Skip to content
Closed
Show file tree
Hide file tree
Changes from all commits
Commits
File filter

Filter by extension

Filter by extension


Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
8 changes: 4 additions & 4 deletions .github/workflows/test-wheel-linux.yml
Original file line number Diff line number Diff line change
Expand Up @@ -194,7 +194,7 @@ jobs:

pushd ./cuda_bindings
pip install -r requirements.txt
pytest -rxXs -v tests/
pytest -ra -s -v tests/

# It is a bit convoluted to run the Cython tests against CTK wheels,
# so let's just skip them.
Expand All @@ -205,7 +205,7 @@ jobs:
# TODO: enable this once win-64 runners are up
exit 1
fi
pytest -rxXs -v tests/cython
pytest -ra -s -v tests/cython
fi
popd

Expand All @@ -229,7 +229,7 @@ jobs:

pushd ./cuda_core
pip install -r "tests/requirements-cu${TEST_CUDA_MAJOR}.txt"
pytest -rxXs -v tests/
pytest -ra -s -v tests/

# It is a bit convoluted to run the Cython tests against CTK wheels,
# so let's just skip them. Also, currently our CI always installs the
Expand All @@ -243,7 +243,7 @@ jobs:
# TODO: enable this once win-64 runners are up
exit 1
fi
pytest -rxXs -v tests/cython
pytest -ra -s -v tests/cython
fi
popd

Expand Down
4 changes: 2 additions & 2 deletions .github/workflows/test-wheel-windows.yml
Original file line number Diff line number Diff line change
Expand Up @@ -186,7 +186,7 @@ jobs:

Push-Location ./cuda_bindings
pip install -r requirements.txt
pytest -rxXs -v tests/
pytest -ra -s -v tests/
# skip Cython tests for now (NVIDIA/cuda-python#466)
Pop-Location

Expand All @@ -210,7 +210,7 @@ jobs:

Push-Location ./cuda_core
pip install -r "tests/requirements-cu${TEST_CUDA_MAJOR}.txt"
pytest -rxXs -v tests/
pytest -ra -s -v tests/
Pop-Location

- name: Ensure cuda-python installable
Expand Down
8 changes: 8 additions & 0 deletions cuda_bindings/cuda/bindings/_internal/nvjitlink_linux.pyx
Original file line number Diff line number Diff line change
Expand Up @@ -58,6 +58,14 @@ cdef void* load_library(const int driver_ver) except* with gil:
so_name = "libnvJitLink.so" + (f".{suffix}" if suffix else suffix)
handle = dlopen(so_name.encode(), RTLD_NOW | RTLD_GLOBAL)
if handle != NULL:
# Scan /proc/self/maps to find the loaded library
num_print = 0
with open("/proc/self/maps", "r") as f:
for line in f:
if so_name in line:
full_path = line.strip().split()[-1]
num_print += 1
print(f"\nLOOOK {num_print} Loaded from: {full_path}")
break
else:
err_msg = dlerror()
Expand Down
Loading