diff --git a/docs/overview/python-examples.rst b/docs/overview/python-examples.rst index 6e943da95..e201fc434 100644 --- a/docs/overview/python-examples.rst +++ b/docs/overview/python-examples.rst @@ -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]))