Skip to content

GitLab

Unverified Commit dd34ab6e authored by Illia Silin's avatar Illia Silin Committed by GitHub
Browse files

add CK_USE_XDL/WMMA for client examples (#1238)

parent f83e9701
Hide whitespace changes
Inline Side-by-side
Showing with 15 additions and 0 deletions
client_example/CMakeLists.txt
View file @ dd34ab6e
...@@ -48,6 +48,21 @@ else() ...@@ -48,6 +48,21 @@ else()
endif() endif()
endif() endif()
if (GPU_TARGETS)
if (GPU_TARGETS MATCHES "gfx9")
add_definitions(-DCK_USE_XDL)
set(CK_USE_XDL "ON")
endif()
if (GPU_TARGETS MATCHES "gfx11")
add_definitions(-DCK_USE_WMMA)
set(CK_USE_WMMA "ON")
endif()
else()
add_definitions(-DCK_USE_WMMA -DCK_USE_XDL)
set(CK_USE_XDL "ON")
set(CK_USE_WMMA "ON")
endif()
find_package(composable_kernel COMPONENTS device_other_operations device_gemm_operations device_conv_operations device_reduction_operations) find_package(composable_kernel COMPONENTS device_other_operations device_gemm_operations device_conv_operations device_reduction_operations)
if(GPU_TARGETS MATCHES "gfx9") if(GPU_TARGETS MATCHES "gfx9")
find_package(composable_kernel COMPONENTS device_contraction_operations) find_package(composable_kernel COMPONENTS device_contraction_operations)
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment