As discussed offline, the search logic here
| # First check if the DLL has been loaded by 3rd parties |
| try: |
| handle = win32api.GetModuleHandle(dll_name) |
| except: |
| pass |
| else: |
| break |
| |
| # Next, check if DLLs are installed via pip |
| for sp in get_site_packages(): |
| mod_path = os.path.join(sp, "nvidia", "nvJitLink", "bin") |
| ifnot os.path.isdir(mod_path): |
| continue |
| os.add_dll_directory(mod_path) |
| try: |
| handle = win32api.LoadLibraryEx( |
| # Note: LOAD_LIBRARY_SEARCH_DLL_LOAD_DIR needs an abs path... |
| os.path.join(mod_path, dll_name), |
| 0, LOAD_LIBRARY_SEARCH_DEFAULT_DIRS | LOAD_LIBRARY_SEARCH_DLL_LOAD_DIR) |
| except: |
| pass |
| else: |
| break |
| |
| # Finally, try default search |
| try: |
| handle = win32api.LoadLibrary(dll_name) |
| except: |
| pass |
| else: |
| break |
does not look for conda installations , which has the following path:
%CONDA_PREFIX%\Library\nvvm\lib\x64\nvvm.lib
One way to fix it is to patch to the internal template and regenerate the above file.
As discussed offline, the search logic here
cuda-python/cuda_bindings/cuda/bindings/_internal/nvjitlink_windows.pyx
Lines 57 to 87 in 19563d5
does not look for conda installations , which has the following path:
One way to fix it is to patch to the internal template and regenerate the above file.