Merge pull request #519 from mlcommons/profiler_oom_fix

Profiler oom fix
mlcommons · Sep 22, 2023 · 92b2d1d · 92b2d1d
2 parents a6d06df + 33a8a9f
commit 92b2d1d
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/algorithmic_efficiency/profiler.py b/algorithmic_efficiency/profiler.py
@@ -15,7 +15,7 @@
 
 
 def _get_monotonic_time() -> float:
-  if torch.cuda.is_available():
+  if torch.cuda.is_available() and torch.cuda.is_initialized():
     torch.cuda.synchronize()
   return time.monotonic()