Skip to content

Commit 06f1604

Browse files
github-actions[bot]athitten
authored andcommitted
Disable nvFuser setup with PyTorch 23.11 and later (NVIDIA#9837) (NVIDIA#9870)
* Disable nvFuser setup with PyTorch 23.11 and later * Apply isort and black reformatting --------- Signed-off-by: Abhishree <[email protected]> Signed-off-by: athitten <[email protected]> Co-authored-by: Abhishree Thittenamane <[email protected]> Co-authored-by: athitten <[email protected]>
1 parent 8f3ef8c commit 06f1604

File tree

1 file changed

+3
-1
lines changed

1 file changed

+3
-1
lines changed

nemo/collections/nlp/models/language_modeling/megatron_base_model.py

+3-1
Original file line numberDiff line numberDiff line change
@@ -408,7 +408,9 @@ def is_official_release_version(nvidia_torch_version):
408408
self.cfg.persist_layer_norm = False
409409

410410
# NVFUSER available starting with 21.11
411-
if NVIDIA_TORCH_MAJOR >= 21 or (NVIDIA_TORCH_MAJOR == 21 and NVIDIA_TORCH_MINOR >= 11):
411+
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+
):
412414

413415
# NVFUSER
414416
torch._C._jit_set_profiling_executor(True)

0 commit comments

Comments
 (0)