Skip to content

Commit 414ca05

Browse files
committed
Revert for Prod "temp for test: test decorator"
This reverts commit c6dc162.
1 parent 244714f commit 414ca05

File tree

2 files changed

+1
-5
lines changed

2 files changed

+1
-5
lines changed

megatron/core/distributed/distributed_data_parallel.py

Lines changed: 0 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -7,7 +7,6 @@
77
import torch
88

99
from .. import parallel_state
10-
from ..backwards_compatibility_decorators import exempt_from_compat_check
1110
from ..config_logger import has_config_logger_enabled, log_config_to_disk
1211
from ..fp8_utils import is_float8tensor
1312
from ..process_groups_config import ProcessGroupCollection
@@ -21,7 +20,6 @@
2120
logger = logging.getLogger(__name__)
2221

2322

24-
@exempt_from_compat_check
2523
class DistributedDataParallel(_BaseDataParallel):
2624
"""
2725
DDP wrapper which stores grads in contiguous buffers. Also has option of overlapping

megatron/core/model_parallel_config.py

Lines changed: 1 addition & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -4,11 +4,9 @@
44
from dataclasses import dataclass
55
from typing import Callable, ContextManager, Optional
66

7-
from .backwards_compatibility_decorators import exempt_from_compat_check
8-
97
import torch
108

11-
@exempt_from_compat_check
9+
1210
@dataclass
1311
class ModelParallelConfig:
1412
"""Base configuration for Megatron Core

0 commit comments

Comments
 (0)