[TritonIR] Make mask operand optional (#74)

2022-08-22 22:00:17 -07:00
parent de2dd04c8a
commit 0ebef11c77
14 changed files with 113 additions and 102 deletions
--- a/test/Analysis/test-alignment.mlir
+++ b/test/Analysis/test-alignment.mlir
@@ -47,6 +47,6 @@ func @permute_2d(%arg0: !tt.ptr<f32> {tt.divisibility = 16 : i32}, %arg1: i32 {t
  %19 = tt.getelementptr %17, %18 : tensor<128x128x!tt.ptr<f32>>
  // CHECK-NEXT: Contiguity: [1, 1] ; Divisibility: [1, 1] ; Constancy: [1, 1]
  %20 = tt.load %10, %cst, %cst_0 {cache = 1 : i32, evict = 1 : i32, isVolatile = false} : tensor<128x128xf32>
-  tt.store %19, %20, %cst, : tensor<128x128xf32>
+  tt.store %19, %20, %cst : tensor<128x128xf32>
  return
-}
+}