Updated pre-generated LibDevice to use Cuda SDK v8.

m4rs-mt · Jan 22, 2024 · b46856d · b46856d
1 parent ac36bf2
commit b46856d
Show file tree

Hide file tree

Showing 3 changed files with 336 additions and 329 deletions.
diff --git a/Src/ILGPU/Backends/PTX/PTXLibDevice.cs b/Src/ILGPU/Backends/PTX/PTXLibDevice.cs
@@ -59,7 +59,7 @@ public static void FindLibDevicePaths(
             nvvmBinDir = Path.Combine(nvvmRoot, nvvmBinName);
             var nvvmSearchPattern =
                 RuntimeInformation.IsOSPlatform(OSPlatform.Windows)
-                ? "nvvm*.dll"
+                ? "nvvm64*.dll"
                 : "libnvvm*.so";
             var nvvmFiles = Directory.EnumerateFiles(nvvmBinDir, nvvmSearchPattern);
             libNvvmPath = nvvmFiles.FirstOrDefault();

diff --git a/Src/ILGPU/Runtime/Cuda/NvvmAPI.cs b/Src/ILGPU/Runtime/Cuda/NvvmAPI.cs
@@ -1,6 +1,6 @@
 // ---------------------------------------------------------------------------------------
 //                                        ILGPU
-//                        Copyright (c) 2021-2023 ILGPU Project
+//                        Copyright (c) 2021-2024 ILGPU Project
 //                                    www.ilgpu.net
 //
 // File: NvvmAPI.cs
@@ -138,7 +138,7 @@ public static NvvmAPI Create(string libNvvmPath, string libDevicePath) =>
         private readonly NvvmCreateProgram nvvmCreateProgram;
         private readonly NvvmDestroyProgram nvvmDestroyProgram;
         private readonly NvvmAddModuleToProgram nvvmAddModuleToProgram;
-        private readonly NvvmLazyAddModuleToProgram nvvmLazyAddModuleToProgram;
+        private readonly NvvmLazyAddModuleToProgram? nvvmLazyAddModuleToProgram;
         private readonly NvvmCompileProgram nvvmCompileProgram;
         private readonly NvvmVerifyProgram nvvmVerifyProgram;
         private readonly NvvmGetCompiledResultSize nvvmGetCompiledResultSize;
@@ -183,11 +183,6 @@ private NvvmAPI(string libNvvmPath, string libDevicePath)
                     NativeLibrary.GetExport(
                         libNvvmModule,
                         "nvvmAddModuleToProgram"));
-            nvvmLazyAddModuleToProgram =
-                Marshal.GetDelegateForFunctionPointer<NvvmLazyAddModuleToProgram>(
-                    NativeLibrary.GetExport(
-                        libNvvmModule,
-                        "nvvmLazyAddModuleToProgram"));
             nvvmCompileProgram =
                 Marshal.GetDelegateForFunctionPointer<NvvmCompileProgram>(
                     NativeLibrary.GetExport(
@@ -218,6 +213,16 @@ private NvvmAPI(string libNvvmPath, string libDevicePath)
                     NativeLibrary.GetExport(
                         libNvvmModule,
                         "nvvmGetProgramLog"));
+
+            if (NativeLibrary.TryGetExport(
+                libNvvmModule,
+                "nvvmLazyAddModuleToProgram",
+                out var nvvmLazyAddModuleToProgramPtr))
+            {
+                nvvmLazyAddModuleToProgram =
+                    Marshal.GetDelegateForFunctionPointer<NvvmLazyAddModuleToProgram>(
+                        nvvmLazyAddModuleToProgramPtr);
+            }
         }
 
         /// <inheritdoc/>
@@ -311,7 +316,9 @@ public NvvmResult LazyAddModuleToProgram(
             IntPtr buffer,
             IntPtr size,
             string? name) =>
-            nvvmLazyAddModuleToProgram(program, buffer, size, name);
+            nvvmLazyAddModuleToProgram != null
+            ? nvvmLazyAddModuleToProgram(program, buffer, size, name)
+            : nvvmAddModuleToProgram(program, buffer, size, name);
 
         /// <summary>
         /// Compiles the program.