chore: remove use_deterministic_algorithms=True since it causes cuda …

…errors (#137)
jbloomAus · May 11, 2024 · 1a3bedb · 1a3bedb
1 parent 2d3e990
commit 1a3bedb
Showing 1 changed file with 0 additions and 1 deletion.
diff --git a/tests/unit/conftest.py b/tests/unit/conftest.py
@@ -17,7 +17,6 @@ def reproducibility():
     seed = 0x1234_5678_9ABC_DEF0
     torch.manual_seed(seed)
     torch.cuda.manual_seed_all(seed)
-    torch.use_deterministic_algorithms(True)
     torch.backends.cudnn.benchmark = False
     # Python native RNG; docs don't give any limitations on seed range
     random.seed(seed)