Skip to content

Commit 9e359c4

Browse files
authored
Add support for native FP4 Tensor Core MMA in build script
1 parent 99cc3b9 commit 9e359c4

1 file changed

Lines changed: 3 additions & 0 deletions

File tree

‎.github/workflows/build-wheels-cuda.yaml‎

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -227,6 +227,9 @@ jobs:
227227
# CUDA 13 dropped offline compilation support for pre-Turing targets.
228228
$cudaArchs = "75-real;80-real;86-real;89-real;90-real;90-virtual"
229229
}
230+
if ([version]$nvccVersion -ge [version]"12.8") {
231+
$cudaArchs = "$cudaArchs;120a-real"
232+
}
230233
# Build real cubins for the supported GPUs and keep
231234
# one forward-compatible PTX target instead of embedding PTX for every
232235
# SM. This keeps the wheel under GitHub's 2 GiB release-asset limit.

0 commit comments

Comments
 (0)