Skip to content

Commit a77e054

Browse files
aporialiaofacebook-github-bot
authored andcommitted
Add is_weighted() call to FPEBC (#3140)
Summary: Pull Request resolved: #3140 Resolving Bug surfaced here: https://docs.google.com/document/d/1K5V1UGRj9Hn85dV6Ka8Fa1nLr7TwsQGGdn6cmFqS75s/edit?tab=t.0 Reviewed By: aliafzal Differential Revision: D77331144 fbshipit-source-id: 8d5570bdb8a6f8c66acc9857a2165de07900b3d3
1 parent a6d6f62 commit a77e054

File tree

1 file changed

+3
-0
lines changed

1 file changed

+3
-0
lines changed

torchrec/modules/fp_embedding_modules.py

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -146,6 +146,9 @@ def split(
146146
assert isinstance(self._feature_processors, FeatureProcessorsCollection)
147147
return self._feature_processors, self._embedding_bag_collection
148148

149+
def is_weighted(self) -> bool:
150+
return self._embedding_bag_collection.is_weighted()
151+
149152
def forward(
150153
self,
151154
features: KeyedJaggedTensor,

0 commit comments

Comments
 (0)