Skip to content

Commit 115aaa8

Browse files
aliafzalmeta-codesync[bot]
authored andcommitted
Update DeltaCheckpointing and DeltaPublish with generic model tracker (#3543)
Summary: Pull Request resolved: #3543 internal Updating DeltaCP and DeltaPublish to with ModelDeltaTracker to allow these functionalities to use either of DeltaTracker implementation General Context: We are in the process of transition to a unified DeltaTracker and this is 9/n diffs representing changes towards the transition. Differential Revision: D80615817 Privacy Context Container: L1158701 fbshipit-source-id: 3bf88feb28a490648a83864cba995968ba45e759
1 parent 6ffcd37 commit 115aaa8

File tree

1 file changed

+4
-4
lines changed

1 file changed

+4
-4
lines changed

torchrec/distributed/model_tracker/model_delta_tracker.py

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -94,7 +94,7 @@ def record_lookup(
9494
kjt (KeyedJaggedTensor): The KeyedJaggedTensor containing IDs to record.
9595
states (torch.Tensor): The embeddings or states corresponding to the IDs in the kjt.
9696
"""
97-
pass
97+
...
9898

9999
@abstractmethod
100100
def get_unique_ids(self, consumer: Optional[str] = None) -> Dict[str, torch.Tensor]:
@@ -104,7 +104,7 @@ def get_unique_ids(self, consumer: Optional[str] = None) -> Dict[str, torch.Tens
104104
Args:
105105
consumer (str, optional): The consumer to retrieve unique IDs for.
106106
"""
107-
pass
107+
...
108108

109109
@abstractmethod
110110
def get_unique(
@@ -120,7 +120,7 @@ def get_unique(
120120
Args:
121121
consumer (str, optional): The consumer to retrieve delta values for.
122122
"""
123-
pass
123+
...
124124

125125
@abstractmethod
126126
def clear(self, consumer: Optional[str] = None) -> None:
@@ -130,7 +130,7 @@ def clear(self, consumer: Optional[str] = None) -> None:
130130
Args:
131131
consumer (str, optional): The consumer to clear IDs/States for.
132132
"""
133-
pass
133+
...
134134

135135
def step(self) -> None:
136136
"""

0 commit comments

Comments
 (0)