Generate 1 acc graph for esr mb5 by removing fx wrapper for kjt (#3571)

Summary: Pull Request resolved: #3571

Reviewed By: yingufan, trieuat

Differential Revision: D87662877

fbshipit-source-id: 9caa15cb61469352f25012ac8e1e30982449e89b

Author: Yixin Bao

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,