File tree Expand file tree Collapse file tree 1 file changed +6
-0
lines changed
Filter options
Expand file tree Collapse file tree 1 file changed +6
-0
lines changed
Original file line number Diff line number Diff line change @@ -2804,23 +2804,29 @@ static vk_device ggml_vk_get_device(size_t idx) {
2804
2804
pipeline_robustness = true;
2805
2805
} else if (strcmp("VK_EXT_subgroup_size_control", properties.extensionName) == 0) {
2806
2806
device->subgroup_size_control = true;
2807
+ #if defined(GGML_VULKAN_COOPMAT_GLSLC_SUPPORT)
2807
2808
} else if (strcmp("VK_KHR_cooperative_matrix", properties.extensionName) == 0 &&
2808
2809
!getenv("GGML_VK_DISABLE_COOPMAT")) {
2809
2810
device->coopmat_support = true;
2810
2811
device->coopmat_m = 0;
2811
2812
device->coopmat_n = 0;
2812
2813
device->coopmat_k = 0;
2814
+ #endif
2815
+ #if defined(GGML_VULKAN_COOPMAT2_GLSLC_SUPPORT)
2813
2816
} else if (strcmp("VK_NV_cooperative_matrix2", properties.extensionName) == 0 &&
2814
2817
!getenv("GGML_VK_DISABLE_COOPMAT2")) {
2815
2818
coopmat2_support = true;
2819
+ #endif
2816
2820
#if defined(GGML_VULKAN_INTEGER_DOT_GLSLC_SUPPORT)
2817
2821
} else if (strcmp("VK_KHR_shader_integer_dot_product", properties.extensionName) == 0 &&
2818
2822
!getenv("GGML_VK_DISABLE_INTEGER_DOT_PRODUCT")) {
2819
2823
device->integer_dot_product = true;
2820
2824
#endif
2825
+ #if defined(GGML_VULKAN_BFLOAT16_GLSLC_SUPPORT)
2821
2826
} else if (strcmp("VK_KHR_shader_bfloat16", properties.extensionName) == 0 &&
2822
2827
!getenv("GGML_VK_DISABLE_BFLOAT16")) {
2823
2828
bfloat16_support = true;
2829
+ #endif
2824
2830
}
2825
2831
}
2826
2832
You can’t perform that action at this time.
0 commit comments