use f32 scalar FA if f16 is not supported by device
This commit is contained in:
parent
828b7e9bb1
commit
f92d7eddab
|
|
@ -3257,10 +3257,17 @@ static void ggml_vk_load_shaders(vk_device& device) {
|
||||||
} \
|
} \
|
||||||
}
|
}
|
||||||
|
|
||||||
CREATE_FA(GGML_TYPE_F32, f32, FA_SCALAR, )
|
if (device->fp16) {
|
||||||
CREATE_FA(GGML_TYPE_F16, f16, FA_SCALAR, )
|
CREATE_FA(GGML_TYPE_F32, f32, FA_SCALAR, )
|
||||||
CREATE_FA(GGML_TYPE_Q4_0, q4_0, FA_SCALAR, )
|
CREATE_FA(GGML_TYPE_F16, f16, FA_SCALAR, )
|
||||||
CREATE_FA(GGML_TYPE_Q8_0, q8_0, FA_SCALAR, )
|
CREATE_FA(GGML_TYPE_Q4_0, q4_0, FA_SCALAR, )
|
||||||
|
CREATE_FA(GGML_TYPE_Q8_0, q8_0, FA_SCALAR, )
|
||||||
|
} else {
|
||||||
|
CREATE_FA(GGML_TYPE_F32, f32, FA_SCALAR, _fp32)
|
||||||
|
CREATE_FA(GGML_TYPE_F16, f16, FA_SCALAR, _fp32)
|
||||||
|
CREATE_FA(GGML_TYPE_Q4_0, q4_0, FA_SCALAR, _fp32)
|
||||||
|
CREATE_FA(GGML_TYPE_Q8_0, q8_0, FA_SCALAR, _fp32)
|
||||||
|
}
|
||||||
#if defined(VK_KHR_cooperative_matrix) && defined(GGML_VULKAN_COOPMAT_GLSLC_SUPPORT)
|
#if defined(VK_KHR_cooperative_matrix) && defined(GGML_VULKAN_COOPMAT_GLSLC_SUPPORT)
|
||||||
if (device->coopmat1_fa_support) {
|
if (device->coopmat1_fa_support) {
|
||||||
CREATE_FA(GGML_TYPE_F32, f32, FA_COOPMAT1, _cm1)
|
CREATE_FA(GGML_TYPE_F32, f32, FA_COOPMAT1, _cm1)
|
||||||
|
|
|
||||||
Loading…
Reference in New Issue