Reland D87662877: Generate 1 acc graph for esr mb5 by removing fx wrapper for kjt (#3582)

Summary:
Pull Request resolved: #3582

D87662877 was reverted in D87747630 due to incompatibility between publish package and lowering package. E.g., In test it uses prod lowering package which does not include this diff change.

Reviewed By: yingufan

Differential Revision: D87848633

fbshipit-source-id: 1a45e5a79491873523e206829820d2b771a93013

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,