pali112
/
ollama
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

vk_check_flash_attention is not needed (coompat2 coopmapt and scalar implementation exist)

This commit is contained in:
Inforithmics 2025-09-25 06:33:15 +02:00
parent a2f2d41d89
commit a7e2d21f59
2 changed files with 0 additions and 29 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

28
discover/gpu_info_vulkan.c
View File

@ -150,34 +150,6 @@ int vk_device_is_supported(vk_handle_t rh, int i) {
return supported;
}
int vk_check_flash_attention(vk_handle_t rh, int i) {
VkInstance instance = rh.vk;
uint32_t deviceCount = rh.num_devices;
VkPhysicalDevice* devices = malloc(deviceCount * sizeof(VkPhysicalDevice));
if (devices == NULL) {
return 0;
}
VkResult result = (*rh.vkEnumeratePhysicalDevices)(instance, &deviceCount, devices);
if (result != VK_SUCCESS) {
free(devices);
return 0;
}
VkPhysicalDeviceProperties properties = {};
(*rh.vkGetPhysicalDeviceProperties)(devices[i], &properties);
int supports_nv_coopmat2 = is_extension_supported(&rh, devices[i], VK_NV_COOPERATIVE_MATRIX_2_EXTENSION_NAME);
if (!supports_nv_coopmat2) {
free(devices);
return 1;
}
free(devices);
return 0;
}
void vk_check_vram(vk_handle_t rh, int i, mem_info_t *resp) {
VkInstance instance = rh.vk;
uint32_t deviceCount = rh.num_devices;

1
discover/gpu_info_vulkan.h
View File

@ -473,7 +473,6 @@ typedef struct vk_init_resp
void vk_init(char* vk_lib_path, vk_init_resp_t *resp);
void vk_check_vram(vk_handle_t rh, int i, mem_info_t *resp);
int vk_check_flash_attention(vk_handle_t rh, int i);
int vk_device_is_supported(vk_handle_t rh, int i);
void vk_release(vk_handle_t rh);