Skip to content

Commit cca078a

Browse files
Merge branch 'develop' into jograner/conv-bwd-weight-kbatch-in-2gb-limit
2 parents 765fcfc + f211156 commit cca078a

5,257 files changed

Lines changed: 45557 additions & 26218 deletions

File tree

Some content is hidden

Large Commits have some content hidden by default. Use the searchbox below for content that may be hidden.

‎.github/workflows/therock-ci-linux.yml‎

Lines changed: 4 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -53,8 +53,8 @@ jobs:
5353
uses: actions/checkout@11bd71901bbe5b1630ceea73d27597364c9af683 # v4.2.2
5454
with:
5555
repository: "ROCm/TheRock"
56-
ref: c2921b151b8285a1d29942aceb33cfe0fea77ac9 # 10-15-2025 commit
5756
path: "TheRock"
57+
ref: f3f77a3161922df3eee006b888b439d75b2b4668 # 2025-10-29 commit
5858

5959
- name: Setup ccache
6060
run: |
@@ -77,6 +77,8 @@ jobs:
7777
- name: Patch rocm-libraries
7878
run: |
7979
git config --global --add safe.directory '*'
80+
# Remove patches here if they cannot be applied cleanly, and they have not been deleted from TheRock repo
81+
rm -f ./TheRock/patches/amd-mainline/rocm-libraries/0008-Revert-remove-options-no-enumerate-966.patch
8082
git -c user.name="therockbot" -c "user.email=therockbot@amd.com" am --whitespace=nowarn ./TheRock/patches/amd-mainline/rocm-libraries/*.patch
8183
8284
- name: Install python deps
@@ -128,7 +130,7 @@ jobs:
128130
run: |
129131
python3 TheRock/build_tools/github_actions/post_build_upload.py \
130132
--run-id ${{ github.run_id }} \
131-
--amdgpu-family ${{ env.AMDGPU_FAMILIES }} \
133+
--artifact-group ${{ env.AMDGPU_FAMILIES }} \
132134
--build-dir TheRock/build \
133135
--upload
134136

‎.github/workflows/therock-test-component.yml‎

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -51,13 +51,13 @@ jobs:
5151
uses: actions/checkout@08c6903cd8c0fde910a37f88322edcfb5dd907a8 # v5.0.0
5252
with:
5353
repository: "ROCm/TheRock"
54-
ref: c2921b151b8285a1d29942aceb33cfe0fea77ac9 # 10-15-2025 commit
54+
ref: f3f77a3161922df3eee006b888b439d75b2b4668 # 2025-10-29 commit
5555

5656
- name: Run setup test environment workflow
5757
uses: './.github/actions/setup_test_environment'
5858
with:
5959
ARTIFACT_RUN_ID: ${{ env.ARTIFACT_RUN_ID }}
60-
AMDGPU_FAMILIES: ${{ inputs.amdgpu_families }}
60+
ARTIFACT_GROUP: ${{ inputs.amdgpu_families }}
6161
OUTPUT_ARTIFACTS_DIR: ${{ env.OUTPUT_ARTIFACTS_DIR }}
6262
VENV_DIR: ${{ env.VENV_DIR }}
6363
FETCH_ARTIFACT_ARGS: ${{ fromJSON(inputs.component).fetch_artifact_args }}

‎.github/workflows/therock-test-packages.yml‎

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -27,7 +27,7 @@ jobs:
2727
uses: actions/checkout@11bd71901bbe5b1630ceea73d27597364c9af683 # v4.2.2
2828
with:
2929
repository: "ROCm/TheRock"
30-
ref: c2921b151b8285a1d29942aceb33cfe0fea77ac9 # 10-15-2025 commit
30+
ref: f3f77a3161922df3eee006b888b439d75b2b4668 # 2025-10-29 commit
3131

3232
- name: "Configuring CI options"
3333
env:

‎CHANGELOG.md‎

Lines changed: 12 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -2,11 +2,10 @@
22

33
Documentation for Composable Kernel available at [https://rocm.docs.amd.com/projects/composable_kernel/en/latest/](https://rocm.docs.amd.com/projects/composable_kernel/en/latest/).
44

5-
## (Unreleased) Composable Kernel for ROCm
6-
7-
### Added
5+
## Composable Kernel 1.2.0 for ROCm 7.2.0
86

97
### Added
8+
* Added support for bf16 data type to grouped_gemm and grouped_gemm_preshuffle.
109
* Added support for mixed precision fp8 x bf8 universal GEMM and weight preshuffle GEMM
1110
* Added a compute async pipeline in the CK TILE universal GEMM on gfx950
1211
* Added support for B Tensor type pk_int4_t in the CK TILE weight preshuffle GEMM.
@@ -32,6 +31,16 @@ Documentation for Composable Kernel available at [https://rocm.docs.amd.com/proj
3231
* Added an optional template parameter `Arch` (`gfx9_t`, `gfx12_t` etc.) to `make_kernel` to support linking multiple object files that have the same kernel compiled for different architectures.
3332
* FMHA examples and tests can be built for multiple architectures (gfx9, gfx950, gfx12) at the same time.
3433

34+
### Upcoming changes
35+
36+
* Composable Kernel will be adopting C++20 features in an upcoming ROCm release, updating the minimum compiler requirement to C++20. Ensure that your development environment complies with this requirement to facilitate a seamless transition.
37+
38+
## Composable Kernel 1.1.0 for ROCm 7.1.1
39+
40+
### Upcoming changes
41+
42+
* Composable Kernel will be adopting C++20 features in an upcoming ROCm release, updating the minimum compiler requirement to C++20. Ensure that your development environment complies with this requirement to facilitate a seamless transition.
43+
3544
## Composable Kernel 1.1.0 for ROCm 7.1.0
3645

3746
### Added

‎CMakeLists.txt‎

Lines changed: 16 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -1,3 +1,6 @@
1+
# Copyright (c) Advanced Micro Devices, Inc., or its affiliates.
2+
# SPDX-License-Identifier: MIT
3+
14
cmake_minimum_required(VERSION 3.14)
25
if(POLICY CMP0140)
36
# policies CMP0140 not known to CMake until 3.25
@@ -39,6 +42,8 @@ option(ENABLE_CLANG_CPP_CHECKS "Enables clang tidy, cppcheck" ON)
3942
option(MIOPEN_REQ_LIBS_ONLY "Build only the MIOpen required libraries" OFF)
4043
option(CK_EXPERIMENTAL_BUILDER "Enable experimental builder" OFF)
4144
option(BUILD_MHA_LIB "Build the static library for flash attention" OFF)
45+
option(FORCE_DISABLE_XDL "Skip compiling XDL specific instances (even if supported GPUs are included in GPU_TARGETS)" OFF)
46+
option(FORCE_DISABLE_WMMA "Skip compiling WMMA specific instances (even if supported GPUs are included in GPU_TARGETS)" OFF)
4247

4348
if(CK_EXPERIMENTAL_BUILDER)
4449
add_definitions(-DCK_EXPERIMENTAL_BUILDER)
@@ -122,7 +127,7 @@ add_compile_options(
122127
# Recent change in compiler makes this warning ON by default, which led to compile errors.
123128
add_compile_options(-Wno-nrvo)
124129

125-
if(NOT DISABLE_DL_KERNELS)
130+
if(NOT DISABLE_DL_KERNELS AND GPU_TARGETS MATCHES "gfx101|gfx103|gfx10-1|gfx10-3")
126131
add_definitions(-DDL_KERNELS)
127132
set(DL_KERNELS "ON")
128133
set(CK_ENABLE_DL_KERNELS "ON")
@@ -229,12 +234,12 @@ message(STATUS "Building CK for the following targets: ${SUPPORTED_GPU_TARGETS}"
229234
# Cache SUPPORTED_GPU_TARGETS for debug
230235
set(SUPPORTED_GPU_TARGETS "${SUPPORTED_GPU_TARGETS}" CACHE STRING "List of supported GPU targets")
231236

232-
if (SUPPORTED_GPU_TARGETS MATCHES "gfx9|gfx11|gfx12")
237+
if (SUPPORTED_GPU_TARGETS MATCHES "gfx9|gfx11|gfx12" AND NOT FORCE_DISABLE_XDL)
233238
message(STATUS "Enabling XDL instances")
234239
add_definitions(-DCK_USE_XDL)
235240
set(CK_USE_XDL "ON")
236241
endif()
237-
if (SUPPORTED_GPU_TARGETS MATCHES "gfx94" OR SUPPORTED_GPU_TARGETS MATCHES "gfx95")
242+
if ((SUPPORTED_GPU_TARGETS MATCHES "gfx94" OR SUPPORTED_GPU_TARGETS MATCHES "gfx95") AND NOT FORCE_DISABLE_XDL)
238243
message(STATUS "Enabling XDL FP8 gemms on native architectures")
239244
add_definitions(-DCK_USE_GFX94)
240245
set(CK_USE_GFX94 "ON")
@@ -247,7 +252,7 @@ if (SUPPORTED_GPU_TARGETS MATCHES "gfx10")
247252
add_definitions(-DCK_GFX1030_SUPPORT)
248253
endif()
249254

250-
if (SUPPORTED_GPU_TARGETS MATCHES "gfx11" OR SUPPORTED_GPU_TARGETS MATCHES "gfx12")
255+
if ((SUPPORTED_GPU_TARGETS MATCHES "gfx11" OR SUPPORTED_GPU_TARGETS MATCHES "gfx12") AND NOT FORCE_DISABLE_WMMA)
251256
message(STATUS "Enabling WMMA instances")
252257
add_definitions(-DCK_USE_WMMA)
253258
set(CK_USE_WMMA "ON")
@@ -257,7 +262,7 @@ endif()
257262
# define the macro with the current value (0 or 1)
258263
add_definitions(-DCK_TILE_USE_WMMA=${CK_TILE_USE_WMMA})
259264

260-
if (SUPPORTED_GPU_TARGETS MATCHES "gfx12")
265+
if (SUPPORTED_GPU_TARGETS MATCHES "gfx12" AND NOT FORCE_DISABLE_WMMA)
261266
message(STATUS "Enabling WMMA FP8 gemms on native architectures")
262267
add_definitions(-DCK_USE_WMMA_FP8)
263268
set(CK_USE_WMMA_FP8 "ON")
@@ -683,6 +688,12 @@ if(NOT GPU_ARCHS AND USER_GPU_TARGETS AND NOT MIOPEN_REQ_LIBS_ONLY)
683688
PACKAGE_NAME examples
684689
)
685690
add_subdirectory(example)
691+
692+
add_subdirectory(tutorial)
693+
rocm_package_setup_component(tutorials
694+
LIBRARY_NAME composablekernel
695+
PACKAGE_NAME tutorials
696+
)
686697
add_subdirectory(tile_engine)
687698
if(BUILD_TESTING)
688699
add_subdirectory(test)

‎Dockerfile.aiter‎

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -17,4 +17,6 @@ RUN pip install pandas zmq einops ninja && \
1717
useradd -u 1001 -g 1001 -m -s /bin/bash jenkins && \
1818
chown -R jenkins:jenkins /home/jenkins && \
1919
chmod -R a+rwx /home/jenkins && \
20+
chown -R jenkins:jenkins /tmp && \
21+
chmod -R a+rwx /tmp && \
2022
sudo usermod -aG irc jenkins

0 commit comments

Comments
 (0)