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
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
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)