Adding forgotten iq5_k_r4

This commit is contained in:
Iwan Kawrakow
2025-05-19 18:00:16 +03:00
parent 380ab3f33a
commit 06efa17fa9

View File

@@ -2063,6 +2063,8 @@ bool iqk_set_kernels_iqk_quants(int ne00, int typeA, int typeB, std::array<mul_m
return false; return false;
} }
func16 = nullptr;
switch (typeA) { switch (typeA) {
case GGML_TYPE_IQ2_KS: case GGML_TYPE_IQ2_KS:
set_functions<DequantizerIQ2KS>(kernels); set_functions<DequantizerIQ2KS>(kernels);
@@ -2118,6 +2120,10 @@ bool iqk_set_kernels_iqk_quants(int ne00, int typeA, int typeB, std::array<mul_m
func16 = mul_mat_iq5_ks_r4_q8_k<16>; func16 = mul_mat_iq5_ks_r4_q8_k<16>;
#endif #endif
break; break;
case GGML_TYPE_IQ5_K_R4:
IQK_SET_MUL_MAT_FUNCTIONS(mul_mat_iq5_k_r4_q8_k, kernels);
func16 = mul_mat_iq5_k_r4_q8_k<16>;
break;
default: default:
return false; return false;
} }
@@ -3261,6 +3267,8 @@ bool iqk_set_kernels_iqk_quants(int ne00, int typeA, int typeB, std::array<mul_m
break; break;
case GGML_TYPE_IQ5_KS_R4: case GGML_TYPE_IQ5_KS_R4:
IQK_SET_MUL_MAT_FUNCTIONS(mul_mat_iq5_ks_r4_q8_k, kernels); IQK_SET_MUL_MAT_FUNCTIONS(mul_mat_iq5_ks_r4_q8_k, kernels);
case GGML_TYPE_IQ5_K_R4:
IQK_SET_MUL_MAT_FUNCTIONS(mul_mat_iq5_k_r4_q8_k, kernels);
default: default:
return false; return false;
} }