Merge branch 'error-description'

akhildevelops · Jan 24, 2024 · 48ecf66 · 48ecf66
2 parents 10f5adf + 63072a3
commit 48ecf66
Show file tree

Hide file tree

Showing 2 changed files with 15 additions and 2 deletions.
diff --git a/example/increment/build.zig.zon b/example/increment/build.zig.zon
@@ -17,7 +17,7 @@
     .dependencies = .{
         .cudaz = .{
             .url = "https://github.com/akhildevelops/cudaz/tarball/master",
-            .hash = "1220cc1c51c2544239dda0899d17c5866560595e3e5008d31b6d33cdfb80c519a9ef",
+            .hash = "122042f55e5b492702d91ac0c09f0d74c7a1842a19ef2e75c1a833d64df51a0be9eb",
         },
     },
     .paths = .{

diff --git a/src/compile.zig b/src/compile.zig
@@ -74,7 +74,20 @@ pub fn cudaProgram(prg: nvrtc.nvrtcProgram, options: ?Options, allocator: std.me
     for (params_own, cparams) |p, *cp| {
         cp.* = p.ptr;
     }
-    try Error.fromNvrtcErrorCode(nvrtc.nvrtcCompileProgram(prg, std.math.cast(c_int, cparams.len).?, cparams.ptr));
+    Error.fromNvrtcErrorCode(nvrtc.nvrtcCompileProgram(prg, std.math.cast(c_int, cparams.len).?, cparams.ptr)) catch |e| {
+        switch (e) {
+            error.NVRTC_ERROR_COMPILATION => {
+                var log_size: usize = undefined;
+                try Error.fromNvrtcErrorCode(nvrtc.nvrtcGetProgramLogSize(prg, &log_size));
+                const log_msg = try allocator.allocSentinel(u8, log_size, '0');
+                try Error.fromNvrtcErrorCode(nvrtc.nvrtcGetProgramLog(prg, log_msg.ptr));
+                std.debug.print("{s}\n", .{log_msg});
+                allocator.free(log_msg);
+            },
+            else => return e,
+        }
+    };
+    return {};
 }
 
 pub fn getPtx(prg: nvrtc.nvrtcProgram, allocator: std.mem.Allocator) CompileError![:0]const u8 {