Skip to content

Commit 2c08b63

Browse files
committed
Disable bfloat16 operations on HIP
1 parent 812619a commit 2c08b63

File tree

4 files changed

+16
-11
lines changed

4 files changed

+16
-11
lines changed

.github/workflows/cmake-run-cuda.yml

Lines changed: 5 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -6,6 +6,10 @@ on:
66
cuda-version:
77
required: true
88
type: string
9+
cuda-architectures:
10+
required: false
11+
type: string
12+
default: "80-real"
913

1014
env:
1115
# Customize the CMake build type here (Release, Debug, RelWithDebInfo, etc.)
@@ -33,7 +37,7 @@ jobs:
3337
- name: Configure CMake
3438
# Configure CMake in a 'build' subdirectory. `CMAKE_BUILD_TYPE` is only required if you are using a single-configuration generator such as make.
3539
# See https://cmake.org/cmake/help/latest/variable/CMAKE_BUILD_TYPE.html?highlight=cmake_build_type
36-
run: CUDAARCHS=all cmake -B ${{github.workspace}}/build -DCMAKE_BUILD_TYPE=${{env.BUILD_TYPE}} -DKERNEL_FLOAT_BUILD_TEST=1 -DKERNEL_FLOAT_BUILD_EXAMPLE=1
40+
run: CUDAARCHS=all cmake -B ${{github.workspace}}/build -DCMAKE_BUILD_TYPE=${{env.BUILD_TYPE}} -DCMAKE_CUDA_ARCHITECTURES=${{ inputs.cuda-architectures }} -DKERNEL_FLOAT_BUILD_TEST=1 -DKERNEL_FLOAT_BUILD_EXAMPLE=1
3741

3842
- name: Build
3943
# Build your program with the given configuration

.github/workflows/cmake.yml

Lines changed: 7 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -12,20 +12,21 @@ jobs:
1212
build-cuda:
1313
uses: ./.github/workflows/cmake-run-cuda.yml
1414
with:
15-
cuda-version: "12.8.0"
15+
cuda-version: "12.9.0"
16+
cuda-architectures: 60-real;70-real;75-real;80-real;86-real
1617

1718
build-hip:
1819
needs: build-cuda # Only attempt HIP after CUDA was succesfull
1920
uses: ./.github/workflows/cmake-run-hip.yml
2021
with:
21-
rocm-version: "6.3.0"
22+
rocm-version: "6.4.0"
2223

23-
build-cuda-12-9:
24+
build-cuda-12-8:
2425
if: github.ref_name == 'main'
2526
needs: build-cuda
2627
uses: ./.github/workflows/cmake-run-cuda.yml
2728
with:
28-
cuda-version: "12.9.0"
29+
cuda-version: "12.8.0"
2930

3031
build-cuda-12-6:
3132
if: github.ref_name == 'main'
@@ -48,9 +49,9 @@ jobs:
4849
with:
4950
rocm-version: "6.2.0"
5051

51-
build-hip-6-4:
52+
build-hip-6-3:
5253
if: github.ref_name == 'main'
5354
needs: build-hip
5455
uses: ./.github/workflows/cmake-run-hip.yml
5556
with:
56-
rocm-version: "6.4.0"
57+
rocm-version: "6.3.0"

include/kernel_float/macros.h

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -46,7 +46,7 @@
4646
#endif // KERNEL_FLOAT_BF16_AVAILABLE
4747

4848
#ifndef KERNEL_FLOAT_BF16_OPS_AVAILABLE
49-
#define KERNEL_FLOAT_BF16_OPS_AVAILABLE ((KERNEL_FLOAT_IS_CUDA && __CUDA_ARCH__ >= 800) || KERNEL_FLOAT_IS_HIP)
49+
#define KERNEL_FLOAT_BF16_OPS_AVAILABLE (KERNEL_FLOAT_IS_CUDA && __CUDA_ARCH__ >= 800)
5050
#endif
5151

5252
#ifndef KERNEL_FLOAT_FP8_AVAILABLE

single_include/kernel_float.h

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -16,8 +16,8 @@
1616

1717
//================================================================================
1818
// this file has been auto-generated, do not modify its contents!
19-
// date: 2025-09-02 18:31:16.281730
20-
// git hash: 023bc75e8ec67145cdcb447c5fd9aa7d7f180cc6
19+
// date: 2025-09-09 10:07:16.211257
20+
// git hash: 812619a8a211c35416f42945e01be86eb22efbb3
2121
//================================================================================
2222

2323
#ifndef KERNEL_FLOAT_MACROS_H
@@ -68,7 +68,7 @@
6868
#endif // KERNEL_FLOAT_BF16_AVAILABLE
6969

7070
#ifndef KERNEL_FLOAT_BF16_OPS_AVAILABLE
71-
#define KERNEL_FLOAT_BF16_OPS_AVAILABLE ((KERNEL_FLOAT_IS_CUDA && __CUDA_ARCH__ >= 800) || KERNEL_FLOAT_IS_HIP)
71+
#define KERNEL_FLOAT_BF16_OPS_AVAILABLE (KERNEL_FLOAT_IS_CUDA && __CUDA_ARCH__ >= 800)
7272
#endif
7373

7474
#ifndef KERNEL_FLOAT_FP8_AVAILABLE

0 commit comments

Comments
 (0)