CMakeLists.txt 1.21 KB
Newer Older
Chao Liu's avatar
Chao Liu committed
1
2
3
4
cmake_minimum_required(VERSION 3.15)
project(ck_app)
add_compile_options(-std=c++17)

5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
if (DTYPES)
        add_definitions(-DDTYPES)
        if (DTYPES MATCHES "int8")
                add_definitions(-D__int8__)
        endif()
        if (DTYPES MATCHES "fp8")
                add_definitions(-D__fp8__)
        endif()
        if (DTYPES MATCHES "fp16")
                add_definitions(-D__fp16__)
        endif()
        if (DTYPES MATCHES "fp32")
                add_definitions(-D__fp32__)
        endif()
        if (DTYPES MATCHES "fp64")
                add_definitions(-D__fp64__)
        endif()
        if (DTYPES MATCHES "bf16")
                add_definitions(-D__bf16__)
        endif()
        message("DTYPES macro set to ${DTYPES}")
else()
        add_definitions(-D__int8__ -D__fp8__ -D__fp16__ -D__fp32__ -D__fp64__ -D__bf16__)
endif()

Chao Liu's avatar
Chao Liu committed
30
31
32
33
find_package(composable_kernel 1.0.0 COMPONENTS device_operations)
find_package(hip REQUIRED PATHS /opt/rocm)
message(STATUS "Build with HIP ${hip_VERSION}")

Shaojie WANG's avatar
Shaojie WANG committed
34
35
36
# add all example subdir
file(GLOB dir_list LIST_DIRECTORIES true *)
FOREACH(subdir ${dir_list})
37
    IF(IS_DIRECTORY "${subdir}" AND (NOT "${subdir}" MATCHES "build"))
Shaojie WANG's avatar
Shaojie WANG committed
38
39
40
        add_subdirectory(${subdir})
    ENDIF()
ENDFOREACH()