debug

mlcommons · Sep 25, 2023 · 59c08ab · 59c08ab
1 parent 7cf7f86
commit 59c08ab
Showing 1 changed file with 2 additions and 3 deletions.
diff --git a/submission_runner.py b/submission_runner.py
@@ -145,14 +145,13 @@
 
 
 def _get_time():
-  if torch.cuda.is_available() and torch.cuda.is_initialized():
+  if torch.cuda.is_available():
     torch.cuda.synchronize()
   return time.time()
 
 
 def _get_time_ddp():
-  if torch.cuda.is_initialized():
-    torch.cuda.synchronize()
+  torch.cuda.synchronize()
   t = time.time()
   return sync_ddp_time(t, DEVICE)