diff --git a/docs/overview/python-examples.rst b/docs/overview/python-examples.rst index ac6417928..e201fc434 100644 --- a/docs/overview/python-examples.rst +++ b/docs/overview/python-examples.rst @@ -26,7 +26,7 @@ Then you can interact with it from your interpreter. Below is the same sample as sq = mgr.sequence() - sq.eval(OpTensorSyncLocal([tensor_in_a, tensor_in_b, tensor_out])) + sq.eval(OpTensorSyncDevice([tensor_in_a, tensor_in_b, tensor_out])) # Define the function via PyShader or directly as glsl string or spirv bytes @python2shader @@ -37,11 +37,11 @@ Then you can interact with it from your interpreter. Below is the same sample as i = index.x data3[i] = data1[i] * data2[i] - algo = mgr.algorithm([tensor_in_a, tensor_in_b, tensor_out], compute_shader_multiply.to_spirv()) + algo = mgr.algorithm([tensor_in_a, tensor_in_b, tensor_out], compute_shader_multiply.to_spirv()) # Run shader operation synchronously sq.eval(OpAlgoDispatch(algo)) - sq.eval(OpAlgoSyncLocal([tensor_out])) + sq.eval(OpTensorSyncLocal([tensor_out])) assert tensor_out.data().tolist() == [2.0, 4.0, 6.0] @@ -61,13 +61,12 @@ Similarly you can find the same extended example as above: mgr = Manager(0, [2]) # Can be initialized with List[] or np.Array - tensor_in_a = Tensor([2, 2, 2]) - tensor_in_b = Tensor([1, 2, 3]) - tensor_out = Tensor([0, 0, 0]) + tensor_in_a = mgr.tensor([2, 2, 2]) + tensor_in_b = mgr.tensor([1, 2, 3]) + tensor_out = mgr.tensor([0, 0, 0]) - mgr.eval_tensor_create_def([tensor_in_a, tensor_in_b, tensor_out]) - - seq = mgr.create_sequence("op") + seq = mgr.sequence() + seq.eval(kp.OpTensorSyncDevice([tensor_in_a, tensor_in_b, tensor_out])) # Define the function via PyShader or directly as glsl string or spirv bytes @python2shader @@ -81,8 +80,8 @@ Similarly you can find the same extended example as above: algo = mgr.algorithm([tensor_in_a, tensor_in_b, tensor_out], compute_shader_multiply.to_spirv()) # Run shader operation asynchronously and then await - mgr.eval_async(kp.OpAlgoDispatch(algo))) - mgr.eval_await() + seq.eval_async(kp.OpAlgoDispatch(algo)) + seq.eval_await() seq.record(kp.OpTensorSyncLocal([tensor_in_a])) seq.record(kp.OpTensorSyncLocal([tensor_in_b]))