Lines Matching refs:MMA_LDST
19746 #define MMA_LDST(n, geom_op_type) \ macro
19753 return MMA_LDST(8, m16n16k16_load_a_f16); in getNVPTXMmaLdstInfo()
19755 return MMA_LDST(8, m16n16k16_load_b_f16); in getNVPTXMmaLdstInfo()
19757 return MMA_LDST(4, m16n16k16_load_c_f16); in getNVPTXMmaLdstInfo()
19759 return MMA_LDST(8, m16n16k16_load_c_f32); in getNVPTXMmaLdstInfo()
19761 return MMA_LDST(8, m32n8k16_load_a_f16); in getNVPTXMmaLdstInfo()
19763 return MMA_LDST(8, m32n8k16_load_b_f16); in getNVPTXMmaLdstInfo()
19765 return MMA_LDST(4, m32n8k16_load_c_f16); in getNVPTXMmaLdstInfo()
19767 return MMA_LDST(8, m32n8k16_load_c_f32); in getNVPTXMmaLdstInfo()
19769 return MMA_LDST(8, m8n32k16_load_a_f16); in getNVPTXMmaLdstInfo()
19771 return MMA_LDST(8, m8n32k16_load_b_f16); in getNVPTXMmaLdstInfo()
19773 return MMA_LDST(4, m8n32k16_load_c_f16); in getNVPTXMmaLdstInfo()
19775 return MMA_LDST(8, m8n32k16_load_c_f32); in getNVPTXMmaLdstInfo()
19779 return MMA_LDST(2, m16n16k16_load_a_s8); in getNVPTXMmaLdstInfo()
19781 return MMA_LDST(2, m16n16k16_load_a_u8); in getNVPTXMmaLdstInfo()
19783 return MMA_LDST(2, m16n16k16_load_b_s8); in getNVPTXMmaLdstInfo()
19785 return MMA_LDST(2, m16n16k16_load_b_u8); in getNVPTXMmaLdstInfo()
19787 return MMA_LDST(8, m16n16k16_load_c_s32); in getNVPTXMmaLdstInfo()
19789 return MMA_LDST(4, m32n8k16_load_a_s8); in getNVPTXMmaLdstInfo()
19791 return MMA_LDST(4, m32n8k16_load_a_u8); in getNVPTXMmaLdstInfo()
19793 return MMA_LDST(1, m32n8k16_load_b_s8); in getNVPTXMmaLdstInfo()
19795 return MMA_LDST(1, m32n8k16_load_b_u8); in getNVPTXMmaLdstInfo()
19797 return MMA_LDST(8, m32n8k16_load_c_s32); in getNVPTXMmaLdstInfo()
19799 return MMA_LDST(1, m8n32k16_load_a_s8); in getNVPTXMmaLdstInfo()
19801 return MMA_LDST(1, m8n32k16_load_a_u8); in getNVPTXMmaLdstInfo()
19803 return MMA_LDST(4, m8n32k16_load_b_s8); in getNVPTXMmaLdstInfo()
19805 return MMA_LDST(4, m8n32k16_load_b_u8); in getNVPTXMmaLdstInfo()
19807 return MMA_LDST(8, m8n32k16_load_c_s32); in getNVPTXMmaLdstInfo()
19820 return MMA_LDST(2, m8n8k32_load_c_s32); in getNVPTXMmaLdstInfo()
19826 return MMA_LDST(2, m8n8k128_load_c_s32); in getNVPTXMmaLdstInfo()
19830 return MMA_LDST(1, m8n8k4_load_a_f64); in getNVPTXMmaLdstInfo()
19832 return MMA_LDST(1, m8n8k4_load_b_f64); in getNVPTXMmaLdstInfo()
19834 return MMA_LDST(2, m8n8k4_load_c_f64); in getNVPTXMmaLdstInfo()
19838 return MMA_LDST(4, m16n16k16_load_a_bf16); in getNVPTXMmaLdstInfo()
19840 return MMA_LDST(4, m16n16k16_load_b_bf16); in getNVPTXMmaLdstInfo()
19842 return MMA_LDST(2, m8n32k16_load_a_bf16); in getNVPTXMmaLdstInfo()
19844 return MMA_LDST(8, m8n32k16_load_b_bf16); in getNVPTXMmaLdstInfo()
19846 return MMA_LDST(8, m32n8k16_load_a_bf16); in getNVPTXMmaLdstInfo()
19848 return MMA_LDST(2, m32n8k16_load_b_bf16); in getNVPTXMmaLdstInfo()
19850 return MMA_LDST(4, m16n16k8_load_a_tf32); in getNVPTXMmaLdstInfo()
19852 return MMA_LDST(4, m16n16k8_load_b_tf32); in getNVPTXMmaLdstInfo()
19854 return MMA_LDST(8, m16n16k8_load_c_f32); in getNVPTXMmaLdstInfo()
19861 return MMA_LDST(4, m16n16k16_store_d_f16); in getNVPTXMmaLdstInfo()
19863 return MMA_LDST(8, m16n16k16_store_d_f32); in getNVPTXMmaLdstInfo()
19865 return MMA_LDST(4, m32n8k16_store_d_f16); in getNVPTXMmaLdstInfo()
19867 return MMA_LDST(8, m32n8k16_store_d_f32); in getNVPTXMmaLdstInfo()
19869 return MMA_LDST(4, m8n32k16_store_d_f16); in getNVPTXMmaLdstInfo()
19871 return MMA_LDST(8, m8n32k16_store_d_f32); in getNVPTXMmaLdstInfo()
19877 return MMA_LDST(8, m16n16k16_store_d_s32); in getNVPTXMmaLdstInfo()
19879 return MMA_LDST(8, m32n8k16_store_d_s32); in getNVPTXMmaLdstInfo()
19881 return MMA_LDST(8, m8n32k16_store_d_s32); in getNVPTXMmaLdstInfo()
19883 return MMA_LDST(2, m8n8k32_store_d_s32); in getNVPTXMmaLdstInfo()
19885 return MMA_LDST(2, m8n8k128_store_d_s32); in getNVPTXMmaLdstInfo()
19889 return MMA_LDST(2, m8n8k4_store_d_f64); in getNVPTXMmaLdstInfo()
19893 return MMA_LDST(8, m16n16k8_store_d_f32); in getNVPTXMmaLdstInfo()
19899 #undef MMA_LDST