[BACKEND] Extracting numWarps from tritonGPU module (#39)
This commit is contained in:
@@ -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
|
||||
}
|
||||
|
||||
}
|
||||
|
@@ -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
|
||||
}
|
||||
|
||||
}
|
||||
|
Reference in New Issue
Block a user