Handle not found for cuda

Author: RossBrunton

offload/plugins-nextgen/cuda/dynamic_cuda/cuda.h

@@ -106,6 +106,7 @@ typedef enum cudaError_enum {
   CUDA_ERROR_INVALID_VALUE = 1,
   CUDA_ERROR_NO_DEVICE = 100,
   CUDA_ERROR_INVALID_HANDLE = 400,
+  CUDA_ERROR_NOT_FOUND = 500,
   CUDA_ERROR_NOT_READY = 600,
   CUDA_ERROR_TOO_MANY_PEERS = 711,
 } CUresult;

offload/plugins-nextgen/cuda/src/rtl.cpp

@@ -1526,8 +1526,18 @@ static Error Plugin::check(int32_t Code, const char *ErrFmt, ArgsTy... Args) {
   if (Ret != CUDA_SUCCESS)
     REPORT("Unrecognized " GETNAME(TARGET_NAME) " error code %d\n", Code);
 
+  // TODO: Add more entries to this switch
+  ErrorCode OffloadErrCode;
+  switch (ResultCode) {
+  case CUDA_ERROR_NOT_FOUND:
+    OffloadErrCode = ErrorCode::NOT_FOUND;
+    break;
+  default:
+    OffloadErrCode = ErrorCode::UNKNOWN;
+  }
+
   // TODO: Create a map for CUDA error codes to Offload error codes
-  return Plugin::error(ErrorCode::UNKNOWN, ErrFmt, Args..., Desc);
+  return Plugin::error(OffloadErrCode, ErrFmt, Args..., Desc);
 }
 
 } // namespace plugin