torchrec icon indicating copy to clipboard operation
torchrec copied to clipboard

Published 20 hours ago • verified publisher icon pytorch

Reland [TorchRec][PT2] KJT custom op for 1d lengths input

Open TroyGarden opened this issue 1 year ago • 1 comments

Summary:

context

  • previous landed D59031938 was reverted due to torchscript push schedule is behind
  • adding torch.jit.is_scripting() to protect the exposure.

Differential Revision: D59081243

TroyGarden avatar Jun 26 '24 22:06 TroyGarden