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

This commit is contained in:
Yan Chunwei
2022-08-09 00:40:20 +08:00
committed by GitHub
parent 920723cf3d
commit 83ef74f248
8 changed files with 66 additions and 17 deletions

View File

@@ -6,8 +6,11 @@
// CHECK: !nvvm.annotations
// CHECK: !{void (i64, half addrspace(1)*)* @test_empty_kernel, !"maxntidx", i32 128}
module attributes {"triton_gpu.num-warps" = 4 : i32} {
func @test_empty_kernel(%lb : index, %A : !tt.ptr<f16>) {
return
}
}

View File

@@ -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
}
}