Lines Matching refs:MMA_LDST
15713 #define MMA_LDST(n, geom_op_type) \ macro
15720 return MMA_LDST(8, m16n16k16_load_a_f16); in getNVPTXMmaLdstInfo()
15722 return MMA_LDST(8, m16n16k16_load_b_f16); in getNVPTXMmaLdstInfo()
15724 return MMA_LDST(4, m16n16k16_load_c_f16); in getNVPTXMmaLdstInfo()
15726 return MMA_LDST(8, m16n16k16_load_c_f32); in getNVPTXMmaLdstInfo()
15728 return MMA_LDST(8, m32n8k16_load_a_f16); in getNVPTXMmaLdstInfo()
15730 return MMA_LDST(8, m32n8k16_load_b_f16); in getNVPTXMmaLdstInfo()
15732 return MMA_LDST(4, m32n8k16_load_c_f16); in getNVPTXMmaLdstInfo()
15734 return MMA_LDST(8, m32n8k16_load_c_f32); in getNVPTXMmaLdstInfo()
15736 return MMA_LDST(8, m8n32k16_load_a_f16); in getNVPTXMmaLdstInfo()
15738 return MMA_LDST(8, m8n32k16_load_b_f16); in getNVPTXMmaLdstInfo()
15740 return MMA_LDST(4, m8n32k16_load_c_f16); in getNVPTXMmaLdstInfo()
15742 return MMA_LDST(8, m8n32k16_load_c_f32); in getNVPTXMmaLdstInfo()
15746 return MMA_LDST(2, m16n16k16_load_a_s8); in getNVPTXMmaLdstInfo()
15748 return MMA_LDST(2, m16n16k16_load_a_u8); in getNVPTXMmaLdstInfo()
15750 return MMA_LDST(2, m16n16k16_load_b_s8); in getNVPTXMmaLdstInfo()
15752 return MMA_LDST(2, m16n16k16_load_b_u8); in getNVPTXMmaLdstInfo()
15754 return MMA_LDST(8, m16n16k16_load_c_s32); in getNVPTXMmaLdstInfo()
15756 return MMA_LDST(4, m32n8k16_load_a_s8); in getNVPTXMmaLdstInfo()
15758 return MMA_LDST(4, m32n8k16_load_a_u8); in getNVPTXMmaLdstInfo()
15760 return MMA_LDST(1, m32n8k16_load_b_s8); in getNVPTXMmaLdstInfo()
15762 return MMA_LDST(1, m32n8k16_load_b_u8); in getNVPTXMmaLdstInfo()
15764 return MMA_LDST(8, m32n8k16_load_c_s32); in getNVPTXMmaLdstInfo()
15766 return MMA_LDST(1, m8n32k16_load_a_s8); in getNVPTXMmaLdstInfo()
15768 return MMA_LDST(1, m8n32k16_load_a_u8); in getNVPTXMmaLdstInfo()
15770 return MMA_LDST(4, m8n32k16_load_b_s8); in getNVPTXMmaLdstInfo()
15772 return MMA_LDST(4, m8n32k16_load_b_u8); in getNVPTXMmaLdstInfo()
15774 return MMA_LDST(8, m8n32k16_load_c_s32); in getNVPTXMmaLdstInfo()
15787 return MMA_LDST(2, m8n8k32_load_c_s32); in getNVPTXMmaLdstInfo()
15793 return MMA_LDST(2, m8n8k128_load_c_s32); in getNVPTXMmaLdstInfo()
15800 return MMA_LDST(4, m16n16k16_store_d_f16); in getNVPTXMmaLdstInfo()
15802 return MMA_LDST(8, m16n16k16_store_d_f32); in getNVPTXMmaLdstInfo()
15804 return MMA_LDST(4, m32n8k16_store_d_f16); in getNVPTXMmaLdstInfo()
15806 return MMA_LDST(8, m32n8k16_store_d_f32); in getNVPTXMmaLdstInfo()
15808 return MMA_LDST(4, m8n32k16_store_d_f16); in getNVPTXMmaLdstInfo()
15810 return MMA_LDST(8, m8n32k16_store_d_f32); in getNVPTXMmaLdstInfo()
15816 return MMA_LDST(8, m16n16k16_store_d_s32); in getNVPTXMmaLdstInfo()
15818 return MMA_LDST(8, m32n8k16_store_d_s32); in getNVPTXMmaLdstInfo()
15820 return MMA_LDST(8, m8n32k16_store_d_s32); in getNVPTXMmaLdstInfo()
15822 return MMA_LDST(2, m8n8k32_store_d_s32); in getNVPTXMmaLdstInfo()
15824 return MMA_LDST(2, m8n8k128_store_d_s32); in getNVPTXMmaLdstInfo()
15830 #undef MMA_LDST