Merge pull request #66 from yan-rui/main

add random seed settings for BI100
FlagOpen · May 10, 2023 · eee35cd · eee35cd
2 parents a3b19b8 + c4340c0
commit eee35cd
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/training/benchmarks/driver/helper.py b/training/benchmarks/driver/helper.py
@@ -52,6 +52,12 @@ def set_seed(self, seed: int, vendor: str):
             torch.cuda.manual_seed_all(seed)
             torch.backends.cudnn.benchmark = False
             torch.backends.cudnn.deterministic = True
+        elif lower_vendor == "iluvatar":
+            torch.manual_seed(seed)
+            torch.cuda.manual_seed(seed)
+            torch.cuda.manual_seed_all(seed)
+            torch.backends.cudnn.benchmark = False
+            torch.backends.cudnn.deterministic = True
         elif lower_vendor == "kunlunxin":
             torch.manual_seed(seed)
         else: