[BACKEND] Extracting numWarps from tritonGPU module (#39)

2022-08-09 00:40:20 +08:00
parent 920723cf3d
commit 83ef74f248
8 changed files with 66 additions and 17 deletions
--- a/test/Target/tritongpu_to_ptx.mlir
+++ b/test/Target/tritongpu_to_ptx.mlir
@@ -5,7 +5,11 @@
 // CHECK: .target sm_80
 // CHECK: .address_size 64

+module attributes {"triton_gpu.num-warps" = 4 : i32} {
+
 func @test_empty_kernel(%lb : index, %A : !tt.ptr<f16>) {

  return
 }
+
+}