diff --git a/examples/nlp/language_modeling/megatron_gpt_pretraining.py b/examples/nlp/language_modeling/megatron_gpt_pretraining.py index 44834b35a0cf..80158446d95a 100644 --- a/examples/nlp/language_modeling/megatron_gpt_pretraining.py +++ b/examples/nlp/language_modeling/megatron_gpt_pretraining.py @@ -13,6 +13,8 @@ # limitations under the License. +# To suppress BF16 compile related issue in the CI runs with turing/V100 +import torch._dynamo import torch.multiprocessing as mp from omegaconf.omegaconf import OmegaConf, open_dict @@ -22,6 +24,8 @@ from nemo.utils import logging from nemo.utils.exp_manager import exp_manager +torch._dynamo.config.suppress_errors = True + mp.set_start_method("spawn", force=True)