mirror of
https://github.com/ROCm/composable_kernel.git
synced 2026-05-26 08:00:13 +00:00
7ce0127e8f
* WIP POC of dispatcher
* Dispatcher python workflow setup.
* Dispatcher cleanup and updates.
Further dispatcher cleanup and updates.
Build fixes
Improvements and python to CK example
Improvements to readme
* Fixes to python paths
* Cleaning up code
* Improving dispatcher support for different arch
Fixing typos
* Fix formatting errors
* Cleaning up examples
* Improving codegeneration
* Improving and fixing C++ examples
* Adding conv functionality (fwd,bwd,bwdw) and examples.
* Fixes based on feedback.
* Further fixes based on feedback.
* Adding stress test for autogeneration and autocorrection, and fixing preshuffle bug.
* Another round of improvements based on feedback.
* Trimming out unnecessary code.
* Fixing the multi-D implementation.
* Using gpu verification for gemms and fixing convolutions tflops calculation.
* Fix counter usage issue and arch filtering per ops.
* Adding changelog and other fixes.
* Improve examples and resolve critical bugs.
* Reduce build time for python examples.
* Fixing minor bug.
* Fix compilation error.
* Improve installation instructions for dispatcher.
* Add docker based installation instructions for dispatcher.
* Fixing arch-based filtering to match tile engine.
* Remove dead code and fix arch filtering.
* Minor bugfix.
* Updates after rebase.
* Trimming code.
* Fix copyright headers.
* Consolidate examples, cut down code.
* Minor fixes.
* Improving python examples.
* Update readmes.
* Remove conv functionality.
* Cleanup following conv removable.
[ROCm/composable_kernel commit: 9e049a32a1]
118 lines
3.0 KiB
CMake
118 lines
3.0 KiB
CMake
# Copyright (c) Advanced Micro Devices, Inc., or its affiliates.
|
|
# SPDX-License-Identifier: MIT
|
|
|
|
cmake_minimum_required(VERSION 3.16)
|
|
|
|
project(ck_tile_dispatcher VERSION 1.0.0 LANGUAGES CXX)
|
|
|
|
# C++17 required
|
|
set(CMAKE_CXX_STANDARD 17)
|
|
set(CMAKE_CXX_STANDARD_REQUIRED ON)
|
|
set(CMAKE_CXX_EXTENSIONS OFF)
|
|
|
|
# Find HIP for headers (needed for validation kernels)
|
|
find_package(hip QUIET)
|
|
if(NOT hip_FOUND)
|
|
list(APPEND CMAKE_PREFIX_PATH /opt/rocm /opt/rocm/hip)
|
|
find_package(hip REQUIRED)
|
|
endif()
|
|
|
|
# Dispatcher library
|
|
add_library(ck_tile_dispatcher
|
|
src/registry.cpp
|
|
src/dispatcher.cpp
|
|
)
|
|
|
|
# Enable PIC for Python bindings
|
|
set_target_properties(ck_tile_dispatcher PROPERTIES
|
|
POSITION_INDEPENDENT_CODE ON
|
|
)
|
|
|
|
target_include_directories(ck_tile_dispatcher
|
|
PUBLIC
|
|
$<BUILD_INTERFACE:${CMAKE_CURRENT_SOURCE_DIR}/include>
|
|
$<INSTALL_INTERFACE:include>
|
|
)
|
|
|
|
# Link against CK Tile headers (header-only)
|
|
target_include_directories(ck_tile_dispatcher
|
|
PUBLIC
|
|
$<BUILD_INTERFACE:${CMAKE_CURRENT_SOURCE_DIR}/../include>
|
|
$<INSTALL_INTERFACE:include>
|
|
)
|
|
|
|
# Link against HIP headers if available
|
|
if(hip_FOUND)
|
|
target_link_libraries(ck_tile_dispatcher PUBLIC hip::host)
|
|
endif()
|
|
|
|
# Compiler warnings
|
|
if(CMAKE_CXX_COMPILER_ID MATCHES "GNU|Clang")
|
|
target_compile_options(ck_tile_dispatcher PRIVATE
|
|
-Wall -Wextra -Wpedantic
|
|
)
|
|
elseif(CMAKE_CXX_COMPILER_ID MATCHES "MSVC")
|
|
target_compile_options(ck_tile_dispatcher PRIVATE
|
|
/W4
|
|
)
|
|
endif()
|
|
|
|
# Optional: Build tests
|
|
option(BUILD_DISPATCHER_TESTS "Build dispatcher unit tests" OFF)
|
|
if(BUILD_DISPATCHER_TESTS)
|
|
enable_testing()
|
|
add_subdirectory(tests)
|
|
endif()
|
|
|
|
# Optional: Build Python bindings
|
|
option(BUILD_DISPATCHER_PYTHON "Build Python bindings for dispatcher" OFF)
|
|
if(BUILD_DISPATCHER_PYTHON)
|
|
add_subdirectory(python)
|
|
endif()
|
|
|
|
# Optional: Codegen for tile_engine integration
|
|
option(DISPATCHER_AUTO_GENERATE_WRAPPERS "Auto-generate wrappers from tile_engine" OFF)
|
|
if(DISPATCHER_AUTO_GENERATE_WRAPPERS)
|
|
add_subdirectory(codegen)
|
|
endif()
|
|
|
|
# Optional: Build examples
|
|
option(BUILD_DISPATCHER_EXAMPLES "Build dispatcher examples" OFF)
|
|
if(BUILD_DISPATCHER_EXAMPLES)
|
|
add_subdirectory(examples)
|
|
endif()
|
|
|
|
# Optional: Build ctypes bindings
|
|
option(BUILD_DISPATCHER_BINDINGS "Build language bindings for dispatcher" OFF)
|
|
if(BUILD_DISPATCHER_BINDINGS)
|
|
add_subdirectory(bindings/ctypes)
|
|
endif()
|
|
|
|
# If codegen is enabled, add generated include directory
|
|
if(DISPATCHER_AUTO_GENERATE_WRAPPERS AND DISPATCHER_GENERATED_INCLUDE_DIR)
|
|
target_include_directories(ck_tile_dispatcher
|
|
PUBLIC
|
|
$<BUILD_INTERFACE:${DISPATCHER_GENERATED_INCLUDE_DIR}>
|
|
)
|
|
endif()
|
|
|
|
# Installation
|
|
install(TARGETS ck_tile_dispatcher
|
|
EXPORT ck_tile_dispatcher_targets
|
|
LIBRARY DESTINATION lib
|
|
ARCHIVE DESTINATION lib
|
|
RUNTIME DESTINATION bin
|
|
)
|
|
|
|
install(DIRECTORY include/
|
|
DESTINATION include
|
|
FILES_MATCHING PATTERN "*.hpp"
|
|
)
|
|
|
|
install(EXPORT ck_tile_dispatcher_targets
|
|
FILE ck_tile_dispatcher_targets.cmake
|
|
NAMESPACE ck_tile::
|
|
DESTINATION lib/cmake/ck_tile_dispatcher
|
|
)
|
|
|