Fix handling of EIATTR_EXIT_INSTR_OFFSETS & add libreCuCtxGetCurrent

Author: mikex86

driverapi/include/librecuda.h

@@ -36,6 +36,8 @@ LIBRECUDA_EXPORT libreCudaStatus_t libreCuCtxDestroy(LibreCUcontext ctx);
 
 LIBRECUDA_EXPORT libreCudaStatus_t libreCuCtxSetCurrent(LibreCUcontext ctx);
 
+LIBRECUDA_EXPORT libreCudaStatus_t libreCuCtxGetCurrent(LibreCUcontext *pCtxOut);
+
 LIBRECUDA_EXPORT libreCudaStatus_t libreCuMemAlloc(void **pDevicePointer, size_t bytesize, bool mapToCpu = false);
 
 LIBRECUDA_EXPORT libreCudaStatus_t libreCuMemCpy(void *dst, void *src, size_t byteCount, LibreCUstream stream);

driverapi/src/librecuda.cpp

@@ -1119,8 +1119,7 @@ libreCudaStatus_t libreCuModuleLoadData(LibreCUmodule *pModule, const void *imag
                         break;
                     }
                     case EIATTR_EXIT_INSTR_OFFSETS: {
-                        off += (EIATTR_EXIT_INSTR_OFFSETS_ATTR_WORD_LEN * sizeof(NvU32));
-                        break;
+                        goto parse_attrs_end;
                     }
                     default: {
                         // TODO: This isn't strictly correct,
@@ -1130,6 +1129,7 @@ libreCudaStatus_t libreCuModuleLoadData(LibreCUmodule *pModule, const void *imag
                     }
                 }
             }
+            parse_attrs_end:;
         }
     }
 
@@ -1342,3 +1342,9 @@ libreCudaStatus_t libreCuDeviceGetName(char *pDeviceName, int length, LibreCUdev
 
     LIBRECUDA_SUCCEED();
 }
+
+libreCudaStatus_t libreCuCtxGetCurrent(LibreCUcontext *pCtxOut) {
+    LIBRECUDA_VALIDATE(pCtxOut != nullptr, LIBRECUDA_ERROR_INVALID_VALUE);
+    *pCtxOut = current_ctx;
+    LIBRECUDA_SUCCEED();
+}