custom build directory

IBM · Sep 6, 2024 · fdf9740 · fdf9740
1 parent 8fcaacb
commit fdf9740
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/dolomite_engine/data/megatron/utils/__init__.py b/dolomite_engine/data/megatron/utils/__init__.py
@@ -22,7 +22,7 @@ def compile_helpers() -> None:
 
     log_rank_0(logging.INFO, "compiling helpers.cpp")
 
-    build_directory = build_directory = os.path.join(os.path.dirname(__file__), "build")
+    build_directory = os.path.join(os.path.dirname(__file__), "build")
     os.makedirs(build_directory, exist_ok=True)
 
     if torch.cuda.current_device() == 0: