Refactor CUDA library regex patterns to search nvidia libraries.

microsoft · Nov 15, 2024 · d2cbf27 · d2cbf27
1 parent 014833f
commit d2cbf27
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/onnxruntime/__init__.py b/onnxruntime/__init__.py
@@ -103,6 +103,8 @@
             "cudnn",
             "cudart",
             "cufft",
+            # "curand",
+            # "nvJitLink",
         )
         # Construct a regex pattern for each library name with optional parts
         # Pattern explanation: