fix(hardware): typo

Signed-off-by: YdrMaster <[email protected]>
InfiniTensor · Jan 29, 2024 · 12e264a · 12e264a
1 parent 722e6d9
commit 12e264a
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/02hardware/src/devices/nvidia/device.cc b/src/02hardware/src/devices/nvidia/device.cc
@@ -25,7 +25,7 @@ namespace refactor::hardware {
         CUDA_ASSERT(cudaMemGetInfo(&free, &total));
         auto size = free * 9 / 10;
         cudaDeviceProp prop;
-        CUDA_ASSERT(cudaGetDeviceProperties(&prop, 0));
+        CUDA_ASSERT(cudaGetDeviceProperties(&prop, card));
         size_t alignment = prop.textureAlignment;
         fmt::println("initializing Nvidia GPU {}, memory {} / {}, alloc {}, alignment {}",
                      card, free, total, size, alignment);