diff --git a/training/benchmarks/driver/helper.py b/training/benchmarks/driver/helper.py
index 21121cc73..bd36a50bc 100644
--- 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: