CMakeLists.txt 602 Bytes
Newer Older
1
if(DTYPES MATCHES "fp16" OR NOT DEFINED DTYPES)
2
3
4
5
6
list(APPEND gpu_list1 gfx1100 gfx1101 gfx1102)
list(APPEND gpu_list2 gfx908 gfx90a gfx940 gfx941 gfx942)
set(target 0)
foreach(gpu IN LISTS GPU_TARGETS)
 if(gpu IN_LIST gpu_list1 AND target EQUAL 0)
7
    add_example_executable(example_gemm_bilinear_wmma_fp16 gemm_bilinear_wmma_fp16.cpp)
8
9
10
11
12
13
14
   set(target 1)
 endif()
endforeach()

set(target 0)
foreach(gpu IN LISTS GPU_TARGETS)
 if(gpu IN_LIST gpu_list2 AND target EQUAL 0)
15
    add_example_executable(example_gemm_bilinear_xdl_fp16 gemm_bilinear_xdl_fp16.cpp)
16
17
18
   set(target 1)
 endif()
endforeach()
19
endif()