more tests

2022-06-05 15:10:09 +08:00
parent a4a2c72173
commit bbf75b492f
2 changed files with 130 additions and 2 deletions
--- a/test/TritonGPU/layout.mlir
+++ b/test/TritonGPU/layout.mlir
@@ -4,14 +4,14 @@
  threadTileSize = [1, 1],
  warpTileSize = [32, 1],
  blockTileSize = [64, 1],
-  order = [0, 1]
+  order = [0]
 }>

 #reg2 = #triton_gpu.sharded_layout<{
  threadTileSize = [2, 1],
  warpTileSize = [64, 1],
  blockTileSize = [128, 1],
-  order = [0, 1]
+  order = [0]
 }>

 func @add(%arg0: tensor<256xi32, #reg>, %arg1: tensor<256xi32, #reg>) {