Commit 1a92ce5c authored by Jared Casper's avatar Jared Casper
Browse files

Merge branch 'helpers_compilation' into 'main'

call helpers makefile every run so we recompile if the code has changed

See merge request ADLR/megatron-lm!215
parents 996aca68 f0232865
...@@ -80,9 +80,6 @@ def initialize_megatron(extra_args_provider=None, args_defaults={}, ...@@ -80,9 +80,6 @@ def initialize_megatron(extra_args_provider=None, args_defaults={},
_init_autoresume() _init_autoresume()
# Compile dataset C++ code. # Compile dataset C++ code.
try:
from megatron.data import helpers
except:
if torch.distributed.get_rank() == 0: if torch.distributed.get_rank() == 0:
from megatron.data.dataset_utils import compile_helper from megatron.data.dataset_utils import compile_helper
compile_helper() compile_helper()
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment