We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent 8f3ef8c commit 06f1604Copy full SHA for 06f1604
nemo/collections/nlp/models/language_modeling/megatron_base_model.py
@@ -408,7 +408,9 @@ def is_official_release_version(nvidia_torch_version):
408
self.cfg.persist_layer_norm = False
409
410
# NVFUSER available starting with 21.11
411
- if NVIDIA_TORCH_MAJOR >= 21 or (NVIDIA_TORCH_MAJOR == 21 and NVIDIA_TORCH_MINOR >= 11):
+ if (NVIDIA_TORCH_MAJOR >= 21 or (NVIDIA_TORCH_MAJOR == 21 and NVIDIA_TORCH_MINOR >= 11)) and (
412
+ NVIDIA_TORCH_MAJOR < 23 or (NVIDIA_TORCH_MAJOR == 23 and NVIDIA_TORCH_MINOR < 11)
413
+ ):
414
415
# NVFUSER
416
torch._C._jit_set_profiling_executor(True)
0 commit comments