Add Loaded from: {full_path} to nvjitlink_linux.pyx; pytest -s

Author: rwgk

.github/workflows/test-wheel-linux.yml

@@ -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.
@@ -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
 
@@ -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
@@ -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
 

.github/workflows/test-wheel-windows.yml

@@ -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
 
@@ -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

cuda_bindings/cuda/bindings/_internal/nvjitlink_linux.pyx

@@ -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()