[quidditch_snitch] Improve wait_for_dma_transfers without tokens sy…

…ntax (#111)
opencompl · Aug 17, 2024 · 9776fed · 9776fed
1 parent 539c632
commit 9776fed
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 1 deletion.
diff --git a/codegen/compiler/src/Quidditch/Dialect/Snitch/IR/QuidditchSnitchOps.td b/codegen/compiler/src/Quidditch/Dialect/Snitch/IR/QuidditchSnitchOps.td
@@ -282,7 +282,7 @@ def QuidditchSnitch_WaitForDMATransfersOp
   );
 
   let assemblyFormat = [{
-    $tokens `:` type($tokens) attr-dict
+    ($tokens^ `:` type($tokens))? attr-dict
   }];
 
   let hasFolder = 1;

diff --git a/codegen/tests/Dialect/Snitch/IR/roundtrip.mlir b/codegen/tests/Dialect/Snitch/IR/roundtrip.mlir
@@ -5,6 +5,7 @@ func.func @test(%arg0 : memref<f64>) {
   ^bb0(%arg1 : memref<f64>):
 
   }
+  quidditch_snitch.wait_for_dma_transfers
   return
 }
-Original file line number
+Diff line change
@@ Expand Up / @@ -5,6 +5,7 @@ func.func @test(%arg0 : memref<f64>) { @@
       ^bb0(%arg1 : memref<f64>):
       }
+      quidditch_snitch.wait_for_dma_transfers
       return
     }
@@ Expand Down @@