Skip to content

Commit

Permalink
enable test iff pynvjitlink is installed
Browse files Browse the repository at this point in the history
  • Loading branch information
isVoid committed Dec 18, 2024
1 parent b8c03c4 commit 38238bc
Showing 1 changed file with 8 additions and 0 deletions.
8 changes: 8 additions & 0 deletions numba_cuda/numba/cuda/tests/cudadrv/test_nvjitlink.py
Original file line number Diff line number Diff line change
Expand Up @@ -8,6 +8,13 @@
from numba import config
from numba.tests.support import run_in_subprocess, override_config

try:
import pynvjitlink # noqa: F401
PYNVJITLINK_INSTALLED = True
except ImportError:
PYNVJITLINK_INSTALLED = False


import itertools
import os
import io
Expand Down Expand Up @@ -253,6 +260,7 @@ def kernel():
pass


@unittest.skipIf(not PYNVJITLINK_INSTALLED)
class TestLinkerUsage(CUDATestCase):
"""Test that whether pynvjitlink can be enabled by both environment variable
and modification of config at runtime.
Expand Down

0 comments on commit 38238bc

Please sign in to comment.