-

mila-iqia · Oct 9, 2024 · 8d8cf4e · 8d8cf4e
1 parent a2b6694
commit 8d8cf4e
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/benchmarks/lightning/main.py b/benchmarks/lightning/main.py
@@ -4,7 +4,7 @@
 import argparse
 import os
 
-os.environ["PT_HPU_LAZY_MODE"] = str(int(os.getenv("WORLD_SIZE", -1) <= 0))
+os.environ["PT_HPU_LAZY_MODE"] = str(int(int(os.getenv("WORLD_SIZE", -1)) <= 0))
 
 from habana_frameworks.torch import hpu; hpu.init()