Add EIATTR_CRS_STACK_SIZE attribute

Author: mikex86

driverapi/include/librecuda_status.h

@@ -6,12 +6,12 @@
 typedef int libreCudaStatus_t;
 
 #define LIBRECUDA_DEBUG(msg) std::cerr << "[LibreCuda Debug]: " << msg << std::endl
-#define STRINGIFY(x) #x
-#define TOSTRING(x) STRINGIFY(x)
+#define __LIBRECUDA_STRINGIFY(x) #x
+#define __LIBRECUDA_TOSTRING(x) __LIBRECUDA_STRINGIFY(x)
 
 #define LIBRECUDA_VALIDATE(condition, err) { \
     if (!(condition)) {                      \
-        LIBRECUDA_DEBUG(__FILE__ ":" TOSTRING(__LINE__) ": " #condition);         \
+        LIBRECUDA_DEBUG(__FILE__ ":" __LIBRECUDA_TOSTRING(__LINE__) ": " #condition);         \
         return err;                          \
     }                                        \
 }

driverapi/internal/librecuda_internal.h

@@ -149,7 +149,7 @@ static inline libreCudaStatus_t rm_alloc(int fd, NvV32 clss,
     LIBRECUDA_SUCCEED();
 }
 
-#define LIBRECUDA_VALIDATE_RM_ALLOC(status) { libreCudaStatus_t status_val = status; if (status_val != LIBRECUDA_SUCCESS) { LIBRECUDA_DEBUG(__FILE__ ":" TOSTRING(__LINE__) ": rm_alloc failed with status " + std::to_string(status_val)); LIBRECUDA_FAIL(LIBRECUDA_ERROR_UNKNOWN); } }
+#define LIBRECUDA_VALIDATE_RM_ALLOC(status) { libreCudaStatus_t status_val = status; if (status_val != LIBRECUDA_SUCCESS) { LIBRECUDA_DEBUG(__FILE__ ":" __LIBRECUDA_TOSTRING(__LINE__) ": rm_alloc failed with status " + std::to_string(status_val)); LIBRECUDA_FAIL(LIBRECUDA_ERROR_UNKNOWN); } }
 
 #define RM_ALLOC(fd, clss, client, parent, object, params, param_size, pObjectNew) LIBRECUDA_VALIDATE_RM_ALLOC(rm_alloc(fd, clss, client, parent, object, params, param_size, pObjectNew));
 
@@ -158,7 +158,7 @@ static inline libreCudaStatus_t rm_ctrl(int fd,
                                         NvHandle client, NvHandle object,
                                         void *params, NvU32 paramSize);
 
-#define LIBRECUDA_VALIDATE_RM_CTRL(status) { libreCudaStatus_t status_val = status; if (status_val != LIBRECUDA_SUCCESS) { LIBRECUDA_DEBUG(__FILE__ ":" TOSTRING(__LINE__) ": rm_ctrl failed with status " + std::to_string(status_val)); LIBRECUDA_FAIL(LIBRECUDA_ERROR_UNKNOWN); } }
+#define LIBRECUDA_VALIDATE_RM_CTRL(status) { libreCudaStatus_t status_val = status; if (status_val != LIBRECUDA_SUCCESS) { LIBRECUDA_DEBUG(__FILE__ ":" __LIBRECUDA_TOSTRING(__LINE__) ": rm_ctrl failed with status " + std::to_string(status_val)); LIBRECUDA_FAIL(LIBRECUDA_ERROR_UNKNOWN); } }
 
 #define RM_CTRL(fd, cmd, client, object, params, paramSize) LIBRECUDA_VALIDATE_RM_CTRL(rm_ctrl(fd, cmd, client, object, params, paramSize))
 

driverapi/src/librecuda.cpp

@@ -821,6 +821,9 @@ struct RelocInfo {
 #define EIATTR_CUDA_API_VERSION 0x3704
 #define EIATTR_CUDA_API_VERSION_ATTR_WORD_LEN 2
 
+#define EIATTR_CRS_STACK_SIZE 0x1e04
+#define EIATTR_CRS_STACK_SIZE_ATTR_WORD_LEN 2
+
 #define EIATTR_PARAM_CBANK 0x0a04
 #define EIATTR_PARAM_CBANK_ATTR_WORD_LEN 3
 
@@ -1129,6 +1132,10 @@ libreCudaStatus_t libreCuModuleLoadData(LibreCUmodule *pModule, const void *imag
                         off += (EIATTR_CUDA_API_VERSION_ATTR_WORD_LEN * sizeof(NvU32));
                         break;
                     }
+                    case EIATTR_CRS_STACK_SIZE: {
+                        off += (EIATTR_CRS_STACK_SIZE_ATTR_WORD_LEN * sizeof(NvU32));
+                        break;
+                    }
                     case EIATTR_PARAM_CBANK: {
                         off += (EIATTR_PARAM_CBANK_ATTR_WORD_LEN * sizeof(NvU32));
                         break;