[OPTIMIZER] Let the pipeline pass insert async wait. (#63)

2022-08-19 01:31:57 +08:00
parent d69ce77b19
commit 8776ad1a0e
3 changed files with 28 additions and 0 deletions
--- a/include/triton/Dialect/TritonGPU/IR/TritonGPUOps.td
+++ b/include/triton/Dialect/TritonGPU/IR/TritonGPUOps.td
@@ -34,6 +34,8 @@ def TTG_AsyncWaitOp : TTG_Op<"async_wait"> {
  let summary = "async wait";

  let arguments = (ins I32Attr:$num);
+
+  let assemblyFormat = "attr-dict";
 }

 def TTG_CopyAsyncOp : TTG_Op<"copy_async",