Add another rpath for finding libnvvm.so

Author: rwgk

cuda_bindings/setup.py

@@ -352,7 +352,13 @@ def build_extension(self, ext):
             elif ext.name == "cuda.bindings._internal.nvjitlink":
                 ldflag = "-Wl,--disable-new-dtags,-rpath,$ORIGIN/../../../nvidia/nvjitlink/lib"
             elif ext.name == "cuda.bindings._internal.nvvm":
-                ldflag = "-Wl,--disable-new-dtags,-rpath,$ORIGIN/../../../nvidia/cuda_nvcc/nvvm/lib64"
+                # from <loc>/site-packages/cuda/bindings/_internal/
+                #   to <loc>/site-packages/nvidia/cuda_nvcc/nvvm/lib64/
+                rel1 = "$ORIGIN/../../../nvidia/cuda_nvcc/nvvm/lib64"
+                # from <loc>/lib/python3.*/site-packages/cuda/bindings/_internal/
+                #   to <loc>/lib/nvvm/lib64/
+                rel2 = "$ORIGIN/../../../../../../nvvm/lib64"
+                ldflag = f"-Wl,--disable-new-dtags,-rpath,{rel1},-rpath,{rel2}"
             else:
                 ldflag = None