vulkan: Add VK_NV_cooperative_matrix2 support for mul_mat and flash attention (#10206)

This commit is contained in:
Jeff Bolz
2024-12-05 13:15:05 -06:00
committed by GitHub
parent 6fe6247831
commit c9c6e01dae
6 changed files with 1665 additions and 97 deletions

View File

@@ -1,7 +1,9 @@
find_package (Threads REQUIRED)
find_package(Vulkan COMPONENTS glslc REQUIRED)
set(TARGET vulkan-shaders-gen)
add_executable(${TARGET} vulkan-shaders-gen.cpp)
install(TARGETS ${TARGET} RUNTIME)
target_compile_features(${TARGET} PRIVATE cxx_std_17)
target_link_libraries(vulkan-shaders-gen PUBLIC Threads::Threads)
target_link_libraries(vulkan-shaders-gen PRIVATE Vulkan::Vulkan)