Avoid CUDA sync for metadata benchmarks

fepegar · fepegar · commit 035defb5b70f · 2026-07-20T23:40:34.000+01:00
Co-authored-by: Copilot &lt;223556219+Copilot@users.noreply.github.com&gt;

Copilot-Session: baf5f11a-f67c-44dc-804e-6be849fa9160
diff --git a/benchmarks/batching.py b/benchmarks/batching.py
@@ -78,7 +78,7 @@ def _benchmark_device(device: torch.device) -> dict[str, float]:
         "metadata-only": _measure(
             lambda: tio.SubjectsBatch.from_subjects(metadata_subjects),
             _ITERATIONS,
-            device,
+            torch.device("cpu"),
         ),
     }
 

Original file line number	Diff line number	Diff line change
`@@ -78,7 +78,7 @@ def _benchmark_device(device: torch.device) -> dict[str, float]:`
`78`	`78`	`"metadata-only": _measure(`
`79`	`79`	`lambda: tio.SubjectsBatch.from_subjects(metadata_subjects),`
`80`	`80`	`_ITERATIONS,`
`81`		`- device,`
	`81`	`+ torch.device("cpu"),`
`82`	`82`	`),`
`83`	`83`	`}`
`84`	`84`