//===-- MIMGInstructions.td - MIMG Instruction Defintions -----------------===// // // The LLVM Compiler Infrastructure // // This file is distributed under the University of Illinois Open Source // License. See LICENSE.TXT for details. // //===----------------------------------------------------------------------===// class MIMG_Mask { string Op = op; int Channels = channels; } class MIMG_Atomic_Size { string Op = op; int AtomicSize = !if(is32Bit, 1, 2); } class mimg si, bits<7> vi = si> { field bits<7> SI = si; field bits<7> VI = vi; } class MIMG_Helper : MIMG { let mayLoad = 1; let mayStore = 0; let hasPostISelHook = 1; let DecoderNamespace = dns; let isAsmParserOnly = !if(!eq(dns,""), 1, 0); let AsmMatchConverter = "cvtMIMG"; let usesCustomInserter = 1; let SchedRW = [WriteVMEM]; } class MIMG_NoSampler_Helper op, string asm, RegisterClass dst_rc, RegisterClass addr_rc, bit d16_bit=0, string dns=""> : MIMG_Helper < (outs dst_rc:$vdata), (ins addr_rc:$vaddr, SReg_256:$srsrc, dmask:$dmask, unorm:$unorm, GLC:$glc, slc:$slc, r128:$r128, tfe:$tfe, lwe:$lwe, da:$da), asm#" $vdata, $vaddr, $srsrc$dmask$unorm$glc$slc$r128$tfe$lwe$da"#!if(d16_bit, " d16", ""), dns>, MIMGe { let ssamp = 0; let D16 = d16; } multiclass MIMG_NoSampler_Src_Helper_Helper op, string asm, RegisterClass dst_rc, int channels, bit d16_bit, string suffix> { def NAME # _V1 # suffix : MIMG_NoSampler_Helper , MIMG_Mask; def NAME # _V2 # suffix : MIMG_NoSampler_Helper , MIMG_Mask; def NAME # _V3 # suffix : MIMG_NoSampler_Helper , MIMG_Mask; def NAME # _V4 # suffix : MIMG_NoSampler_Helper , MIMG_Mask; } multiclass MIMG_NoSampler_Src_Helper op, string asm, RegisterClass dst_rc, int channels> { defm NAME : MIMG_NoSampler_Src_Helper_Helper ; let d16 = 1 in { let SubtargetPredicate = HasPackedD16VMem in { defm NAME : MIMG_NoSampler_Src_Helper_Helper ; } // End HasPackedD16VMem. let SubtargetPredicate = HasUnpackedD16VMem, DecoderNamespace = "GFX80_UNPACKED" in { defm NAME : MIMG_NoSampler_Src_Helper_Helper ; } // End HasUnpackedD16VMem. } // End d16 = 1. } multiclass MIMG_NoSampler op, string asm> { defm _V1 : MIMG_NoSampler_Src_Helper ; defm _V2 : MIMG_NoSampler_Src_Helper ; defm _V3 : MIMG_NoSampler_Src_Helper ; defm _V4 : MIMG_NoSampler_Src_Helper ; } multiclass MIMG_PckNoSampler op, string asm> { defm NAME # _V1 : MIMG_NoSampler_Src_Helper_Helper ; defm NAME # _V2 : MIMG_NoSampler_Src_Helper_Helper ; defm NAME # _V3 : MIMG_NoSampler_Src_Helper_Helper ; defm NAME # _V4 : MIMG_NoSampler_Src_Helper_Helper ; } class MIMG_Store_Helper op, string asm, RegisterClass data_rc, RegisterClass addr_rc, bit d16_bit=0, string dns = ""> : MIMG_Helper < (outs), (ins data_rc:$vdata, addr_rc:$vaddr, SReg_256:$srsrc, dmask:$dmask, unorm:$unorm, GLC:$glc, slc:$slc, r128:$r128, tfe:$tfe, lwe:$lwe, da:$da), asm#" $vdata, $vaddr, $srsrc$dmask$unorm$glc$slc$r128$tfe$lwe$da"#!if(d16_bit, " d16", ""), dns>, MIMGe { let ssamp = 0; let mayLoad = 0; let mayStore = 1; let hasSideEffects = 0; let hasPostISelHook = 0; let DisableWQM = 1; let D16 = d16; } multiclass MIMG_Store_Addr_Helper_Helper op, string asm, RegisterClass data_rc, int channels, bit d16_bit, string suffix> { def NAME # _V1 # suffix : MIMG_Store_Helper , MIMG_Mask; def NAME # _V2 # suffix : MIMG_Store_Helper , MIMG_Mask; def NAME # _V3 # suffix : MIMG_Store_Helper , MIMG_Mask; def NAME # _V4 # suffix : MIMG_Store_Helper , MIMG_Mask; } multiclass MIMG_Store_Addr_Helper op, string asm, RegisterClass data_rc, int channels> { defm NAME : MIMG_Store_Addr_Helper_Helper ; let d16 = 1 in { let SubtargetPredicate = HasPackedD16VMem in { defm NAME : MIMG_Store_Addr_Helper_Helper ; } // End HasPackedD16VMem. let SubtargetPredicate = HasUnpackedD16VMem, DecoderNamespace = "GFX80_UNPACKED" in { defm NAME : MIMG_Store_Addr_Helper_Helper ; } // End HasUnpackedD16VMem. } // End d16 = 1. } multiclass MIMG_Store op, string asm> { defm _V1 : MIMG_Store_Addr_Helper ; defm _V2 : MIMG_Store_Addr_Helper ; defm _V3 : MIMG_Store_Addr_Helper ; defm _V4 : MIMG_Store_Addr_Helper ; } multiclass MIMG_PckStore op, string asm> { defm NAME # _V1 : MIMG_Store_Addr_Helper_Helper ; defm NAME # _V2 : MIMG_Store_Addr_Helper_Helper ; defm NAME # _V3 : MIMG_Store_Addr_Helper_Helper ; defm NAME # _V4 : MIMG_Store_Addr_Helper_Helper ; } class MIMG_Atomic_Helper : MIMG_Helper < (outs data_rc:$vdst), (ins data_rc:$vdata, addr_rc:$vaddr, SReg_256:$srsrc, dmask:$dmask, unorm:$unorm, GLC:$glc, slc:$slc, r128:$r128, tfe:$tfe, lwe:$lwe, da:$da), asm#" $vdst, $vaddr, $srsrc$dmask$unorm$glc$slc$r128$tfe$lwe$da", !if(enableDasm, dns, "")> { let mayLoad = 1; let mayStore = 1; let hasSideEffects = 1; // FIXME: Remove this let hasPostISelHook = 0; let DisableWQM = 1; let Constraints = "$vdst = $vdata"; let AsmMatchConverter = "cvtMIMGAtomic"; } class MIMG_Atomic_Real_si : MIMG_Atomic_Helper, SIMCInstr, MIMGe { let isCodeGenOnly = 0; let AssemblerPredicates = [isSICI]; let DisableDecoder = DisableSIDecoder; } class MIMG_Atomic_Real_vi : MIMG_Atomic_Helper, SIMCInstr, MIMGe { let isCodeGenOnly = 0; let AssemblerPredicates = [isVI]; let DisableDecoder = DisableVIDecoder; } multiclass MIMG_Atomic_Helper_m { let isPseudo = 1, isCodeGenOnly = 1 in { def "" : MIMG_Atomic_Helper, SIMCInstr; } let ssamp = 0 in { def _si : MIMG_Atomic_Real_si, MIMG_Atomic_Size; def _vi : MIMG_Atomic_Real_vi, MIMG_Atomic_Size; } } multiclass MIMG_Atomic_Addr_Helper_m { // _V* variants have different address size, but the size is not encoded. // So only one variant can be disassembled. V1 looks the safest to decode. defm _V1 : MIMG_Atomic_Helper_m ; defm _V2 : MIMG_Atomic_Helper_m ; defm _V3 : MIMG_Atomic_Helper_m ; defm _V4 : MIMG_Atomic_Helper_m ; } multiclass MIMG_Atomic { // 64-bit atomics // _V* variants have different dst size, but the size is encoded implicitly, // using dmask and tfe. Only 32-bit variant is registered with disassembler. // Other variants are reconstructed by disassembler using dmask and tfe. defm _V1 : MIMG_Atomic_Addr_Helper_m ; defm _V2 : MIMG_Atomic_Addr_Helper_m ; } class MIMG_Sampler_Helper op, string asm, RegisterClass dst_rc, RegisterClass src_rc, bit wqm, bit d16_bit=0, string dns=""> : MIMG_Helper < (outs dst_rc:$vdata), (ins src_rc:$vaddr, SReg_256:$srsrc, SReg_128:$ssamp, dmask:$dmask, unorm:$unorm, GLC:$glc, slc:$slc, r128:$r128, tfe:$tfe, lwe:$lwe, da:$da), asm#" $vdata, $vaddr, $srsrc, $ssamp$dmask$unorm$glc$slc$r128$tfe$lwe$da"#!if(d16_bit, " d16", ""), dns>, MIMGe { let WQM = wqm; let D16 = d16; } multiclass MIMG_Sampler_Src_Helper_Helper op, string asm, RegisterClass dst_rc, int channels, bit wqm, bit d16_bit, string suffix> { def _V1 # suffix : MIMG_Sampler_Helper , MIMG_Mask; def _V2 # suffix : MIMG_Sampler_Helper , MIMG_Mask; def _V3 # suffix : MIMG_Sampler_Helper , MIMG_Mask; def _V4 # suffix : MIMG_Sampler_Helper , MIMG_Mask; def _V8 # suffix : MIMG_Sampler_Helper , MIMG_Mask; def _V16 # suffix : MIMG_Sampler_Helper , MIMG_Mask; } multiclass MIMG_Sampler_Src_Helper op, string asm, RegisterClass dst_rc, int channels, bit wqm> { defm : MIMG_Sampler_Src_Helper_Helper ; let d16 = 1 in { let SubtargetPredicate = HasPackedD16VMem in { defm : MIMG_Sampler_Src_Helper_Helper ; } // End HasPackedD16VMem. let SubtargetPredicate = HasUnpackedD16VMem, DecoderNamespace = "GFX80_UNPACKED" in { defm : MIMG_Sampler_Src_Helper_Helper ; } // End HasUnpackedD16VMem. } // End d16 = 1. } multiclass MIMG_Sampler op, string asm, bit wqm=0> { defm _V1 : MIMG_Sampler_Src_Helper; defm _V2 : MIMG_Sampler_Src_Helper; defm _V3 : MIMG_Sampler_Src_Helper; defm _V4 : MIMG_Sampler_Src_Helper; } multiclass MIMG_Sampler_WQM op, string asm> : MIMG_Sampler; class MIMG_Gather_Helper op, string asm, RegisterClass dst_rc, RegisterClass src_rc, bit wqm, bit d16_bit=0, string dns=""> : MIMG < (outs dst_rc:$vdata), (ins src_rc:$vaddr, SReg_256:$srsrc, SReg_128:$ssamp, dmask:$dmask, unorm:$unorm, GLC:$glc, slc:$slc, r128:$r128, tfe:$tfe, lwe:$lwe, da:$da), asm#" $vdata, $vaddr, $srsrc, $ssamp$dmask$unorm$glc$slc$r128$tfe$lwe$da"#!if(d16_bit, " d16", ""), []>, MIMGe { let mayLoad = 1; let mayStore = 0; // DMASK was repurposed for GATHER4. 4 components are always // returned and DMASK works like a swizzle - it selects // the component to fetch. The only useful DMASK values are // 1=red, 2=green, 4=blue, 8=alpha. (e.g. 1 returns // (red,red,red,red) etc.) The ISA document doesn't mention // this. // Therefore, disable all code which updates DMASK by setting this: let Gather4 = 1; let hasPostISelHook = 0; let WQM = wqm; let D16 = d16; let DecoderNamespace = dns; let isAsmParserOnly = !if(!eq(dns,""), 1, 0); } multiclass MIMG_Gather_Src_Helper op, string asm, RegisterClass dst_rc, bit wqm, bit d16_bit, string prefix, string suffix> { def prefix # _V1 # suffix : MIMG_Gather_Helper ; def prefix # _V2 # suffix : MIMG_Gather_Helper ; def prefix # _V3 # suffix : MIMG_Gather_Helper ; def prefix # _V4 # suffix : MIMG_Gather_Helper ; def prefix # _V8 # suffix : MIMG_Gather_Helper ; def prefix # _V16 # suffix : MIMG_Gather_Helper ; } multiclass MIMG_Gather op, string asm, bit wqm=0> { defm : MIMG_Gather_Src_Helper; let d16 = 1 in { let AssemblerPredicate = HasPackedD16VMem in { defm : MIMG_Gather_Src_Helper; } // End HasPackedD16VMem. let AssemblerPredicate = HasUnpackedD16VMem, DecoderNamespace = "GFX80_UNPACKED" in { defm : MIMG_Gather_Src_Helper; } // End HasUnpackedD16VMem. } // End d16 = 1. } multiclass MIMG_Gather_WQM op, string asm> : MIMG_Gather; //===----------------------------------------------------------------------===// // MIMG Instructions //===----------------------------------------------------------------------===// let SubtargetPredicate = isGCN in { defm IMAGE_LOAD : MIMG_NoSampler <0x00000000, "image_load">; defm IMAGE_LOAD_MIP : MIMG_NoSampler <0x00000001, "image_load_mip">; defm IMAGE_LOAD_PCK : MIMG_PckNoSampler <0x00000002, "image_load_pck">; defm IMAGE_LOAD_PCK_SGN : MIMG_PckNoSampler <0x00000003, "image_load_pck_sgn">; defm IMAGE_LOAD_MIP_PCK : MIMG_PckNoSampler <0x00000004, "image_load_mip_pck">; defm IMAGE_LOAD_MIP_PCK_SGN : MIMG_PckNoSampler <0x00000005, "image_load_mip_pck_sgn">; defm IMAGE_STORE : MIMG_Store <0x00000008, "image_store">; defm IMAGE_STORE_MIP : MIMG_Store <0x00000009, "image_store_mip">; defm IMAGE_STORE_PCK : MIMG_PckStore <0x0000000a, "image_store_pck">; defm IMAGE_STORE_MIP_PCK : MIMG_PckStore <0x0000000b, "image_store_mip_pck">; let mayLoad = 0, mayStore = 0 in { defm IMAGE_GET_RESINFO : MIMG_NoSampler <0x0000000e, "image_get_resinfo">; } defm IMAGE_ATOMIC_SWAP : MIMG_Atomic , "image_atomic_swap">; defm IMAGE_ATOMIC_CMPSWAP : MIMG_Atomic , "image_atomic_cmpswap", VReg_64, VReg_128>; defm IMAGE_ATOMIC_ADD : MIMG_Atomic , "image_atomic_add">; defm IMAGE_ATOMIC_SUB : MIMG_Atomic , "image_atomic_sub">; //def IMAGE_ATOMIC_RSUB : MIMG_NoPattern_ <"image_atomic_rsub", 0x00000013>; -- not on VI defm IMAGE_ATOMIC_SMIN : MIMG_Atomic , "image_atomic_smin">; defm IMAGE_ATOMIC_UMIN : MIMG_Atomic , "image_atomic_umin">; defm IMAGE_ATOMIC_SMAX : MIMG_Atomic , "image_atomic_smax">; defm IMAGE_ATOMIC_UMAX : MIMG_Atomic , "image_atomic_umax">; defm IMAGE_ATOMIC_AND : MIMG_Atomic , "image_atomic_and">; defm IMAGE_ATOMIC_OR : MIMG_Atomic , "image_atomic_or">; defm IMAGE_ATOMIC_XOR : MIMG_Atomic , "image_atomic_xor">; defm IMAGE_ATOMIC_INC : MIMG_Atomic , "image_atomic_inc">; defm IMAGE_ATOMIC_DEC : MIMG_Atomic , "image_atomic_dec">; //def IMAGE_ATOMIC_FCMPSWAP : MIMG_NoPattern_ <"image_atomic_fcmpswap", 0x0000001d>; -- not on VI //def IMAGE_ATOMIC_FMIN : MIMG_NoPattern_ <"image_atomic_fmin", 0x0000001e>; -- not on VI //def IMAGE_ATOMIC_FMAX : MIMG_NoPattern_ <"image_atomic_fmax", 0x0000001f>; -- not on VI defm IMAGE_SAMPLE : MIMG_Sampler_WQM <0x00000020, "image_sample">; defm IMAGE_SAMPLE_CL : MIMG_Sampler_WQM <0x00000021, "image_sample_cl">; defm IMAGE_SAMPLE_D : MIMG_Sampler <0x00000022, "image_sample_d">; defm IMAGE_SAMPLE_D_CL : MIMG_Sampler <0x00000023, "image_sample_d_cl">; defm IMAGE_SAMPLE_L : MIMG_Sampler <0x00000024, "image_sample_l">; defm IMAGE_SAMPLE_B : MIMG_Sampler_WQM <0x00000025, "image_sample_b">; defm IMAGE_SAMPLE_B_CL : MIMG_Sampler_WQM <0x00000026, "image_sample_b_cl">; defm IMAGE_SAMPLE_LZ : MIMG_Sampler <0x00000027, "image_sample_lz">; defm IMAGE_SAMPLE_C : MIMG_Sampler_WQM <0x00000028, "image_sample_c">; defm IMAGE_SAMPLE_C_CL : MIMG_Sampler_WQM <0x00000029, "image_sample_c_cl">; defm IMAGE_SAMPLE_C_D : MIMG_Sampler <0x0000002a, "image_sample_c_d">; defm IMAGE_SAMPLE_C_D_CL : MIMG_Sampler <0x0000002b, "image_sample_c_d_cl">; defm IMAGE_SAMPLE_C_L : MIMG_Sampler <0x0000002c, "image_sample_c_l">; defm IMAGE_SAMPLE_C_B : MIMG_Sampler_WQM <0x0000002d, "image_sample_c_b">; defm IMAGE_SAMPLE_C_B_CL : MIMG_Sampler_WQM <0x0000002e, "image_sample_c_b_cl">; defm IMAGE_SAMPLE_C_LZ : MIMG_Sampler <0x0000002f, "image_sample_c_lz">; defm IMAGE_SAMPLE_O : MIMG_Sampler_WQM <0x00000030, "image_sample_o">; defm IMAGE_SAMPLE_CL_O : MIMG_Sampler_WQM <0x00000031, "image_sample_cl_o">; defm IMAGE_SAMPLE_D_O : MIMG_Sampler <0x00000032, "image_sample_d_o">; defm IMAGE_SAMPLE_D_CL_O : MIMG_Sampler <0x00000033, "image_sample_d_cl_o">; defm IMAGE_SAMPLE_L_O : MIMG_Sampler <0x00000034, "image_sample_l_o">; defm IMAGE_SAMPLE_B_O : MIMG_Sampler_WQM <0x00000035, "image_sample_b_o">; defm IMAGE_SAMPLE_B_CL_O : MIMG_Sampler_WQM <0x00000036, "image_sample_b_cl_o">; defm IMAGE_SAMPLE_LZ_O : MIMG_Sampler <0x00000037, "image_sample_lz_o">; defm IMAGE_SAMPLE_C_O : MIMG_Sampler_WQM <0x00000038, "image_sample_c_o">; defm IMAGE_SAMPLE_C_CL_O : MIMG_Sampler_WQM <0x00000039, "image_sample_c_cl_o">; defm IMAGE_SAMPLE_C_D_O : MIMG_Sampler <0x0000003a, "image_sample_c_d_o">; defm IMAGE_SAMPLE_C_D_CL_O : MIMG_Sampler <0x0000003b, "image_sample_c_d_cl_o">; defm IMAGE_SAMPLE_C_L_O : MIMG_Sampler <0x0000003c, "image_sample_c_l_o">; defm IMAGE_SAMPLE_C_B_O : MIMG_Sampler_WQM <0x0000003d, "image_sample_c_b_o">; defm IMAGE_SAMPLE_C_B_CL_O : MIMG_Sampler_WQM <0x0000003e, "image_sample_c_b_cl_o">; defm IMAGE_SAMPLE_C_LZ_O : MIMG_Sampler <0x0000003f, "image_sample_c_lz_o">; defm IMAGE_GATHER4 : MIMG_Gather_WQM <0x00000040, "image_gather4">; defm IMAGE_GATHER4_CL : MIMG_Gather_WQM <0x00000041, "image_gather4_cl">; defm IMAGE_GATHER4_L : MIMG_Gather <0x00000044, "image_gather4_l">; defm IMAGE_GATHER4_B : MIMG_Gather_WQM <0x00000045, "image_gather4_b">; defm IMAGE_GATHER4_B_CL : MIMG_Gather_WQM <0x00000046, "image_gather4_b_cl">; defm IMAGE_GATHER4_LZ : MIMG_Gather <0x00000047, "image_gather4_lz">; defm IMAGE_GATHER4_C : MIMG_Gather_WQM <0x00000048, "image_gather4_c">; defm IMAGE_GATHER4_C_CL : MIMG_Gather_WQM <0x00000049, "image_gather4_c_cl">; defm IMAGE_GATHER4_C_L : MIMG_Gather <0x0000004c, "image_gather4_c_l">; defm IMAGE_GATHER4_C_B : MIMG_Gather_WQM <0x0000004d, "image_gather4_c_b">; defm IMAGE_GATHER4_C_B_CL : MIMG_Gather_WQM <0x0000004e, "image_gather4_c_b_cl">; defm IMAGE_GATHER4_C_LZ : MIMG_Gather <0x0000004f, "image_gather4_c_lz">; defm IMAGE_GATHER4_O : MIMG_Gather_WQM <0x00000050, "image_gather4_o">; defm IMAGE_GATHER4_CL_O : MIMG_Gather_WQM <0x00000051, "image_gather4_cl_o">; defm IMAGE_GATHER4_L_O : MIMG_Gather <0x00000054, "image_gather4_l_o">; defm IMAGE_GATHER4_B_O : MIMG_Gather_WQM <0x00000055, "image_gather4_b_o">; defm IMAGE_GATHER4_B_CL_O : MIMG_Gather <0x00000056, "image_gather4_b_cl_o">; defm IMAGE_GATHER4_LZ_O : MIMG_Gather <0x00000057, "image_gather4_lz_o">; defm IMAGE_GATHER4_C_O : MIMG_Gather_WQM <0x00000058, "image_gather4_c_o">; defm IMAGE_GATHER4_C_CL_O : MIMG_Gather_WQM <0x00000059, "image_gather4_c_cl_o">; defm IMAGE_GATHER4_C_L_O : MIMG_Gather <0x0000005c, "image_gather4_c_l_o">; defm IMAGE_GATHER4_C_B_O : MIMG_Gather_WQM <0x0000005d, "image_gather4_c_b_o">; defm IMAGE_GATHER4_C_B_CL_O : MIMG_Gather_WQM <0x0000005e, "image_gather4_c_b_cl_o">; defm IMAGE_GATHER4_C_LZ_O : MIMG_Gather <0x0000005f, "image_gather4_c_lz_o">; let mayLoad = 0, mayStore = 0 in { defm IMAGE_GET_LOD : MIMG_Sampler_WQM <0x00000060, "image_get_lod">; } defm IMAGE_SAMPLE_CD : MIMG_Sampler <0x00000068, "image_sample_cd">; defm IMAGE_SAMPLE_CD_CL : MIMG_Sampler <0x00000069, "image_sample_cd_cl">; defm IMAGE_SAMPLE_C_CD : MIMG_Sampler <0x0000006a, "image_sample_c_cd">; defm IMAGE_SAMPLE_C_CD_CL : MIMG_Sampler <0x0000006b, "image_sample_c_cd_cl">; defm IMAGE_SAMPLE_CD_O : MIMG_Sampler <0x0000006c, "image_sample_cd_o">; defm IMAGE_SAMPLE_CD_CL_O : MIMG_Sampler <0x0000006d, "image_sample_cd_cl_o">; defm IMAGE_SAMPLE_C_CD_O : MIMG_Sampler <0x0000006e, "image_sample_c_cd_o">; defm IMAGE_SAMPLE_C_CD_CL_O : MIMG_Sampler <0x0000006f, "image_sample_c_cd_cl_o">; //def IMAGE_RSRC256 : MIMG_NoPattern_RSRC256 <"image_rsrc256", 0x0000007e>; //def IMAGE_SAMPLER : MIMG_NoPattern_ <"image_sampler", 0x0000007f>; } /********** ============================== **********/ /********** Dimension-aware image patterns **********/ /********** ============================== **********/ class getDwordsType { int NumDwords = dwords; string suffix = !if(!lt(dwords, 1), ?, !if(!eq(dwords, 1), "_V1", !if(!eq(dwords, 2), "_V2", !if(!le(dwords, 4), "_V4", !if(!le(dwords, 8), "_V8", !if(!le(dwords, 16), "_V16", ?)))))); ValueType VT = !if(!lt(dwords, 1), ?, !if(!eq(dwords, 1), f32, !if(!eq(dwords, 2), v2f32, !if(!le(dwords, 4), v4f32, !if(!le(dwords, 8), v8f32, !if(!le(dwords, 16), v16f32, ?)))))); RegisterClass VReg = !if(!lt(dwords, 1), ?, !if(!eq(dwords, 1), VGPR_32, !if(!eq(dwords, 2), VReg_64, !if(!le(dwords, 4), VReg_128, !if(!le(dwords, 8), VReg_256, !if(!le(dwords, 16), VReg_512, ?)))))); } class makeRegSequence_Fold { int idx = i; dag lhs = d; } // Generate a dag node which returns a vector register of class RC into which // the source operands given by names have been inserted (assuming that each // name corresponds to an operand whose size is equal to a subregister). class makeRegSequence names> { dag ret = !if(!eq(!size(names), 1), !dag(COPY_TO_REGCLASS, [?, RC], [names[0], ?]), !foldl(makeRegSequence_Fold<0, (vt (IMPLICIT_DEF))>, names, f, name, makeRegSequence_Fold< !add(f.idx, 1), !con((INSERT_SUBREG f.lhs), !dag(INSERT_SUBREG, [?, !cast("sub"#f.idx)], [name, ?]))>).lhs); } class ImageDimPattern : GCNPat<(undef), (undef)> { list AddrArgs = I.P.AddrDefaultArgs; getDwordsType AddrDwords = getDwordsType; Instruction MI = !cast(!strconcat("IMAGE_", I.P.OpMod, dop, AddrDwords.suffix, suffix)); // DAG fragment to match data arguments (vdata for store/atomic, dmask // for non-atomic). dag MatchDataDag = !con(!dag(I, !foreach(arg, I.P.DataArgs, dty), !foreach(arg, I.P.DataArgs, arg.Name)), !if(I.P.IsAtomic, (I), (I i32:$dmask))); // DAG fragment to match vaddr arguments. dag MatchAddrDag = !dag(I, !foreach(arg, AddrArgs, arg.Type.VT), !foreach(arg, AddrArgs, arg.Name)); // DAG fragment to match sampler resource and unorm arguments. dag MatchSamplerDag = !if(I.P.IsSample, (I v4i32:$sampler, i1:$unorm), (I)); // DAG node that generates the MI vdata for store/atomic getDwordsType DataDwords = getDwordsType; dag GenDataDag = !if(I.P.IsAtomic, (MI makeRegSequence.ret), !if(!size(I.P.DataArgs), (MI $vdata), (MI))); // DAG node that generates the MI vaddr dag GenAddrDag = makeRegSequence.ret; // DAG fragments that generate various inline flags dag GenDmask = !if(I.P.IsAtomic, (MI !add(!shl(1, DataDwords.NumDwords), -1)), (MI (as_i32imm $dmask))); dag GenGLC = !if(I.P.IsAtomic, (MI 1), (MI (bitextract_imm<0> $cachepolicy))); dag MatchIntrinsic = !con(MatchDataDag, MatchAddrDag, (I v8i32:$rsrc), MatchSamplerDag, (I 0/*texfailctrl*/, i32:$cachepolicy)); let PatternToMatch = !if(!size(I.RetTypes), (dty MatchIntrinsic), MatchIntrinsic); bit IsCmpSwap = !and(I.P.IsAtomic, !eq(!size(I.P.DataArgs), 2)); dag ImageInstruction = !con(GenDataDag, (MI GenAddrDag), (MI $rsrc), !if(I.P.IsSample, (MI $sampler), (MI)), GenDmask, !if(I.P.IsSample, (MI (as_i1imm $unorm)), (MI 1)), GenGLC, (MI (bitextract_imm<1> $cachepolicy), 0, /* r128 */ 0, /* tfe */ 0 /*(as_i1imm $lwe)*/, { I.P.Dim.DA })); let ResultInstrs = [ !if(IsCmpSwap, (EXTRACT_SUBREG ImageInstruction, sub0), ImageInstruction) ]; } foreach intr = !listconcat(AMDGPUImageDimIntrinsics, AMDGPUImageDimGetResInfoIntrinsics) in { def intr#_pat_v1 : ImageDimPattern; def intr#_pat_v2 : ImageDimPattern; def intr#_pat_v4 : ImageDimPattern; } // v2f16 and v4f16 are used as data types to signal that D16 should be used. // However, they are not (always) legal types, and the SelectionDAG requires us // to legalize them before running any patterns. So we legalize them by // converting to an int type of equal size and using an internal 'd16helper' // intrinsic instead which signifies both the use of D16 and actually allows // this integer-based return type. multiclass ImageDimD16Helper { let SubtargetPredicate = HasUnpackedD16VMem in { def _unpacked_v1 : ImageDimPattern; def _unpacked_v2 : ImageDimPattern; def _unpacked_v4 : ImageDimPattern; } // End HasUnpackedD16VMem. let SubtargetPredicate = HasPackedD16VMem in { def _packed_v1 : ImageDimPattern; // used on gfx810 def _packed_v2 : ImageDimPattern; // used on gfx900 def _packed_v2_gfx9 : ImageDimPattern; def _packed_v4 : ImageDimPattern; } // End HasPackedD16VMem. } foreach intr = AMDGPUImageDimIntrinsics in { def intr#_d16helper_profile : AMDGPUDimProfileCopy { let RetTypes = !foreach(ty, intr.P.RetTypes, llvm_any_ty); let DataArgs = !foreach(arg, intr.P.DataArgs, AMDGPUArg); } let TargetPrefix = "SI", isTarget = 1 in def int_SI_image_d16helper_ # intr.P.OpMod # intr.P.Dim.Name : AMDGPUImageDimIntrinsic(intr#"_d16helper_profile"), intr.IntrProperties, intr.Properties>; defm intr#_d16 : ImageDimD16Helper< intr, !cast( "int_SI_image_d16helper_" # intr.P.OpMod # intr.P.Dim.Name)>; } foreach intr = AMDGPUImageDimGatherIntrinsics in { def intr#_pat3 : ImageDimPattern; def intr#_d16helper_profile : AMDGPUDimProfileCopy { let RetTypes = !foreach(ty, intr.P.RetTypes, llvm_any_ty); let DataArgs = !foreach(arg, intr.P.DataArgs, AMDGPUArg); } let TargetPrefix = "SI", isTarget = 1 in def int_SI_image_d16helper_ # intr.P.OpMod # intr.P.Dim.Name : AMDGPUImageDimIntrinsic(intr#"_d16helper_profile"), intr.IntrProperties, intr.Properties>; let SubtargetPredicate = HasUnpackedD16VMem in { def intr#_unpacked_v4 : ImageDimPattern( "int_SI_image_d16helper_" # intr.P.OpMod # intr.P.Dim.Name), "_V4", v4i32, "_D16_gfx80">; } // End HasUnpackedD16VMem. let SubtargetPredicate = HasPackedD16VMem in { def intr#_packed_v4 : ImageDimPattern( "int_SI_image_d16helper_" # intr.P.OpMod # intr.P.Dim.Name), "_V2", v2i32, "_D16">; } // End HasPackedD16VMem. } foreach intr = AMDGPUImageDimAtomicIntrinsics in { def intr#_pat1 : ImageDimPattern; } /********** ======================= **********/ /********** Image sampling patterns **********/ /********** ======================= **********/ // ImageSample for amdgcn // TODO: // 1. Handle v4i32 rsrc type (Register Class for the instruction to be SReg_128). // 2. Add A16 support when we pass address of half type. multiclass ImageSamplePattern { def : GCNPat< (dt (name vt:$addr, v8i32:$rsrc, v4i32:$sampler, i32:$dmask, i1:$unorm, i1:$glc, i1:$slc, i1:$lwe, i1:$da)), (opcode $addr, $rsrc, $sampler, (as_i32imm $dmask), (as_i1imm $unorm), (as_i1imm $glc), (as_i1imm $slc), 0, 0, (as_i1imm $lwe), (as_i1imm $da)) >; } multiclass ImageSampleDataPatterns { defm : ImageSamplePattern(opcode # _V1 # suffix), dt, f32>; defm : ImageSamplePattern(opcode # _V2 # suffix), dt, v2f32>; defm : ImageSamplePattern(opcode # _V4 # suffix), dt, v4f32>; defm : ImageSamplePattern(opcode # _V8 # suffix), dt, v8f32>; defm : ImageSamplePattern(opcode # _V16 # suffix), dt, v16f32>; } // ImageSample patterns. multiclass ImageSamplePatterns { defm : ImageSampleDataPatterns(opcode # _V1), f32>; defm : ImageSampleDataPatterns(opcode # _V2), v2f32>; defm : ImageSampleDataPatterns(opcode # _V4), v4f32>; let SubtargetPredicate = HasUnpackedD16VMem in { defm : ImageSampleDataPatterns(opcode # _V1), f16, "_D16_gfx80">; } // End HasUnpackedD16VMem. let SubtargetPredicate = HasPackedD16VMem in { defm : ImageSampleDataPatterns(opcode # _V1), f16, "_D16">; defm : ImageSampleDataPatterns(opcode # _V1), v2f16, "_D16">; } // End HasPackedD16VMem. } // ImageSample alternative patterns for illegal vector half Types. multiclass ImageSampleAltPatterns { let SubtargetPredicate = HasUnpackedD16VMem in { defm : ImageSampleDataPatterns(opcode # _V2), v2i32, "_D16_gfx80">; defm : ImageSampleDataPatterns(opcode # _V4), v4i32, "_D16_gfx80">; } // End HasUnpackedD16VMem. let SubtargetPredicate = HasPackedD16VMem in { defm : ImageSampleDataPatterns(opcode # _V1), i32, "_D16">; defm : ImageSampleDataPatterns(opcode # _V2), v2i32, "_D16">; } // End HasPackedD16VMem. } // ImageGather4 patterns. multiclass ImageGather4Patterns { defm : ImageSampleDataPatterns(opcode # _V4), v4f32>; } // ImageGather4 alternative patterns for illegal vector half Types. multiclass ImageGather4AltPatterns { let SubtargetPredicate = HasUnpackedD16VMem in { defm : ImageSampleDataPatterns(opcode # _V4), v4i32, "_D16_gfx80">; } // End HasUnpackedD16VMem. let SubtargetPredicate = HasPackedD16VMem in { defm : ImageSampleDataPatterns(opcode # _V2), v2i32, "_D16">; } // End HasPackedD16VMem. } // ImageLoad for amdgcn. multiclass ImageLoadPattern { def : GCNPat < (dt (name vt:$addr, v8i32:$rsrc, i32:$dmask, i1:$glc, i1:$slc, i1:$lwe, i1:$da)), (opcode $addr, $rsrc, (as_i32imm $dmask), 1, (as_i1imm $glc), (as_i1imm $slc), 0, 0, (as_i1imm $lwe), (as_i1imm $da)) >; } multiclass ImageLoadDataPatterns { defm : ImageLoadPattern(opcode # _V1 # suffix), dt, i32>; defm : ImageLoadPattern(opcode # _V2 # suffix), dt, v2i32>; defm : ImageLoadPattern(opcode # _V4 # suffix), dt, v4i32>; } // ImageLoad patterns. // TODO: support v3f32. multiclass ImageLoadPatterns { defm : ImageLoadDataPatterns(opcode # _V1), f32>; defm : ImageLoadDataPatterns(opcode # _V2), v2f32>; defm : ImageLoadDataPatterns(opcode # _V4), v4f32>; let SubtargetPredicate = HasUnpackedD16VMem in { defm : ImageLoadDataPatterns(opcode # _V1), f16, "_D16_gfx80">; } // End HasUnpackedD16VMem. let SubtargetPredicate = HasPackedD16VMem in { defm : ImageLoadDataPatterns(opcode # _V1), f16, "_D16">; defm : ImageLoadDataPatterns(opcode # _V1), v2f16, "_D16">; } // End HasPackedD16VMem. } // ImageLoad alternative patterns for illegal vector half Types. multiclass ImageLoadAltPatterns { let SubtargetPredicate = HasUnpackedD16VMem in { defm : ImageLoadDataPatterns(opcode # _V2), v2i32, "_D16_gfx80">; defm : ImageLoadDataPatterns(opcode # _V4), v4i32, "_D16_gfx80">; } // End HasUnPackedD16VMem. let SubtargetPredicate = HasPackedD16VMem in { defm : ImageLoadDataPatterns(opcode # _V1), i32, "_D16">; defm : ImageLoadDataPatterns(opcode # _V2), v2i32, "_D16">; } // End HasPackedD16VMem. } // ImageStore for amdgcn. multiclass ImageStorePattern { def : GCNPat < (name dt:$data, vt:$addr, v8i32:$rsrc, i32:$dmask, i1:$glc, i1:$slc, i1:$lwe, i1:$da), (opcode $data, $addr, $rsrc, (as_i32imm $dmask), 1, (as_i1imm $glc), (as_i1imm $slc), 0, 0, (as_i1imm $lwe), (as_i1imm $da)) >; } multiclass ImageStoreDataPatterns { defm : ImageStorePattern(opcode # _V1 # suffix), dt, i32>; defm : ImageStorePattern(opcode # _V2 # suffix), dt, v2i32>; defm : ImageStorePattern(opcode # _V4 # suffix), dt, v4i32>; } // ImageStore patterns. // TODO: support v3f32. multiclass ImageStorePatterns { defm : ImageStoreDataPatterns(opcode # _V1), f32>; defm : ImageStoreDataPatterns(opcode # _V2), v2f32>; defm : ImageStoreDataPatterns(opcode # _V4), v4f32>; let SubtargetPredicate = HasUnpackedD16VMem in { defm : ImageStoreDataPatterns(opcode # _V1), f16, "_D16_gfx80">; } // End HasUnpackedD16VMem. let SubtargetPredicate = HasPackedD16VMem in { defm : ImageStoreDataPatterns(opcode # _V1), f16, "_D16">; defm : ImageStoreDataPatterns(opcode # _V1), v2f16, "_D16">; } // End HasPackedD16VMem. } // ImageStore alternative patterns. multiclass ImageStoreAltPatterns { let SubtargetPredicate = HasUnpackedD16VMem in { defm : ImageStoreDataPatterns(opcode # _V2), v2i32, "_D16_gfx80">; defm : ImageStoreDataPatterns(opcode # _V4), v4i32, "_D16_gfx80">; } // End HasUnpackedD16VMem. let SubtargetPredicate = HasPackedD16VMem in { defm : ImageStoreDataPatterns(opcode # _V1), i32, "_D16">; defm : ImageStoreDataPatterns(opcode # _V2), v2i32, "_D16">; } // End HasPackedD16VMem. } // ImageAtomic for amdgcn. class ImageAtomicPattern : GCNPat < (name i32:$vdata, vt:$addr, v8i32:$rsrc, imm:$r128, imm:$da, imm:$slc), (opcode $vdata, $addr, $rsrc, 1, 1, 1, (as_i1imm $slc), (as_i1imm $r128), 0, 0, (as_i1imm $da)) >; // ImageAtomic patterns. multiclass ImageAtomicPatterns { def : ImageAtomicPattern(opcode # _V1_V1), i32>; def : ImageAtomicPattern(opcode # _V1_V2), v2i32>; def : ImageAtomicPattern(opcode # _V1_V4), v4i32>; } // ImageAtomicCmpSwap for amdgcn. class ImageAtomicCmpSwapPattern : GCNPat < (int_amdgcn_image_atomic_cmpswap i32:$vsrc, i32:$vcmp, vt:$addr, v8i32:$rsrc, imm:$r128, imm:$da, imm:$slc), (EXTRACT_SUBREG (opcode (REG_SEQUENCE VReg_64, $vsrc, sub0, $vcmp, sub1), $addr, $rsrc, 3, 1, 1, (as_i1imm $slc), (as_i1imm $r128), 0, 0, (as_i1imm $da)), sub0) >; // ======= amdgcn Image Intrinsics ============== // Image load. defm : ImageLoadPatterns; defm : ImageLoadPatterns; defm : ImageLoadPatterns; defm : ImageLoadAltPatterns; defm : ImageLoadAltPatterns; // Image store. defm : ImageStorePatterns; defm : ImageStorePatterns; defm : ImageStoreAltPatterns; defm : ImageStoreAltPatterns; // Basic sample. defm : ImageSamplePatterns; defm : ImageSamplePatterns; defm : ImageSamplePatterns; defm : ImageSamplePatterns; defm : ImageSamplePatterns; defm : ImageSamplePatterns; defm : ImageSamplePatterns; defm : ImageSamplePatterns; defm : ImageSamplePatterns; defm : ImageSamplePatterns; // Sample with comparison. defm : ImageSamplePatterns; defm : ImageSamplePatterns; defm : ImageSamplePatterns; defm : ImageSamplePatterns; defm : ImageSamplePatterns; defm : ImageSamplePatterns; defm : ImageSamplePatterns; defm : ImageSamplePatterns; defm : ImageSamplePatterns; defm : ImageSamplePatterns; // Sample with offsets. defm : ImageSamplePatterns; defm : ImageSamplePatterns; defm : ImageSamplePatterns; defm : ImageSamplePatterns; defm : ImageSamplePatterns; defm : ImageSamplePatterns; defm : ImageSamplePatterns; defm : ImageSamplePatterns; defm : ImageSamplePatterns; defm : ImageSamplePatterns; // Sample with comparison and offsets. defm : ImageSamplePatterns; defm : ImageSamplePatterns; defm : ImageSamplePatterns; defm : ImageSamplePatterns; defm : ImageSamplePatterns; defm : ImageSamplePatterns; defm : ImageSamplePatterns; defm : ImageSamplePatterns; defm : ImageSamplePatterns; defm : ImageSamplePatterns; // Basic gather4. defm : ImageGather4Patterns; defm : ImageGather4Patterns; defm : ImageGather4Patterns; defm : ImageGather4Patterns; defm : ImageGather4Patterns; defm : ImageGather4Patterns; // Gather4 with comparison. defm : ImageGather4Patterns; defm : ImageGather4Patterns; defm : ImageGather4Patterns; defm : ImageGather4Patterns; defm : ImageGather4Patterns; defm : ImageGather4Patterns; // Gather4 with offsets. defm : ImageGather4Patterns; defm : ImageGather4Patterns; defm : ImageGather4Patterns; defm : ImageGather4Patterns; defm : ImageGather4Patterns; defm : ImageGather4Patterns; // Gather4 with comparison and offsets. defm : ImageGather4Patterns; defm : ImageGather4Patterns; defm : ImageGather4Patterns; defm : ImageGather4Patterns; defm : ImageGather4Patterns; defm : ImageGather4Patterns; // Basic sample alternative. defm : ImageSampleAltPatterns; defm : ImageSampleAltPatterns; defm : ImageSampleAltPatterns; defm : ImageSampleAltPatterns; defm : ImageSampleAltPatterns; defm : ImageSampleAltPatterns; defm : ImageSampleAltPatterns; defm : ImageSampleAltPatterns; defm : ImageSampleAltPatterns; defm : ImageSampleAltPatterns; // Sample with comparison alternative. defm : ImageSampleAltPatterns; defm : ImageSampleAltPatterns; defm : ImageSampleAltPatterns; defm : ImageSampleAltPatterns; defm : ImageSampleAltPatterns; defm : ImageSampleAltPatterns; defm : ImageSampleAltPatterns; defm : ImageSampleAltPatterns; defm : ImageSampleAltPatterns; defm : ImageSampleAltPatterns; // Sample with offsets alternative. defm : ImageSampleAltPatterns; defm : ImageSampleAltPatterns; defm : ImageSampleAltPatterns; defm : ImageSampleAltPatterns; defm : ImageSampleAltPatterns; defm : ImageSampleAltPatterns; defm : ImageSampleAltPatterns; defm : ImageSampleAltPatterns; defm : ImageSampleAltPatterns; defm : ImageSampleAltPatterns; // Sample with comparison and offsets alternative. defm : ImageSampleAltPatterns; defm : ImageSampleAltPatterns; defm : ImageSampleAltPatterns; defm : ImageSampleAltPatterns; defm : ImageSampleAltPatterns; defm : ImageSampleAltPatterns; defm : ImageSampleAltPatterns; defm : ImageSampleAltPatterns; defm : ImageSampleAltPatterns; defm : ImageSampleAltPatterns; // Basic gather4 alternative. defm : ImageGather4AltPatterns; defm : ImageGather4AltPatterns; defm : ImageGather4AltPatterns; defm : ImageGather4AltPatterns; defm : ImageGather4AltPatterns; defm : ImageGather4AltPatterns; // Gather4 with comparison alternative. defm : ImageGather4AltPatterns; defm : ImageGather4AltPatterns; defm : ImageGather4AltPatterns; defm : ImageGather4AltPatterns; defm : ImageGather4AltPatterns; defm : ImageGather4AltPatterns; // Gather4 with offsets alternative. defm : ImageGather4AltPatterns; defm : ImageGather4AltPatterns; defm : ImageGather4AltPatterns; defm : ImageGather4AltPatterns; defm : ImageGather4AltPatterns; defm : ImageGather4AltPatterns; // Gather4 with comparison and offsets alternative. defm : ImageGather4AltPatterns; defm : ImageGather4AltPatterns; defm : ImageGather4AltPatterns; defm : ImageGather4AltPatterns; defm : ImageGather4AltPatterns; defm : ImageGather4AltPatterns; defm : ImageSamplePatterns; // Image atomics defm : ImageAtomicPatterns; def : ImageAtomicCmpSwapPattern; def : ImageAtomicCmpSwapPattern; def : ImageAtomicCmpSwapPattern; defm : ImageAtomicPatterns; defm : ImageAtomicPatterns; defm : ImageAtomicPatterns; defm : ImageAtomicPatterns; defm : ImageAtomicPatterns; defm : ImageAtomicPatterns; defm : ImageAtomicPatterns; defm : ImageAtomicPatterns; defm : ImageAtomicPatterns; defm : ImageAtomicPatterns; defm : ImageAtomicPatterns; /* SIsample for simple 1D texture lookup */ def : GCNPat < (SIsample i32:$addr, v8i32:$rsrc, v4i32:$sampler, imm), (IMAGE_SAMPLE_V4_V1 $addr, $rsrc, $sampler, 0xf, 0, 0, 0, 0, 0, 0, 0) >; class SamplePattern : GCNPat < (name vt:$addr, v8i32:$rsrc, v4i32:$sampler, imm), (opcode $addr, $rsrc, $sampler, 0xf, 0, 0, 0, 0, 0, 0, 0) >; class SampleRectPattern : GCNPat < (name vt:$addr, v8i32:$rsrc, v4i32:$sampler, TEX_RECT), (opcode $addr, $rsrc, $sampler, 0xf, 1, 0, 0, 0, 0, 0, 0) >; class SampleArrayPattern : GCNPat < (name vt:$addr, v8i32:$rsrc, v4i32:$sampler, TEX_ARRAY), (opcode $addr, $rsrc, $sampler, 0xf, 0, 0, 0, 0, 0, 0, 1) >; class SampleShadowPattern : GCNPat < (name vt:$addr, v8i32:$rsrc, v4i32:$sampler, TEX_SHADOW), (opcode $addr, $rsrc, $sampler, 0xf, 0, 0, 0, 0, 0, 0, 0) >; class SampleShadowArrayPattern : GCNPat < (name vt:$addr, v8i32:$rsrc, v4i32:$sampler, TEX_SHADOW_ARRAY), (opcode $addr, $rsrc, $sampler, 0xf, 0, 0, 0, 0, 0, 0, 1) >; /* SIsample* for texture lookups consuming more address parameters */ multiclass SamplePatterns { def : SamplePattern ; def : SampleRectPattern ; def : SampleArrayPattern ; def : SampleShadowPattern ; def : SampleShadowArrayPattern ; def : SamplePattern ; def : SampleArrayPattern ; def : SampleShadowPattern ; def : SampleShadowArrayPattern ; def : SamplePattern ; def : SampleArrayPattern ; def : SampleShadowPattern ; def : SampleShadowArrayPattern ; def : SamplePattern ; def : SampleArrayPattern ; def : SampleShadowPattern ; def : SampleShadowArrayPattern ; } defm : SamplePatterns; defm : SamplePatterns; defm : SamplePatterns; defm : SamplePatterns;