File tree Expand file tree Collapse file tree 1 file changed +4
-4
lines changed Expand file tree Collapse file tree 1 file changed +4
-4
lines changed Original file line number Diff line number Diff line change @@ -1993,13 +1993,14 @@ static vk_device ggml_vk_get_device(size_t idx) {
1993
1993
amd_shader_core_properties2 = true ;
1994
1994
} else if (strcmp (" VK_EXT_pipeline_robustness" , properties.extensionName ) == 0 ) {
1995
1995
pipeline_robustness = true ;
1996
- } else if (strcmp (" VK_KHR_cooperative_matrix" , properties.extensionName ) == 0 ) {
1996
+ } else if (strcmp (" VK_KHR_cooperative_matrix" , properties.extensionName ) == 0 &&
1997
+ !getenv (" GGML_VK_DISABLE_COOPMAT" )) {
1997
1998
device->coopmat_support = true ;
1998
1999
device->coopmat_m = 0 ;
1999
2000
device->coopmat_n = 0 ;
2000
2001
device->coopmat_k = 0 ;
2001
2002
} else if (strcmp (" VK_NV_cooperative_matrix2" , properties.extensionName ) == 0 &&
2002
- !getenv (" GGML_VULKAN_DISABLE_COOPMAT2 " )) {
2003
+ !getenv (" GGML_VK_DISABLE_COOPMAT2 " )) {
2003
2004
coopmat2_support = true ;
2004
2005
}
2005
2006
}
@@ -2062,8 +2063,7 @@ static vk_device ggml_vk_get_device(size_t idx) {
2062
2063
device->shader_core_count = 0 ;
2063
2064
}
2064
2065
2065
- const char * GGML_VK_DISABLE_F16 = getenv (" GGML_VK_DISABLE_F16" );
2066
- const bool force_disable_f16 = GGML_VK_DISABLE_F16 != nullptr ;
2066
+ const bool force_disable_f16 = getenv (" GGML_VK_DISABLE_F16" ) != nullptr ;
2067
2067
2068
2068
device->fp16 = !force_disable_f16 && fp16_storage && fp16_compute;
2069
2069
You can’t perform that action at this time.
0 commit comments