Skip to content

Commit 7556892

Browse files
committed
[AOTI] Skip emit_multi_arch_kernel when CUDA version is lower than 12.8
Summary: Fix #168353. aot_inductor.emit_multi_arch_kernel requires a newer CUDA version. [ghstack-poisoned]
1 parent deabb3e commit 7556892

File tree

2 files changed

+4
-4
lines changed

2 files changed

+4
-4
lines changed

test/inductor/test_aot_inductor.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -250,8 +250,8 @@ def forward(self, x):
250250
# failure on CI
251251
@common_utils.parametrize("embed_kernel_binary", [False])
252252
@unittest.skipIf(
253-
torch.version.hip is None and _get_torch_cuda_version() < (12, 6),
254-
"Test is only supported on CUDA 12.6+",
253+
torch.version.hip is None and _get_torch_cuda_version() < (12, 8),
254+
"Test is only supported on CUDA 12.8+",
255255
)
256256
def test_simple_multi_arch(self, embed_kernel_binary):
257257
if self.device != GPU_TYPE:

test/inductor/test_aot_inductor_package.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -315,8 +315,8 @@ def forward(self, x, y):
315315
self.assertTrue(torch.allclose(actual, expected))
316316

317317
@unittest.skipIf(
318-
torch.version.hip is None and _get_torch_cuda_version() < (12, 6),
319-
"Test is only supported on CUDA 12.6+",
318+
torch.version.hip is None and _get_torch_cuda_version() < (12, 8),
319+
"Test is only supported on CUDA 12.8+",
320320
)
321321
@unittest.skipIf(IS_FBCODE, "cmake won't work in fbcode")
322322
@skipIfXpu # doesn't support multi-arch binary

0 commit comments

Comments
 (0)