Skip to content

Commit 544065b

Browse files
leofangclaude
andcommitted
Make cubin compilation arch explicit in fatbin fixture
Co-Authored-By: Claude Opus 4.6 (1M context) <noreply@anthropic.com>
1 parent 0de91e1 commit 544065b

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

cuda_core/tests/test_module.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -117,7 +117,7 @@ def get_saxpy_fatbin(init_cuda):
117117
second_arch = "75" if arch != "75" else "80"
118118

119119
# Compile to cubin for current device arch
120-
prog = Program(SAXPY_KERNEL, code_type="c++")
120+
prog = Program(SAXPY_KERNEL, code_type="c++", options=ProgramOptions(arch=f"sm_{arch}"))
121121
mod = prog.compile(
122122
"cubin",
123123
name_expressions=("saxpy<float>", "saxpy<double>"),

0 commit comments

Comments
 (0)