try fix oneAPI device again

MolloiLab · Jan 18, 2025 · 462d0d4 · 462d0d4
1 parent a538d64
commit 462d0d4
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/benchmarks/setup.jl b/benchmarks/setup.jl
@@ -21,7 +21,7 @@ function monitor_gpu_memory(backend::String, duration=0.1)
         elseif backend == "oneAPI"
             # Get the first device since that's what we're using
             device = oneAPI.devices()[1]
-            props = oneL0.memory_properties(device)
+            props = oneL0.memory_properties(device)[1]
             return (props.totalSize - props.freeSize) / (1024 * 1024)
         elseif backend == "AMDGPU"
             free, total = AMDGPU.HIP.memory_info()