Compile time dispatch among multi targets

ROCm · Jul 3, 2024 · 8ad46ed · 8ad46ed
1 parent 508b42a
commit 8ad46ed
Show file tree

Hide file tree

Showing 3 changed files with 7 additions and 1 deletion.
diff --git a/client_example/25_wrapper/CMakeLists.txt b/client_example/25_wrapper/CMakeLists.txt
@@ -2,7 +2,7 @@ add_executable(client_tensor_transform_using_wrapper tensor_transform_using_wrap
 target_link_libraries(client_tensor_transform_using_wrapper PRIVATE composable_kernel::device_other_operations)
 add_executable(client_wrapper_img2col wrapper_img2col.cpp)
 target_link_libraries(client_wrapper_img2col PRIVATE composable_kernel::device_other_operations)
-if((GPU_TARGETS MATCHES "gfx9") AND (NOT GPU_TARGETS MATCHES "gfx1"))
+if(GPU_TARGETS MATCHES "gfx9")
     add_executable(client_wrapper_basic_gemm wrapper_basic_gemm.cpp)
     target_link_libraries(client_wrapper_basic_gemm PRIVATE composable_kernel::device_other_operations)
     add_executable(client_wrapper_optimized_gemm wrapper_optimized_gemm.cpp)

diff --git a/client_example/25_wrapper/wrapper_basic_gemm.cpp b/client_example/25_wrapper/wrapper_basic_gemm.cpp
@@ -1,5 +1,7 @@
 // SPDX-License-Identifier: MIT
 // Copyright (c) 2024, Advanced Micro Devices, Inc. All rights reserved.
+#if(!defined(__HIP_DEVICE_COMPILE__) || defined(__gfx908__) || defined(__gfx90a__) || \
+    defined(__gfx94__))
 
 #include <numeric>
 #include <cstdlib>
@@ -222,3 +224,4 @@ int main(int argc, char* argv[])
         3840, 4096, 4096, tile_shape, thread_layout);
     return 0;
 }
+#endif
diff --git a/client_example/25_wrapper/wrapper_optimized_gemm.cpp b/client_example/25_wrapper/wrapper_optimized_gemm.cpp
@@ -1,5 +1,7 @@
 // SPDX-License-Identifier: MIT
 // Copyright (c) 2024, Advanced Micro Devices, Inc. All rights reserved.
+#if(!defined(__HIP_DEVICE_COMPILE__) || defined(__gfx908__) || defined(__gfx90a__) || \
+    defined(__gfx94__))
 
 #include <numeric>
 #include <cstdlib>
@@ -314,3 +316,4 @@ int main(int argc, char* argv[])
         3840, 4096, 4096, tile_shape, thread_layout);
     return 0;
 }
+#endif