Skip to content

Commit 2b54279

Browse files
committed
skip tileIR test for <13.1 installation
1 parent 965c53b commit 2b54279

1 file changed

Lines changed: 5 additions & 0 deletions

File tree

cuda_bindings/tests/test_nvfatbin.py

Lines changed: 5 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -68,6 +68,10 @@
6868
)
6969

7070

71+
def get_version() -> tuple[int, int]:
72+
return nvfatbin.version()
73+
74+
7175
@pytest.fixture(params=ARCHITECTURES)
7276
def arch(request):
7377
return request.param
@@ -264,6 +268,7 @@ def test_nvfatbin_add_reloc(OBJECT):
264268
nvfatbin.destroy(handle)
265269

266270

271+
@pytest.skipIf(get_version() < (13, 1), reason="TileIR API is not supported in CUDA < 13.1")
267272
def test_nvfatbin_add_tile_ir(TILEIR):
268273
handle = nvfatbin.create([], 0)
269274
nvfatbin.add_tile_ir(handle, TILEIR, len(TILEIR), "VectorAdd", "")

0 commit comments

Comments
 (0)