chore: cleanup

kevmo314 · Dec 15, 2024 · 516909a · 516909a
1 parent 3740492
commit 516909a
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 4 deletions.
diff --git a/client.cpp b/client.cpp
@@ -271,12 +271,10 @@ void* maybe_get_cached_arg_ptr(const int index, void* arg_ptr)
         }
     }
 
+    // no matching host pointer found
     if (found < 0)
         return nullptr;
 
-    // @TODO: this is currently very naive. it depends on the ordering of the args passed to launchKernel...
-    // to be the same order that the memory addresses are allocated.
-    // expand on this soon. - brodey
    return &conns[index].unified_mem_pointers[found][0];
 }
 

diff --git a/test/unified.cu b/test/unified.cu
@@ -25,8 +25,8 @@ int main() {
     // Unified memory allocation
     int *a, *c;
 
-    cudaMallocManaged(&a, bytes);
     cudaMallocManaged(&c, bytes);
+    cudaMallocManaged(&a, bytes);
 
     // Initialize arrays on the CPU
     for (int i = 0; i < size; ++i) {