Lines Matching refs:MMA_LDST
19022 #define MMA_LDST(n, geom_op_type) \ macro
19029 return MMA_LDST(8, m16n16k16_load_a_f16); in getNVPTXMmaLdstInfo()
19031 return MMA_LDST(8, m16n16k16_load_b_f16); in getNVPTXMmaLdstInfo()
19033 return MMA_LDST(4, m16n16k16_load_c_f16); in getNVPTXMmaLdstInfo()
19035 return MMA_LDST(8, m16n16k16_load_c_f32); in getNVPTXMmaLdstInfo()
19037 return MMA_LDST(8, m32n8k16_load_a_f16); in getNVPTXMmaLdstInfo()
19039 return MMA_LDST(8, m32n8k16_load_b_f16); in getNVPTXMmaLdstInfo()
19041 return MMA_LDST(4, m32n8k16_load_c_f16); in getNVPTXMmaLdstInfo()
19043 return MMA_LDST(8, m32n8k16_load_c_f32); in getNVPTXMmaLdstInfo()
19045 return MMA_LDST(8, m8n32k16_load_a_f16); in getNVPTXMmaLdstInfo()
19047 return MMA_LDST(8, m8n32k16_load_b_f16); in getNVPTXMmaLdstInfo()
19049 return MMA_LDST(4, m8n32k16_load_c_f16); in getNVPTXMmaLdstInfo()
19051 return MMA_LDST(8, m8n32k16_load_c_f32); in getNVPTXMmaLdstInfo()
19055 return MMA_LDST(2, m16n16k16_load_a_s8); in getNVPTXMmaLdstInfo()
19057 return MMA_LDST(2, m16n16k16_load_a_u8); in getNVPTXMmaLdstInfo()
19059 return MMA_LDST(2, m16n16k16_load_b_s8); in getNVPTXMmaLdstInfo()
19061 return MMA_LDST(2, m16n16k16_load_b_u8); in getNVPTXMmaLdstInfo()
19063 return MMA_LDST(8, m16n16k16_load_c_s32); in getNVPTXMmaLdstInfo()
19065 return MMA_LDST(4, m32n8k16_load_a_s8); in getNVPTXMmaLdstInfo()
19067 return MMA_LDST(4, m32n8k16_load_a_u8); in getNVPTXMmaLdstInfo()
19069 return MMA_LDST(1, m32n8k16_load_b_s8); in getNVPTXMmaLdstInfo()
19071 return MMA_LDST(1, m32n8k16_load_b_u8); in getNVPTXMmaLdstInfo()
19073 return MMA_LDST(8, m32n8k16_load_c_s32); in getNVPTXMmaLdstInfo()
19075 return MMA_LDST(1, m8n32k16_load_a_s8); in getNVPTXMmaLdstInfo()
19077 return MMA_LDST(1, m8n32k16_load_a_u8); in getNVPTXMmaLdstInfo()
19079 return MMA_LDST(4, m8n32k16_load_b_s8); in getNVPTXMmaLdstInfo()
19081 return MMA_LDST(4, m8n32k16_load_b_u8); in getNVPTXMmaLdstInfo()
19083 return MMA_LDST(8, m8n32k16_load_c_s32); in getNVPTXMmaLdstInfo()
19096 return MMA_LDST(2, m8n8k32_load_c_s32); in getNVPTXMmaLdstInfo()
19102 return MMA_LDST(2, m8n8k128_load_c_s32); in getNVPTXMmaLdstInfo()
19106 return MMA_LDST(1, m8n8k4_load_a_f64); in getNVPTXMmaLdstInfo()
19108 return MMA_LDST(1, m8n8k4_load_b_f64); in getNVPTXMmaLdstInfo()
19110 return MMA_LDST(2, m8n8k4_load_c_f64); in getNVPTXMmaLdstInfo()
19114 return MMA_LDST(4, m16n16k16_load_a_bf16); in getNVPTXMmaLdstInfo()
19116 return MMA_LDST(4, m16n16k16_load_b_bf16); in getNVPTXMmaLdstInfo()
19118 return MMA_LDST(2, m8n32k16_load_a_bf16); in getNVPTXMmaLdstInfo()
19120 return MMA_LDST(8, m8n32k16_load_b_bf16); in getNVPTXMmaLdstInfo()
19122 return MMA_LDST(8, m32n8k16_load_a_bf16); in getNVPTXMmaLdstInfo()
19124 return MMA_LDST(2, m32n8k16_load_b_bf16); in getNVPTXMmaLdstInfo()
19126 return MMA_LDST(4, m16n16k8_load_a_tf32); in getNVPTXMmaLdstInfo()
19128 return MMA_LDST(4, m16n16k8_load_b_tf32); in getNVPTXMmaLdstInfo()
19130 return MMA_LDST(8, m16n16k8_load_c_f32); in getNVPTXMmaLdstInfo()
19137 return MMA_LDST(4, m16n16k16_store_d_f16); in getNVPTXMmaLdstInfo()
19139 return MMA_LDST(8, m16n16k16_store_d_f32); in getNVPTXMmaLdstInfo()
19141 return MMA_LDST(4, m32n8k16_store_d_f16); in getNVPTXMmaLdstInfo()
19143 return MMA_LDST(8, m32n8k16_store_d_f32); in getNVPTXMmaLdstInfo()
19145 return MMA_LDST(4, m8n32k16_store_d_f16); in getNVPTXMmaLdstInfo()
19147 return MMA_LDST(8, m8n32k16_store_d_f32); in getNVPTXMmaLdstInfo()
19153 return MMA_LDST(8, m16n16k16_store_d_s32); in getNVPTXMmaLdstInfo()
19155 return MMA_LDST(8, m32n8k16_store_d_s32); in getNVPTXMmaLdstInfo()
19157 return MMA_LDST(8, m8n32k16_store_d_s32); in getNVPTXMmaLdstInfo()
19159 return MMA_LDST(2, m8n8k32_store_d_s32); in getNVPTXMmaLdstInfo()
19161 return MMA_LDST(2, m8n8k128_store_d_s32); in getNVPTXMmaLdstInfo()
19165 return MMA_LDST(2, m8n8k4_store_d_f64); in getNVPTXMmaLdstInfo()
19169 return MMA_LDST(8, m16n16k8_store_d_f32); in getNVPTXMmaLdstInfo()
19175 #undef MMA_LDST