More fixes

sasha0552 · Aug 27, 2024 · afd3788 · afd3788
1 parent 6aaf715
commit afd3788
Show file tree

Hide file tree

Showing 3 changed files with 14 additions and 5 deletions.
diff --git a/src/main.c b/src/main.c
@@ -269,14 +269,20 @@ int main(int argc, char *argv[]) {
           continue;
         }
 
+        // Get the current state of the GPU
+        gpuState * state = &gpuStates[id];
+
         // Mark the GPU as managed
-        &gpuStates[id]->managed = true;
+        gpuState->managed = true;
       }
     } else {
       // Iterate through each GPU
       for (unsigned int i = 0; i < deviceCount; i++) {
+        // Get the current state of the GPU
+        gpuState * state = &gpuStates[i];
+
         // Mark the GPU as managed
-        &gpuStates[i]->managed = true;
+        gpuState->managed = true;
       }
     }
 
@@ -285,8 +291,11 @@ int main(int argc, char *argv[]) {
 
     // Iterate through each GPU
     for (unsigned int i = 0; i < deviceCount; i++) {
+      // Get the current state of the GPU
+      gpuState * state = &gpuStates[i];
+
       // If GPU is managed
-      if (&gpuStates[i]->managed) {
+      if (gpuState->managed) {
         // Update flag
         hasManagedGpus = true;
       }

diff --git a/src/utils.c b/src/utils.c
@@ -41,7 +41,7 @@ bool parse_ulong(const char *arg, unsigned long *value) {
   return true;
 }
 
-bool parse_ulong_array(const char *arg, const char *delimiter, const size_t max_count, unsigned long *values, size_t *count) {
+bool parse_ulong_array(const char *arg, const char *delimiter, const size_t max_count, unsigned long **values, size_t *count) {
   // Check if the input or output argument is invalid
   if (arg == NULL || values == NULL || count == NULL) {
     return false;

diff --git a/src/utils.h b/src/utils.h
@@ -48,4 +48,4 @@
 /***** ***** ***** ***** ***** FUNCTIONS ***** ***** ***** ***** *****/
 
 bool parse_ulong(const char *arg, unsigned long *value);
-bool parse_ulong_array(const char *arg, const char *delimiter, const size_t max_count, unsigned long *values, size_t *count);
+bool parse_ulong_array(const char *arg, const char *delimiter, const size_t max_count, unsigned long **values, size_t *count);