Build examples only on the supported HW

2d05b1bb · Bartlomiej Wroblewski · a2b03d36 · 2d05b1bb · 2d05b1bb
Commit 2d05b1bb authored Nov 16, 2023 by Bartlomiej Wroblewski
Show whitespace changes
Inline Side-by-side

Showing with 5 additions and 3 deletions

example/01_gemm/CMakeLists.txt example/01_gemm/CMakeLists.txt +1 -1

example/04_gemm_add_add_fastgelu/CMakeLists.txt example/04_gemm_add_add_fastgelu/CMakeLists.txt +4 -2

No files found.
--- a/example/01_gemm/CMakeLists.txt
+++ b/example/01_gemm/CMakeLists.txt
@@ -56,7 +56,7 @@ add_example_dependencies(example_gemm_xdl example_gemm_xdl_fp8)
 add_example_executable(example_gemm_xdl_fp8_bf8 gemm_xdl_fp8_bf8.cpp)
 add_example_dependencies(example_gemm_xdl example_gemm_xdl_fp8_bf8)

-if(GPU_TARGETS MATCHES "gfx90a")
+if(GPU_TARGETS MATCHES "gfx90a" OR GPU_TARGETS MATCHES "gfx940" OR GPU_TARGETS MATCHES "gfx941" OR GPU_TARGETS MATCHES "gfx942")
  add_example_executable(example_gemm_xdl_lds_direct_load_fp32 gemm_xdl_lds_direct_load_fp32.cpp)
  add_example_dependencies(example_gemm_xdl example_gemm_xdl_lds_direct_load_fp32)


--- a/example/04_gemm_add_add_fastgelu/CMakeLists.txt
+++ b/example/04_gemm_add_add_fastgelu/CMakeLists.txt
@@ -12,8 +12,10 @@ foreach(gpu IN LISTS GPU_TARGETS)
        add_example_executable(example_gemm_add_add_fastgelu_xdl_fp32 gemm_add_add_fastgelu_xdl_fp32.cpp)
        add_example_dependencies(example_gemm_add_add_fastgelu_xdl example_gemm_add_add_fastgelu_xdl_fp32)

+        if(GPU_TARGETS MATCHES "gfx90a" OR GPU_TARGETS MATCHES "gfx940" OR GPU_TARGETS MATCHES "gfx941" OR GPU_TARGETS MATCHES "gfx942")
            add_example_executable(example_gemm_add_add_fastgelu_xdl_lds_direct_load_fp32 gemm_add_add_fastgelu_xdl_lds_direct_load_fp32.cpp)
            add_example_dependencies(example_gemm_add_add_fastgelu_xdl example_gemm_add_add_fastgelu_xdl_lds_direct_load_fp32)
+        endif()

        if(USE_BITINT_EXTENSION_INT4)
            add_example_executable(example_gemm_add_add_fastgelu_xdl_int4 gemm_add_add_fastgelu_xdl_int4.cpp)