CMakeLists.txt 5.63 KB
Newer Older
Artur Wojcik's avatar
Artur Wojcik committed
1
2
3
# SPDX-License-Identifier: MIT
# Copyright (c) 2018-2022, Advanced Micro Devices, Inc. All rights reserved.

4
include_directories(BEFORE
Adam Osewski's avatar
Adam Osewski committed
5
    ${PROJECT_SOURCE_DIR}/
6
    ${PROJECT_SOURCE_DIR}/profiler/include
7
8
)

Artur Wojcik's avatar
Artur Wojcik committed
9
include(GTest)
JD's avatar
JD committed
10

11
12
add_custom_target(tests)

Chao Liu's avatar
Chao Liu committed
13
14
function(add_test_executable TEST_NAME)
    message("adding test ${TEST_NAME}")
15
16
    set(result 1)
    if(DEFINED DTYPES)
17
18
        foreach(source IN LISTS ARGN)
            set(test 0)
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
            if((source MATCHES "_fp16" OR source MATCHES "_f16") AND NOT "fp16" IN_LIST DTYPES)
                set(test 1)
            endif()
            if((source MATCHES "_fp32" OR source MATCHES "_f32") AND NOT "fp32" IN_LIST DTYPES)
                set(test 1)
            endif()
            if((source MATCHES "_fp64" OR source MATCHES "_f64") AND NOT "fp64" IN_LIST DTYPES)
                set(test 1)
            endif()
            if((source MATCHES "_fp8" OR source MATCHES "_f8") AND NOT "fp8" IN_LIST DTYPES)
                set(test 1)
            endif()
            if((source MATCHES "_bf8" OR source MATCHES "_bf8") AND NOT "bf8" IN_LIST DTYPES)
                set(test 1)
            endif()
            if((source MATCHES "_bf16" OR source MATCHES "_b16") AND NOT "bf16" IN_LIST DTYPES)
                set(test 1)
            endif()
            if((source MATCHES "_int8" OR source MATCHES "_i8") AND NOT "int8" IN_LIST DTYPES)
                set(test 1)
            endif()
40
41
42
            if(test EQUAL 1)
                message("removing test ${source} ")
                list(REMOVE_ITEM ARGN "${source}")
43
44
            endif()
        endforeach()
45
46
    endif()
    foreach(source IN LISTS ARGN)
47
48
49
50
51
52
53
54
55
        if(NOT DEFINED DL_KERNELS AND source MATCHES "_dl")
            message("removing dl test ${source} ")
            list(REMOVE_ITEM ARGN "${source}")
        endif()
    endforeach()

    #only continue if there are some source files left on the list
    if(ARGN)
        add_executable(${TEST_NAME} ${ARGN})
Artur Wojcik's avatar
Artur Wojcik committed
56
        target_link_libraries(${TEST_NAME} PRIVATE getopt::getopt)
57
58
59
60
61
62
63
        add_test(NAME ${TEST_NAME} COMMAND $<TARGET_FILE:${TEST_NAME}>)
        add_dependencies(tests ${TEST_NAME})
        add_dependencies(check ${TEST_NAME})
        rocm_install(TARGETS ${TEST_NAME} COMPONENT tests)
        set(result 0)
    endif()
    #message("add_test returns ${result}")
64
    set(result ${result} PARENT_SCOPE)
Artur Wojcik's avatar
Artur Wojcik committed
65
endfunction()
66
67
68

function(add_gtest_executable TEST_NAME)
    message("adding gtest ${TEST_NAME}")
69
70
    set(result 1)
    if(DEFINED DTYPES)
71
72
        foreach(source IN LISTS ARGN)
            set(test 0)
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
            if((source MATCHES "_fp16" OR source MATCHES "_f16") AND NOT "fp16" IN_LIST DTYPES)
                set(test 1)
            endif()
            if((source MATCHES "_fp32" OR source MATCHES "_f32") AND NOT "fp32" IN_LIST DTYPES)
                set(test 1)
            endif()
            if((source MATCHES "_fp64" OR source MATCHES "_f64") AND NOT "fp64" IN_LIST DTYPES)
                set(test 1)
            endif()
            if((source MATCHES "_fp8" OR source MATCHES "_f8") AND NOT "fp8" IN_LIST DTYPES)
                set(test 1)
            endif()
            if((source MATCHES "_bf8" OR source MATCHES "_bf8") AND NOT "bf8" IN_LIST DTYPES)
                set(test 1)
            endif()
            if((source MATCHES "_bf16" OR source MATCHES "_b16") AND NOT "bf16" IN_LIST DTYPES)
                set(test 1)
            endif()
            if((source MATCHES "_int8" OR source MATCHES "_i8") AND NOT "int8" IN_LIST DTYPES)
                set(test 1)
            endif()
94
95
96
            if(test EQUAL 1)
                message("removing gtest ${source} ")
                list(REMOVE_ITEM ARGN "${source}")
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
            endif()
        endforeach()
    endif()
    foreach(source IN LISTS ARGN)
        if(NOT DEFINED DL_KERNELS AND source MATCHES "_dl")
            message("removing dl test ${source} ")
            list(REMOVE_ITEM ARGN "${source}")
        endif()
    endforeach()
    #only continue if there are some source files left on the list
    if(ARGN)
        add_executable(${TEST_NAME} ${ARGN})
        add_dependencies(tests ${TEST_NAME})
        add_dependencies(check ${TEST_NAME})

        # suppress gtest warnings
        target_compile_options(${TEST_NAME} PRIVATE -Wno-global-constructors -Wno-undef)
Artur Wojcik's avatar
Artur Wojcik committed
114
        target_link_libraries(${TEST_NAME} PRIVATE gtest_main getopt::getopt)
115
116
117
118
119
        add_test(NAME ${TEST_NAME} COMMAND $<TARGET_FILE:${TEST_NAME}>)
        rocm_install(TARGETS ${TEST_NAME} COMPONENT tests)
        set(result 0)
    endif()
    #message("add_gtest returns ${result}")
120
    set(result ${result} PARENT_SCOPE)
Artur Wojcik's avatar
Artur Wojcik committed
121
endfunction()
122

Chao Liu's avatar
Chao Liu committed
123
124
125
126
127
add_subdirectory(magic_number_division)
add_subdirectory(space_filling_curve)
add_subdirectory(conv_util)
add_subdirectory(reference_conv_fwd)
add_subdirectory(gemm)
128
add_subdirectory(gemm_layernorm)
Chao Liu's avatar
Chao Liu committed
129
add_subdirectory(gemm_split_k)
Chao Liu's avatar
Chao Liu committed
130
131
add_subdirectory(gemm_reduce)
add_subdirectory(batched_gemm)
132
add_subdirectory(batched_gemm_reduce)
Anthony Chang's avatar
Anthony Chang committed
133
add_subdirectory(batched_gemm_gemm)
Anthony Chang's avatar
Anthony Chang committed
134
add_subdirectory(batched_gemm_softmax_gemm)
135
add_subdirectory(batched_gemm_softmax_gemm_permute)
Chao Liu's avatar
Chao Liu committed
136
add_subdirectory(grouped_gemm)
137
add_subdirectory(reduce)
138
add_subdirectory(convnd_fwd)
JD's avatar
JD committed
139
add_subdirectory(convnd_bwd_data)
140
add_subdirectory(grouped_convnd_fwd)
141
add_subdirectory(grouped_convnd_bwd_weight)
142
add_subdirectory(block_to_ctile_map)
143
add_subdirectory(softmax)
144
add_subdirectory(normalization)
Adam Osewski's avatar
Adam Osewski committed
145
add_subdirectory(data_type)
146
add_subdirectory(elementwise_normalization)
147
add_subdirectory(batchnorm)
148
add_subdirectory(contraction)
149
add_subdirectory(pool)
150
add_subdirectory(batched_gemm_multi_d)
151
add_subdirectory(grouped_convnd_bwd_data)
152
add_subdirectory(conv_tensor_rearrange)
153
if(GPU_TARGETS MATCHES "gfx11")
154
155
    add_subdirectory(wmma_op)
endif()