CMakeLists.txt 11 KB
Newer Older
1
cmake_minimum_required(VERSION 3.14)
JD's avatar
JD committed
2
3

# Check support for CUDA/HIP in Cmake
Chao Liu's avatar
Chao Liu committed
4
project(composable_kernel)
Chao Liu's avatar
Chao Liu committed
5

6
7
list(APPEND CMAKE_MODULE_PATH "${PROJECT_SOURCE_DIR}/cmake")

JD's avatar
JD committed
8
9
enable_testing()

10
set(ROCM_SYMLINK_LIBS OFF)
Anthony Chang's avatar
Anthony Chang committed
11
find_package(ROCM REQUIRED PATHS /opt/rocm)
JD's avatar
JD committed
12
13
14
15
16
17

include(ROCMInstallTargets)
include(ROCMPackageConfigHelpers)
include(ROCMSetupVersion)
include(ROCMInstallSymlinks)
include(ROCMCreatePackage)
Chao Liu's avatar
Chao Liu committed
18
19
include(CheckCXXCompilerFlag)

Alan Turner's avatar
Alan Turner committed
20
21
option(CK_BUILD_JIT_LIB, "Only build the CK JIT Helper Library" OFF)

22
rocm_setup_version(VERSION 0.2.0)
JD's avatar
JD committed
23
24
25
include(TargetFlags)
list(APPEND CMAKE_PREFIX_PATH ${CMAKE_INSTALL_PREFIX} ${CMAKE_INSTALL_PREFIX}/llvm ${CMAKE_INSTALL_PREFIX}/hip /opt/rocm /opt/rocm/llvm /opt/rocm/hip)

26
## C++
Chao Liu's avatar
Chao Liu committed
27
enable_language(CXX)
Chao Liu's avatar
Chao Liu committed
28
set(CMAKE_CXX_STANDARD 17)
Chao Liu's avatar
Chao Liu committed
29
set(CMAKE_CXX_STANDARD_REQUIRED ON)
Chao Liu's avatar
Chao Liu committed
30
set(CMAKE_CXX_EXTENSIONS OFF)
Chao Liu's avatar
Chao Liu committed
31
32
message("CMAKE_CXX_COMPILER_ID: ${CMAKE_CXX_COMPILER_ID}")

Alan Turner's avatar
Alan Turner committed
33
34
if (NOT CK_BUILD_JIT_LIB)
    option(USE_BITINT_EXTENSION_INT4, "Whether to enable clang's BitInt extension to provide int4 data type." OFF)
Chao Liu's avatar
Chao Liu committed
35

Alan Turner's avatar
Alan Turner committed
36
37
38
39
40
    if(USE_BITINT_EXTENSION_INT4)
        add_compile_definitions(CK_EXPERIMENTAL_BIT_INT_EXTENSION_INT4)
        add_compile_options(-Wno-bit-int-extension)
        message("CK compiled with USE_BITINT_EXTENSION_INT4 set to ${USE_BITINT_EXTENSION_INT4}")
    endif()
Chao Liu's avatar
Chao Liu committed
41

Alan Turner's avatar
Alan Turner committed
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
    ## Threads
    set(THREADS_PREFER_PTHREAD_FLAG ON)
    find_package(Threads REQUIRED)
    link_libraries(Threads::Threads)
    
    ## OpenMP
    if(CMAKE_CXX_COMPILER_ID MATCHES "Clang")
        # workaround issue hipcc in rocm3.5 cannot find openmp
        set(OpenMP_CXX "${CMAKE_CXX_COMPILER}")
        set(OpenMP_CXX_FLAGS "-fopenmp=libomp -Wno-unused-command-line-argument")
        set(OpenMP_CXX_LIB_NAMES "libomp" "libgomp" "libiomp5")
        set(OpenMP_libomp_LIBRARY ${OpenMP_CXX_LIB_NAMES})
        set(OpenMP_libgomp_LIBRARY ${OpenMP_CXX_LIB_NAMES})
        set(OpenMP_libiomp5_LIBRARY ${OpenMP_CXX_LIB_NAMES})
    else()
        find_package(OpenMP REQUIRED)
    endif()
Chao Liu's avatar
Chao Liu committed
59

Alan Turner's avatar
Alan Turner committed
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
    message("OpenMP_CXX_LIB_NAMES: ${OpenMP_CXX_LIB_NAMES}")
    message("OpenMP_gomp_LIBRARY: ${OpenMP_gomp_LIBRARY}")
    message("OpenMP_pthread_LIBRARY: ${OpenMP_pthread_LIBRARY}")
    message("OpenMP_CXX_FLAGS: ${OpenMP_CXX_FLAGS}")

    link_libraries(${OpenMP_gomp_LIBRARY})
    link_libraries(${OpenMP_pthread_LIBRARY})

    ## HIP
    find_package(HIP REQUIRED)
    # Override HIP version in config.h, if necessary.
    # The variables set by find_package() can't be overwritten,
    # therefore let's use intermediate variables.
    set(CK_HIP_VERSION_MAJOR "${HIP_VERSION_MAJOR}")
    set(CK_HIP_VERSION_MINOR "${HIP_VERSION_MINOR}")
    set(CK_HIP_VERSION_PATCH "${HIP_VERSION_PATCH}")
    if( DEFINED CK_OVERRIDE_HIP_VERSION_MAJOR )
        set(CK_HIP_VERSION_MAJOR "${CK_OVERRIDE_HIP_VERSION_MAJOR}")
        message(STATUS "CK_HIP_VERSION_MAJOR overriden with ${CK_OVERRIDE_HIP_VERSION_MAJOR}")
    endif()
    if( DEFINED CK_OVERRIDE_HIP_VERSION_MINOR )
        set(CK_HIP_VERSION_MINOR "${CK_OVERRIDE_HIP_VERSION_MINOR}")
        message(STATUS "CK_HIP_VERSION_MINOR overriden with ${CK_OVERRIDE_HIP_VERSION_MINOR}")
    endif()
    if( DEFINED CK_OVERRIDE_HIP_VERSION_PATCH )
        set(CK_HIP_VERSION_PATCH "${CK_OVERRIDE_HIP_VERSION_PATCH}")
        message(STATUS "CK_HIP_VERSION_PATCH overriden with ${CK_OVERRIDE_HIP_VERSION_PATCH}")
    endif()
    message(STATUS "Build with HIP ${HIP_VERSION}")
    link_libraries(hip::device)
    add_compile_definitions(__HIP_PLATFORM_HCC__=1)
JD's avatar
JD committed
91
endif()
92

Chao Liu's avatar
Chao Liu committed
93
94
## tidy
include(EnableCompilerWarnings)
JD's avatar
JD committed
95
set(CK_TIDY_ERRORS ERRORS * -readability-inconsistent-declaration-parameter-name)
Chao Liu's avatar
Chao Liu committed
96
if(CMAKE_CXX_COMPILER MATCHES ".*hcc" OR CMAKE_CXX_COMPILER MATCHES ".*clang\\+\\+")
JD's avatar
JD committed
97
    set(CK_TIDY_CHECKS -modernize-use-override -readability-non-const-parameter)
Chao Liu's avatar
Chao Liu committed
98
# Enable tidy on hip
JD's avatar
JD committed
99
100
elseif(CK_BACKEND STREQUAL "HIP" OR CK_BACKEND STREQUAL "HIPNOGPU")
    set(CK_TIDY_ERRORS ALL)
Chao Liu's avatar
Chao Liu committed
101
102
endif()

JD's avatar
JD committed
103

Chao Liu's avatar
Chao Liu committed
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
include(ClangTidy)
enable_clang_tidy(
    CHECKS
        *
        -abseil-*
        -android-cloexec-fopen
        # Yea we shouldn't be using rand()
        -cert-msc30-c
        -bugprone-exception-escape
        -bugprone-macro-parentheses
        -cert-env33-c
        -cert-msc32-c
        -cert-msc50-cpp
        -cert-msc51-cpp
        -cert-dcl37-c
        -cert-dcl51-cpp
        -clang-analyzer-alpha.core.CastToStruct
        -clang-analyzer-optin.performance.Padding
        -clang-diagnostic-deprecated-declarations
        -clang-diagnostic-extern-c-compat
        -clang-diagnostic-unused-command-line-argument
        -cppcoreguidelines-avoid-c-arrays
        -cppcoreguidelines-avoid-magic-numbers
        -cppcoreguidelines-explicit-virtual-functions
        -cppcoreguidelines-init-variables
        -cppcoreguidelines-macro-usage
        -cppcoreguidelines-non-private-member-variables-in-classes
        -cppcoreguidelines-pro-bounds-array-to-pointer-decay
        -cppcoreguidelines-pro-bounds-constant-array-index
        -cppcoreguidelines-pro-bounds-pointer-arithmetic
        -cppcoreguidelines-pro-type-member-init
        -cppcoreguidelines-pro-type-reinterpret-cast
        -cppcoreguidelines-pro-type-union-access
        -cppcoreguidelines-pro-type-vararg
        -cppcoreguidelines-special-member-functions
        -fuchsia-*
        -google-explicit-constructor
        -google-readability-braces-around-statements
        -google-readability-todo
        -google-runtime-int
        -google-runtime-references
        -hicpp-vararg
        -hicpp-braces-around-statements
        -hicpp-explicit-conversions
        -hicpp-named-parameter
        -hicpp-no-array-decay
        # We really shouldn't use bitwise operators with signed integers, but
        # opencl leaves us no choice
        -hicpp-avoid-c-arrays
        -hicpp-signed-bitwise
        -hicpp-special-member-functions
        -hicpp-uppercase-literal-suffix
        -hicpp-use-auto
        -hicpp-use-equals-default
        -hicpp-use-override
        -llvm-header-guard
        -llvm-include-order
        #-llvmlibc-*
        -llvmlibc-restrict-system-libc-headers
        -llvmlibc-callee-namespace
        -llvmlibc-implementation-in-namespace
        -llvm-else-after-return
        -llvm-qualified-auto
        -misc-misplaced-const
        -misc-non-private-member-variables-in-classes
        -misc-no-recursion
        -modernize-avoid-bind
        -modernize-avoid-c-arrays
        -modernize-pass-by-value
        -modernize-use-auto
        -modernize-use-default-member-init
        -modernize-use-equals-default
        -modernize-use-trailing-return-type
        -modernize-use-transparent-functors
        -performance-unnecessary-value-param
        -readability-braces-around-statements
        -readability-else-after-return
        # we are not ready to use it, but very useful
        -readability-function-cognitive-complexity
        -readability-isolate-declaration
        -readability-magic-numbers
        -readability-named-parameter
        -readability-uppercase-literal-suffix
        -readability-convert-member-functions-to-static
        -readability-qualified-auto
        -readability-redundant-string-init
        # too many narrowing conversions in our code
        -bugprone-narrowing-conversions
        -cppcoreguidelines-narrowing-conversions
        -altera-struct-pack-align
        -cppcoreguidelines-prefer-member-initializer
JD's avatar
JD committed
195
196
        ${CK_TIDY_CHECKS}
        ${CK_TIDY_ERRORS}
Chao Liu's avatar
Chao Liu committed
197
198
199
    HEADER_FILTER
        "\.hpp$"
    EXTRA_ARGS
JD's avatar
JD committed
200
        -DCK_USE_CLANG_TIDY
Chao Liu's avatar
Chao Liu committed
201
202
)

Alan Turner's avatar
Alan Turner committed
203

Chao Liu's avatar
Chao Liu committed
204
205
206
207
208
209
210
211
212
213
214
215
216
include(CppCheck)
enable_cppcheck(
    CHECKS
        warning
        style
        performance
        portability
    SUPPRESS
        ConfigurationNotChecked
        constStatement
        duplicateCondition
        noExplicitConstructor
        passedByValue
Chao Liu's avatar
Chao Liu committed
217
        preprocessorErrorDirective
Chao Liu's avatar
Chao Liu committed
218
219
220
221
222
223
224
        shadowVariable
        unusedFunction
        unusedPrivateFunction
        unusedStructMember
        unmatchedSuppression
    FORCE
    SOURCES
Chao Liu's avatar
Chao Liu committed
225
        library/src
Chao Liu's avatar
Chao Liu committed
226
227
228
    INCLUDE
        ${CMAKE_CURRENT_SOURCE_DIR}/include
        ${CMAKE_CURRENT_BINARY_DIR}/include
Chao Liu's avatar
Chao Liu committed
229
        ${CMAKE_CURRENT_SOURCE_DIR}/library/include
Chao Liu's avatar
Chao Liu committed
230
231
232
233
    DEFINE
        CPPCHECK=1
        __linux__=1
)
Chao Liu's avatar
Chao Liu committed
234

JD's avatar
JD committed
235
236
237
238
set(CMAKE_LIBRARY_OUTPUT_DIRECTORY ${CMAKE_CURRENT_BINARY_DIR}/lib)
set(CMAKE_ARCHIVE_OUTPUT_DIRECTORY ${CMAKE_CURRENT_BINARY_DIR}/lib)
set(CMAKE_RUNTIME_OUTPUT_DIRECTORY ${CMAKE_CURRENT_BINARY_DIR}/bin)

Alan Turner's avatar
Alan Turner committed
239

Chao Liu's avatar
Chao Liu committed
240
241
242
include_directories(BEFORE
    ${PROJECT_SOURCE_DIR}/include
    ${PROJECT_SOURCE_DIR}/library/include
243
    ${HIP_INCLUDE_DIRS}
JD's avatar
JD committed
244
)
Chao Liu's avatar
Chao Liu committed
245

Alan Turner's avatar
Alan Turner committed
246
247
248
249
250
251
252
if (NOT CK_BUILD_JIT_LIB)
    SET(BUILD_DEV ON CACHE BOOL "BUILD_DEV")
    if(BUILD_DEV)
        add_compile_options(-Werror)
        add_compile_options(-Weverything)
    endif()
    message("CMAKE_CXX_FLAGS: ${CMAKE_CXX_FLAGS}")
253

Alan Turner's avatar
Alan Turner committed
254
    add_custom_target(check COMMAND ${CMAKE_CTEST_COMMAND} --output-on-failure -C ${CMAKE_CFG_INTDIR})
JD's avatar
JD committed
255

Alan Turner's avatar
Alan Turner committed
256
257
258
259
260
261
262
263
264
    file(GLOB_RECURSE INSTANCE_FILES "${PROJECT_SOURCE_DIR}/*/device_*_instance.cpp")
    file(GLOB dir_list RELATIVE ${PROJECT_SOURCE_DIR}/library/src/tensor_operation_instance/gpu ${PROJECT_SOURCE_DIR}/library/src/tensor_operation_instance/gpu/*)
    set(CK_DEVICE_INSTANCES)
    FOREACH(subdir_path ${dir_list})
        IF(IS_DIRECTORY "${PROJECT_SOURCE_DIR}/library/src/tensor_operation_instance/gpu/${subdir_path}")
        list(APPEND CK_DEVICE_INSTANCES device_${subdir_path}_instance)
        ENDIF()
    ENDFOREACH()
    add_custom_target(instances DEPENDS utility;${CK_DEVICE_INSTANCES}  SOURCES ${INSTANCE_FILES})
Anthony Chang's avatar
Anthony Chang committed
265

Alan Turner's avatar
Alan Turner committed
266
267
268
269
    rocm_package_setup_component(tests
            LIBRARY_NAME composablekernel
            PACKAGE_NAME tests # Prevent -static suffix on package name
    )
270

Alan Turner's avatar
Alan Turner committed
271
272
273
274
    rocm_package_setup_component(examples
            LIBRARY_NAME composablekernel
            PACKAGE_NAME examples
    )
275

Alan Turner's avatar
Alan Turner committed
276
277
278
279
    rocm_package_setup_component(profiler
            LIBRARY_NAME composablekernel
            PACKAGE_NAME ckProfiler
    )
280

Alan Turner's avatar
Alan Turner committed
281
282
283
284
285
286
    
    add_subdirectory(example)
    add_subdirectory(test)
    add_subdirectory(profiler)
else()
    add_subdirectory(library)
287

Alan Turner's avatar
Alan Turner committed
288
289
290
291
292
    rocm_package_setup_component(jit_library
            LIBRARY_NAME composablekernel
            PACKAGE_NAME jit_library
    )
endif()
JD's avatar
JD committed
293
294
295
296
297
298
299
300
301
302
303

#Create an interface target for the include only files and call it "composablekernels"
include(CMakePackageConfigHelpers)

set(version 1.0.0)
write_basic_package_version_file(
    "${CMAKE_CURRENT_BINARY_DIR}/composable_kernelConfigVersion.cmake"
    VERSION "${version}"
    COMPATIBILITY AnyNewerVersion
)

Anthony Chang's avatar
Anthony Chang committed
304
configure_package_config_file(${CMAKE_CURRENT_SOURCE_DIR}/Config.cmake.in
JD's avatar
JD committed
305
        "${CMAKE_CURRENT_BINARY_DIR}/composable_kernelConfig.cmake"
Anthony Chang's avatar
Anthony Chang committed
306
        INSTALL_DESTINATION ${CMAKE_INSTALL_LIBDIR}/cmake/composable_kernel
JD's avatar
JD committed
307
308
309
        NO_CHECK_REQUIRED_COMPONENTS_MACRO
)

310
rocm_install(FILES
JD's avatar
JD committed
311
312
    "${CMAKE_CURRENT_BINARY_DIR}/composable_kernelConfig.cmake"
    "${CMAKE_CURRENT_BINARY_DIR}/composable_kernelConfigVersion.cmake"
Anthony Chang's avatar
Anthony Chang committed
313
    DESTINATION ${CMAKE_INSTALL_LIBDIR}/cmake/composable_kernel
JD's avatar
JD committed
314
)
315
316
317
318
319
320
321
322
323
324
325

set(CPACK_RESOURCE_FILE_LICENSE "${CMAKE_CURRENT_SOURCE_DIR}/LICENSE")
set(CPACK_RPM_PACKAGE_LICENSE "MIT")

rocm_create_package(
    NAME composablekernel
    DESCRIPTION "High Performance Composable Kernel for AMD GPUs"
    MAINTAINER "MIOpen Kernels Dev Team <dl.MIOpen@amd.com>"
    LDCONFIG
    HEADER_ONLY
)