Skip to content

Commit 977254c

Browse files
tjruwasemrwyattii
andauthored
Disable z3 tracing profiler (#4106)
Co-authored-by: Michael Wyatt <[email protected]>
1 parent abe293b commit 977254c

File tree

1 file changed

+3
-1
lines changed

1 file changed

+3
-1
lines changed

deepspeed/runtime/zero/partitioned_param_coordinator.py

+3-1
Original file line numberDiff line numberDiff line change
@@ -18,6 +18,8 @@
1818
from deepspeed.accelerator import get_accelerator
1919
import logging
2020

21+
ENABLE_PROFILER = False
22+
2123

2224
def debug_rank0(message: str) -> None:
2325
if dist.get_rank() == 0:
@@ -117,7 +119,7 @@ def __init__(
117119
self.__ongoing_fetch_events: Deque[get_accelerator().Event] = collections.deque()
118120
# TODO. make this configurable via JSON
119121
self.__max_ongoing_fetch_events: int = 2
120-
self.__profiler = PartitionedParameterProfiler(timers)
122+
self.__profiler = PartitionedParameterProfiler(timers if ENABLE_PROFILER else None)
121123

122124
"""Tracing and Tracking
123125
TODO. consider performing trace before initializing PartitionedParameterCoordinator

0 commit comments

Comments
 (0)