Skip to content

Commit e710dd6

Browse files
authored
Merge pull request #47 from NVIDIA/switch-ci-pytorch-26.06
do not skip cuTile test on CC 9.x as CI switched to 13.3
2 parents 7cbfbc1 + b1c8f49 commit e710dd6

1 file changed

Lines changed: 0 additions & 12 deletions

File tree

‎examples/test_examples.py‎

Lines changed: 0 additions & 12 deletions
Original file line numberDiff line numberDiff line change
@@ -2,15 +2,8 @@
22
# SPDX-License-Identifier: Apache-2.0
33

44
import importlib
5-
from typing import Any
65

76
import pytest
8-
import torch
9-
10-
11-
def get_cuda_dev_cc_major(device_id: int) -> Any:
12-
props = torch.cuda.get_device_properties(device_id)
13-
return props.major
147

158

169
def test_00_minimal() -> None:
@@ -74,11 +67,6 @@ def test_11_output_csv() -> None:
7467
output_csv.main()
7568

7669

77-
# skip cuTile test on CC 9.x as Cuda Toolkit 13.2 used for testing does not support cuda-tile on CC 9.x (Hopper)
78-
@pytest.mark.skipif(
79-
get_cuda_dev_cc_major(0) == 9,
80-
reason="cuda-tile not supported on CC 9.x in CUDA Toolkit 13.2",
81-
) # type: ignore[untyped-decorator]
8270
def test_12_cutile() -> None:
8371
cutile = importlib.import_module("examples.12_cutile")
8472
cutile.main()

0 commit comments

Comments
 (0)