From 1a3bedbc3192ca919fc0716ce52d06f060ad2550 Mon Sep 17 00:00:00 2001 From: David Chanin Date: Sat, 11 May 2024 12:20:40 +0100 Subject: [PATCH] chore: remove use_deterministic_algorithms=True since it causes cuda errors (#137) --- tests/unit/conftest.py | 1 - 1 file changed, 1 deletion(-) diff --git a/tests/unit/conftest.py b/tests/unit/conftest.py index e88bc41f..53143386 100644 --- 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)