torchrec icon indicating copy to clipboard operation
torchrec copied to clipboard

Published 20 hours ago • verified publisher icon pytorch

Non-strict export fix in split and hardening test framework

Open PaulZhang12 opened this issue 9 months ago • 2 comments

Summary: As there are more instances of KJT/TorchRec data types used in PT2 IR, more edge cases are popping up. This diff fixes a bug and hardens the test framework by introducing weights for the KJT for all tests alongside simulating dynamic shapes for values and weights.

Differential Revision: D57163211

PaulZhang12 avatar May 09 '24 21:05 PaulZhang12

This pull request was exported from Phabricator. Differential Revision: D57163211

facebook-github-bot avatar May 09 '24 21:05 facebook-github-bot

This pull request was exported from Phabricator. Differential Revision: D57163211

facebook-github-bot avatar May 09 '24 21:05 facebook-github-bot