CMakeLists.txt 8.69 KB
Newer Older
Umang Yadav's avatar
Umang Yadav committed
1
# ####################################################################################
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
# The MIT License (MIT)
#
# Copyright (c) 2015-2022 Advanced Micro Devices, Inc. All rights reserved.
#
# Permission is hereby granted, free of charge, to any person obtaining a copy
# of this software and associated documentation files (the "Software"), to deal
# in the Software without restriction, including without limitation the rights
# to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
# copies of the Software, and to permit persons to whom the Software is
# furnished to do so, subject to the following conditions:
#
# The above copyright notice and this permission notice shall be included in
# all copies or substantial portions of the Software.
#
# THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
# IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
# FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT.  IN NO EVENT SHALL THE
# AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
# LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
# OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
# THE SOFTWARE.
Umang Yadav's avatar
Umang Yadav committed
23
# ####################################################################################
Paul's avatar
Paul committed
24
25
26
27
28
29

cmake_policy(SET CMP0057 NEW)

include(CTest)

find_package(Threads REQUIRED)
Paul's avatar
Paul committed
30
31
32
include(ProcessorCount)
ProcessorCount(N)
set(CTEST_PARALLEL_LEVEL ${N} CACHE STRING "CTest parallel level")
Paul Fultz II's avatar
Paul Fultz II committed
33
add_custom_target(check COMMAND ${CMAKE_CTEST_COMMAND} --output-on-failure -j ${CTEST_PARALLEL_LEVEL} -C ${CMAKE_CFG_INTDIR} --timeout 5000)
Paul's avatar
Paul committed
34
35
add_custom_target(tests)

Paul's avatar
Paul committed
36
find_program(MIGRAPHX_GDB gdb)
Paul's avatar
Paul committed
37

Paul's avatar
Paul committed
38
39
if(MIGRAPHX_GDB)
    set(MIGRAPHX_TEST_GDB On CACHE BOOL "")
Paul's avatar
Paul committed
40
else()
Paul's avatar
Paul committed
41
    set(MIGRAPHX_TEST_GDB Off CACHE BOOL "")
Paul's avatar
Paul committed
42
endif()
Paul's avatar
Paul committed
43

Paul's avatar
Paul committed
44
45
46
47
48
49
50
51
set(SKIP_TESTS)

function(add_test_command NAME EXE)
    if(NAME IN_LIST SKIP_TESTS)
        add_test(NAME ${NAME} COMMAND echo skipped)
        set_tests_properties(${NAME} PROPERTIES DISABLED On)
    elseif(WIN32)
        set(WINPATH)
Umang Yadav's avatar
Umang Yadav committed
52

Paul's avatar
Paul committed
53
54
55
        foreach(PATH ${CMAKE_FIND_ROOT_PATH})
            list(APPEND WINPATH ${PATH}/bin)
        endforeach()
Umang Yadav's avatar
Umang Yadav committed
56

Paul's avatar
Paul committed
57
58
59
60
61
        file(GENERATE OUTPUT "${CMAKE_CURRENT_BINARY_DIR}/test_${NAME}.cmd"
            CONTENT "set PATH=${WINPATH};%PATH%
                    %1 ${ARGN}")
        add_test(NAME ${NAME} COMMAND ${WINE_CMD} cmd /c "${CMAKE_CURRENT_BINARY_DIR}/test_${NAME}.cmd" $<TARGET_FILE:${EXE}>)
    else()
Paul's avatar
Paul committed
62
        if(MIGRAPHX_TEST_GDB)
Umang Yadav's avatar
Umang Yadav committed
63
64
65
66
67
68
69
            # add_test(NAME ${NAME} COMMAND ${MIGRAPHX_GDB}
            # --batch
            # --return-child-result
            # -ex "set disable-randomization off"
            # -ex run
            # -ex backtrace
            # --args $<TARGET_FILE:${EXE}> ${ARGN})
Paul's avatar
Paul committed
70
71
            set(TEST_DIR ${CMAKE_CURRENT_BINARY_DIR}/gdb/test_${NAME})
            file(MAKE_DIRECTORY ${TEST_DIR})
Umang Yadav's avatar
Umang Yadav committed
72
73

            if(NOT EXISTS ${TEST_DIR})
74
75
                message(FATAL_ERROR "Failed to create test directory: ${TEST_DIR}")
            endif()
Umang Yadav's avatar
Umang Yadav committed
76

Paul's avatar
Paul committed
77
            file(GENERATE OUTPUT "${TEST_DIR}/run.cmake"
Paul's avatar
Paul committed
78
                CONTENT "
Paul's avatar
Paul committed
79
80
                # Remove previous core dump
                file(REMOVE ${TEST_DIR}/core)
Paul's avatar
Paul committed
81
                execute_process(COMMAND $<TARGET_FILE:${EXE}> ${ARGN} WORKING_DIRECTORY ${TEST_DIR} RESULT_VARIABLE RESULT)
Paul's avatar
Paul committed
82
83
                if(NOT RESULT EQUAL 0)
                    # TODO: check for core files based on pid when setting /proc/sys/kernel/core_uses_pid
Paul's avatar
Paul committed
84
                    if(EXISTS ${TEST_DIR}/core)
85
86
                        set(\$ENV{UBSAN_OPTIONS} print_stacktrace=1)
                        set(\$ENV{ASAN_OPTIONS} print_stacktrace=1)
Paul's avatar
Paul committed
87
                        execute_process(COMMAND ${MIGRAPHX_GDB} $<TARGET_FILE:${EXE}> ${TEST_DIR}/core -batch -ex bt)
Paul's avatar
Paul committed
88
89
90
91
                    endif()
                    message(FATAL_ERROR \"Test failed\")
                endif()
            ")
Paul's avatar
Paul committed
92
            add_test(NAME ${NAME} COMMAND ${CMAKE_COMMAND} -P "${TEST_DIR}/run.cmake")
Paul's avatar
Paul committed
93
94
95
96
        else()
            add_test(NAME ${NAME} COMMAND ${EXE} ${ARGN})
        endif()
    endif()
Umang Yadav's avatar
Umang Yadav committed
97

98
    set_tests_properties(${NAME} PROPERTIES FAIL_REGULAR_EXPRESSION "FAILED")
Paul's avatar
Paul committed
99
100
101
endfunction()

function(add_test_executable TEST_NAME)
Umang Yadav's avatar
Umang Yadav committed
102
    add_executable(${TEST_NAME} EXCLUDE_FROM_ALL ${ARGN})
Paul's avatar
Paul committed
103
    target_link_libraries(${TEST_NAME} ${CMAKE_THREAD_LIBS_INIT})
Umang Yadav's avatar
Umang Yadav committed
104

Paul's avatar
Paul committed
105
    # Cmake does not add flags correctly for gcc
Umang Yadav's avatar
Umang Yadav committed
106
    if(CMAKE_CXX_COMPILER_ID MATCHES "GNU")
Paul's avatar
Paul committed
107
108
        set_target_properties(${TEST_NAME} PROPERTIES COMPILE_FLAGS -pthread LINK_FLAGS -pthread)
    endif()
Umang Yadav's avatar
Umang Yadav committed
109

Paul's avatar
Paul committed
110
    separate_arguments(MIOPEN_TEST_FLAGS_ARGS UNIX_COMMAND ${MIOPEN_TEST_FLAGS})
Umang Yadav's avatar
Umang Yadav committed
111

Paul's avatar
Paul committed
112
113
114
115
116
    if(MIOPEN_TEST_ALL)
        set(TEST_COMMAND ${TEST_NAME} ${MIOPEN_TEST_FLOAT_ARG} --all ${MIOPEN_TEST_FLAGS_ARGS})
    else()
        set(TEST_COMMAND ${TEST_NAME} ${MIOPEN_TEST_FLOAT_ARG} ${MIOPEN_TEST_FLAGS_ARGS})
    endif()
Umang Yadav's avatar
Umang Yadav committed
117

Paul's avatar
Paul committed
118
119
120
    add_test_command(${TEST_NAME} ${TEST_COMMAND})
    add_dependencies(tests ${TEST_NAME})
    add_dependencies(check ${TEST_NAME})
121
    target_link_libraries(${TEST_NAME} migraphx migraphx_ref migraphx_onnx)
Paul's avatar
Paul committed
122
    target_include_directories(${TEST_NAME} PUBLIC include)
Paul's avatar
Paul committed
123
124
endfunction(add_test_executable)

125
file(GLOB TESTS ${CONFIGURE_DEPENDS} *.cpp)
Paul's avatar
Paul committed
126
127
128
129

foreach(TEST ${TESTS})
    get_filename_component(BASE_NAME ${TEST} NAME_WE)
    add_test_executable(test_${BASE_NAME} ${TEST})
Paul's avatar
Paul committed
130
    rocm_clang_tidy_check(test_${BASE_NAME})
Paul's avatar
Paul committed
131
endforeach()
Paul's avatar
Paul committed
132

Paul's avatar
Paul committed
133
if(MIGRAPHX_ENABLE_GPU)
Paul's avatar
Paul committed
134
    # gpu tests
135
    file(GLOB GPU_TESTS ${CONFIGURE_DEPENDS} gpu/*.cpp)
Paul's avatar
Paul committed
136

Paul's avatar
Paul committed
137
    foreach(TEST ${GPU_TESTS})
Paul's avatar
Paul committed
138
        get_filename_component(BASE_NAME ${TEST} NAME_WE)
Paul's avatar
Paul committed
139
        add_test_executable(test_gpu_${BASE_NAME} ${TEST})
Paul's avatar
Paul committed
140
        rocm_clang_tidy_check(test_gpu_${BASE_NAME})
Umang Yadav's avatar
Umang Yadav committed
141
142
        set_tests_properties(test_gpu_${BASE_NAME} PROPERTIES
            COST 10
Paul's avatar
Paul committed
143
144
            RESOURCE_LOCK gpu
        )
Umang Yadav's avatar
Umang Yadav committed
145
        target_link_libraries(test_gpu_${BASE_NAME} migraphx_gpu migraphx_kernels)
Paul's avatar
Paul committed
146
147
    endforeach()
endif()
Scott Thornton's avatar
Scott Thornton committed
148

varunsh's avatar
varunsh committed
149
150
151
152
153
154
155
156
if(MIGRAPHX_ENABLE_FPGA)
    # fpga tests
    file(GLOB FPGA_TESTS ${CONFIGURE_DEPENDS} fpga/*.cpp)

    foreach(TEST ${FPGA_TESTS})
        get_filename_component(BASE_NAME ${TEST} NAME_WE)
        add_test_executable(test_fpga_${BASE_NAME} ${TEST})
        rocm_clang_tidy_check(test_fpga_${BASE_NAME})
Umang Yadav's avatar
Umang Yadav committed
157
158
        set_tests_properties(test_fpga_${BASE_NAME} PROPERTIES
            COST 10
varunsh's avatar
varunsh committed
159
160
161
162
163
164
            RESOURCE_LOCK fpga
        )
        target_link_libraries(test_fpga_${BASE_NAME} migraphx_fpga)
    endforeach()
endif()

Paul's avatar
Paul committed
165
# Onnx test
Paul's avatar
Paul committed
166
set(TEST_ONNX_DIR ${CMAKE_CURRENT_SOURCE_DIR}/onnx)
Umang Yadav's avatar
Umang Yadav committed
167
168
file(GLOB ONNX_TESTS ${TEST_ONNX_DIR}/*.cpp)

169
170
foreach(ONNX_TEST ${ONNX_TESTS})
    get_filename_component(BASE_NAME ${ONNX_TEST} NAME_WE)
171
    set(TEST_NAME test_${BASE_NAME})
172
    add_executable(${TEST_NAME} ${ONNX_TEST})
173
    rocm_clang_tidy_check(${TEST_NAME})
174
    target_link_libraries(${TEST_NAME} migraphx_onnx migraphx_ref)
175
    target_include_directories(${TEST_NAME} PUBLIC include)
Umang Yadav's avatar
Umang Yadav committed
176
    add_test(NAME ${TEST_NAME} COMMAND $<TARGET_FILE:${TEST_NAME}> WORKING_DIRECTORY ${TEST_ONNX_DIR})
177
178
    add_dependencies(tests ${TEST_NAME})
    add_dependencies(check ${TEST_NAME})
179
endforeach()
Paul's avatar
Paul committed
180

181
# tf test
kahmed10's avatar
kahmed10 committed
182
set(TEST_TF_DIR ${CMAKE_CURRENT_SOURCE_DIR}/tf)
183
184
add_executable(test_tf tf/tf_test.cpp)
rocm_clang_tidy_check(test_tf)
185
target_link_libraries(test_tf migraphx_tf migraphx_ref)
186
target_include_directories(test_tf PUBLIC include)
Umang Yadav's avatar
Umang Yadav committed
187
add_test(NAME test_tf COMMAND $<TARGET_FILE:test_tf> WORKING_DIRECTORY ${TEST_TF_DIR})
188
189
190
add_dependencies(tests test_tf)
add_dependencies(check test_tf)

Paul Fultz II's avatar
Paul Fultz II committed
191
add_subdirectory(api)
192
add_subdirectory(verify)
Umang Yadav's avatar
Umang Yadav committed
193

194
if(MIGRAPHX_ENABLE_PYTHON)
Umang Yadav's avatar
Umang Yadav committed
195
    add_subdirectory(py)
196
endif()
Paul's avatar
Paul committed
197

Paul's avatar
Paul committed
198
function(test_header NAME HEADER)
Umang Yadav's avatar
Umang Yadav committed
199
    file(WRITE ${CMAKE_CURRENT_BINARY_DIR}/header-main-include-${NAME}.cpp
Paul's avatar
Paul committed
200
201
        "#include <${HEADER}>\nint main() {}\n"
    )
Umang Yadav's avatar
Umang Yadav committed
202
    file(WRITE ${CMAKE_CURRENT_BINARY_DIR}/header-static-include-${NAME}.cpp
Paul's avatar
Paul committed
203
204
205
        "#include <${HEADER}>\n"
    )
    add_test_executable(${NAME}
Umang Yadav's avatar
Umang Yadav committed
206
        ${CMAKE_CURRENT_BINARY_DIR}/header-main-include-${NAME}.cpp
Paul's avatar
Paul committed
207
208
209
210
        ${CMAKE_CURRENT_BINARY_DIR}/header-static-include-${NAME}.cpp
    )
endfunction()

211
function(test_headers PREFIX)
212
    file(GLOB HEADERS ${CONFIGURE_DEPENDS} ${ARGN})
213
214
215
216
217
218

    foreach(HEADER ${HEADERS})
        file(RELATIVE_PATH HEADER_REL ${CMAKE_SOURCE_DIR} ${HEADER})
        string(MAKE_C_IDENTIFIER ${HEADER_REL} TEST_NAME)
        get_filename_component(BASE_NAME ${HEADER} NAME_WE)
        test_header(header_${TEST_NAME} ${PREFIX}/${BASE_NAME}.hpp)
Umang Yadav's avatar
Umang Yadav committed
219

Paul's avatar
Paul committed
220
        if(MIGRAPHX_ENABLE_GPU)
Paul's avatar
Paul committed
221
            target_link_libraries(header_${TEST_NAME} migraphx_gpu)
222
223
224
        endif()
    endforeach()
endfunction()
Paul's avatar
Paul committed
225

Paul's avatar
Paul committed
226
test_headers(migraphx ${CMAKE_SOURCE_DIR}/src/include/migraphx/*.hpp)
227
test_headers(migraphx/ref ${CMAKE_SOURCE_DIR}/src/targets/ref/include/migraphx/ref/*.hpp)
Umang Yadav's avatar
Umang Yadav committed
228

Paul's avatar
Paul committed
229
if(MIGRAPHX_ENABLE_GPU)
Umang Yadav's avatar
Umang Yadav committed
230
    test_headers(migraphx/gpu ${CMAKE_SOURCE_DIR}/src/targets/gpu/include/migraphx/gpu/*.hpp)
231
endif()