diff options
Diffstat (limited to 'mlir/test/Examples/NVGPU/Ch5.py')
| -rw-r--r-- | mlir/test/Examples/NVGPU/Ch5.py | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/mlir/test/Examples/NVGPU/Ch5.py b/mlir/test/Examples/NVGPU/Ch5.py index 92e9314e1b81..f98cfd758a75 100644 --- a/mlir/test/Examples/NVGPU/Ch5.py +++ b/mlir/test/Examples/NVGPU/Ch5.py @@ -252,7 +252,7 @@ def epilogue(D: WGMMAMatrix, d_dev): @NVDSL.mlir_func def gemm_warp_specialized(a, b, d, num_stages): - token_ty = ir.Type.parse("!gpu.async.token") + token_ty = gpu.AsyncTokenType.get() t1 = gpu.wait(token_ty, []) a_dev, t2 = gpu.alloc(a.type, token_ty, [t1], [], []) b_dev, t3 = gpu.alloc(b.type, token_ty, [t2], [], []) |