Skip to content

Commit 887e139

Browse files
committed
refactor and refresh
1 parent 6931c80 commit 887e139

File tree

2 files changed

+19
-32
lines changed

2 files changed

+19
-32
lines changed

llvm/include/llvm/IR/IntrinsicsNVVM.td

Lines changed: 5 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -2088,18 +2088,18 @@ foreach dim = 1...5 in {
20882088
// Intrinsics for Prefetch and Prefetchu
20892089
let IntrProperties = [IntrArgMemOnly, ReadOnly<ArgIndex<0>>, NoCapture<ArgIndex<0>>] in {
20902090
foreach level = ["L1", "L2"] in {
2091-
def int_nvvm_prefetch_ # level : Intrinsic<[], [llvm_ptr_ty]>;
2092-
def int_nvvm_prefetch_global_ # level : Intrinsic<[], [llvm_global_ptr_ty]>;
2093-
def int_nvvm_prefetch_local_ # level : Intrinsic<[], [llvm_local_ptr_ty]>;
2091+
def int_nvvm_prefetch_ # level : DefaultAttrsIntrinsic<[], [llvm_ptr_ty]>;
2092+
def int_nvvm_prefetch_global_ # level : DefaultAttrsIntrinsic<[], [llvm_global_ptr_ty]>;
2093+
def int_nvvm_prefetch_local_ # level : DefaultAttrsIntrinsic<[], [llvm_local_ptr_ty]>;
20942094
}
20952095

20962096
def int_nvvm_prefetch_generic_tensormap : DefaultAttrsIntrinsic<[], [llvm_ptr_ty]>;
20972097
def int_nvvm_prefetch_const_tensormap : DefaultAttrsIntrinsic<[], [llvm_constant_ptr_ty]>;
20982098

20992099
foreach eviction_priority = ["evict_normal", "evict_last"] in
2100-
def int_nvvm_prefetch_global_L2_ # eviction_priority : Intrinsic<[], [llvm_global_ptr_ty]>;
2100+
def int_nvvm_prefetch_global_L2_ # eviction_priority : DefaultAttrsIntrinsic<[], [llvm_global_ptr_ty]>;
21012101

2102-
def int_nvvm_prefetchu_L1 : Intrinsic<[], [llvm_ptr_ty]>;
2102+
def int_nvvm_prefetchu_L1 : DefaultAttrsIntrinsic<[], [llvm_ptr_ty]>;
21032103
}
21042104

21052105
// applypriority

llvm/lib/Target/NVPTX/NVPTXIntrinsics.td

Lines changed: 14 additions & 27 deletions
Original file line numberDiff line numberDiff line change
@@ -746,38 +746,25 @@ foreach dim = [1, 2, 3, 4, 5] in {
746746

747747
//Prefetch and Prefetchu
748748

749-
class PREFETCH_INTRS<string InstName> :
749+
class PREFETCH_INTRS<string InstName, string IntrName> :
750750
BasicNVPTXInst<(outs), (ins ADDR:$addr),
751751
InstName,
752-
[(!cast<Intrinsic>(!strconcat("int_nvvm_",
753-
!subst(".", "_", InstName))) addr:$addr)]>,
752+
[(!cast<Intrinsic>(IntrName) addr:$addr)]>,
754753
Requires<[hasPTX<80>, hasSM<90>]>;
755754

756755

757-
def PREFETCH_L1 : PREFETCH_INTRS<"prefetch.L1">;
758-
def PREFETCH_L2 : PREFETCH_INTRS<"prefetch.L2">;
759-
def PREFETCH_GLOBAL_L1 : PREFETCH_INTRS<"prefetch.global.L1">;
760-
def PREFETCH_LOCAL_L1 : PREFETCH_INTRS<"prefetch.local.L1">;
761-
def PREFETCH_GLOBAL_L2 : PREFETCH_INTRS<"prefetch.global.L2">;
762-
def PREFETCH_LOCAL_L2 : PREFETCH_INTRS<"prefetch.local.L2">;
763-
def PREFETCH_CONST_TENSORMAP : PREFETCH_INTRS<"prefetch.const.tensormap">;
764-
def PREFETCH_GENERIC_TENSORMAP : BasicNVPTXInst<(outs), (ins ADDR:$addr),
765-
"prefetch.tensormap",
766-
[(int_nvvm_prefetch_generic_tensormap addr:$addr)]>,
767-
Requires<[hasPTX<80>, hasSM<90>]>;
768-
769-
def PREFETCH_GLOBAL_L2_EVICT_NORMAL : BasicNVPTXInst<(outs), (ins ADDR:$addr),
770-
"prefetch.global.L2::evict_normal",
771-
[(int_nvvm_prefetch_global_L2_evict_normal addr:$addr)]>,
772-
Requires<[hasPTX<80>, hasSM<90>]>;
773-
774-
def PREFETCH_GLOBAL_L2_EVICT_LAST : BasicNVPTXInst<(outs), (ins ADDR:$addr),
775-
"prefetch.global.L2::evict_last",
776-
[(int_nvvm_prefetch_global_L2_evict_last addr:$addr)]>,
777-
Requires<[hasPTX<80>, hasSM<90>]>;
778-
779-
780-
def PREFETCHU_L1 : PREFETCH_INTRS<"prefetchu.L1">;
756+
def PREFETCH_L1 : PREFETCH_INTRS<"prefetch.L1", "int_nvvm_prefetch_L1">;
757+
def PREFETCH_L2 : PREFETCH_INTRS<"prefetch.L2", "int_nvvm_prefetch_L2">;
758+
def PREFETCH_GLOBAL_L1 : PREFETCH_INTRS<"prefetch.global.L1", "int_nvvm_prefetch_global_L1">;
759+
def PREFETCH_LOCAL_L1 : PREFETCH_INTRS<"prefetch.local.L1", "int_nvvm_prefetch_local_L1">;
760+
def PREFETCH_GLOBAL_L2 : PREFETCH_INTRS<"prefetch.global.L2", "int_nvvm_prefetch_global_L2">;
761+
def PREFETCH_LOCAL_L2 : PREFETCH_INTRS<"prefetch.local.L2", "int_nvvm_prefetch_local_L2">;
762+
def PREFETCH_CONST_TENSORMAP : PREFETCH_INTRS<"prefetch.const.tensormap", "int_nvvm_prefetch_const_tensormap">;
763+
def PREFETCH_GENERIC_TENSORMAP : PREFETCH_INTRS<"prefetch.tensormap", "int_nvvm_prefetch_generic_tensormap">;
764+
def PREFETCH_GLOBAL_L2_EVICT_NORMAL : PREFETCH_INTRS<"prefetch.global.L2::evict_normal", "int_nvvm_prefetch_global_L2_evict_normal">;
765+
def PREFETCH_GLOBAL_L2_EVICT_LAST : PREFETCH_INTRS<"prefetch.global.L2::evict_last", "int_nvvm_prefetch_global_L2_evict_last">;
766+
767+
def PREFETCHU_L1 : PREFETCH_INTRS<"prefetchu.L1", "int_nvvm_prefetchu_L1">;
781768

782769
//Applypriority intrinsics
783770
class APPLYPRIORITY_L2_INTRS<string addrspace> :

0 commit comments

Comments
 (0)