diff --git a/torchrec/distributed/quant_embedding.py b/torchrec/distributed/quant_embedding.py index 2e86053bb..0f91edc30 100644 --- a/torchrec/distributed/quant_embedding.py +++ b/torchrec/distributed/quant_embedding.py @@ -343,7 +343,6 @@ def _construct_jagged_tensors_tw( return ret -@torch.fx.wrap def _fx_marker_construct_jagged_tensor( values: torch.Tensor, lengths: torch.Tensor,