We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent abe293b commit 977254cCopy full SHA for 977254c
deepspeed/runtime/zero/partitioned_param_coordinator.py
@@ -18,6 +18,8 @@
18
from deepspeed.accelerator import get_accelerator
19
import logging
20
21
+ENABLE_PROFILER = False
22
+
23
24
def debug_rank0(message: str) -> None:
25
if dist.get_rank() == 0:
@@ -117,7 +119,7 @@ def __init__(
117
119
self.__ongoing_fetch_events: Deque[get_accelerator().Event] = collections.deque()
118
120
# TODO. make this configurable via JSON
121
self.__max_ongoing_fetch_events: int = 2
- self.__profiler = PartitionedParameterProfiler(timers)
122
+ self.__profiler = PartitionedParameterProfiler(timers if ENABLE_PROFILER else None)
123
124
"""Tracing and Tracking
125
TODO. consider performing trace before initializing PartitionedParameterCoordinator
0 commit comments