Skip to content

Commit 298b636

Browse files
committed
Raising a warning instead of error if deterministic operation not available
1 parent 1ceabad commit 298b636

1 file changed

Lines changed: 1 addition & 1 deletion

File tree

src/thunder/utils/utils.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -192,7 +192,7 @@ def set_seed(seed: int) -> None:
192192
torch.cuda.manual_seed(seed)
193193
torch.backends.cudnn.deterministic = True
194194
torch.backends.cudnn.benchmark = False
195-
torch.use_deterministic_algorithms(True)
195+
torch.use_deterministic_algorithms(True, warn_only=True)
196196
os.environ["CUBLAS_WORKSPACE_CONFIG"] = ":4096:8"
197197

198198

0 commit comments

Comments
 (0)