diff options
Diffstat (limited to 'mlir/test/Examples/NVGPU/Ch2.py')
| -rw-r--r-- | mlir/test/Examples/NVGPU/Ch2.py | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/mlir/test/Examples/NVGPU/Ch2.py b/mlir/test/Examples/NVGPU/Ch2.py index 78c14cb2c7ad..729913c6d5c4 100644 --- a/mlir/test/Examples/NVGPU/Ch2.py +++ b/mlir/test/Examples/NVGPU/Ch2.py @@ -27,7 +27,7 @@ import numpy as np @NVDSL.mlir_func def saxpy(x, y, alpha): - token_ty = ir.Type.parse("!gpu.async.token") + token_ty = gpu.AsyncTokenType.get() t1 = gpu.wait(token_ty, []) x_dev, t2 = gpu.alloc(x.type, token_ty, [t1], [], []) y_dev, t3 = gpu.alloc(y.type, token_ty, [t2], [], []) |