From c142e37323fe1d438dbd6fca5d0a6d39f2071646 Mon Sep 17 00:00:00 2001
From: jwzhanggy <jwzhanggy@gmail.com>
Date: Mon, 21 Oct 2024 20:48:38 -0700
Subject: [PATCH] Commit with rpn gnn tuning

---
 tinybig/util/utility.py | 1 +
 1 file changed, 1 insertion(+)

diff --git a/tinybig/util/utility.py b/tinybig/util/utility.py
index 22b46f04..a0775532 100644
--- a/tinybig/util/utility.py
+++ b/tinybig/util/utility.py
@@ -57,6 +57,7 @@ def set_random_seed(random_seed: int = 0):
     torch.backends.cudnn.deterministic = True
     torch.backends.cudnn.benchmark = False
     torch.use_deterministic_algorithms(True)
+    os.environ['CUBLAS_WORKSPACE_CONFIG'] = ':4096:8'
 
 
 def check_file_existence(complete_file_path):