search

pytorch / torchrec

Pytorch domain library for recommendation systems
BSD 3-Clause "New" or "Revised" License
1.79k stars 381 forks source link

Fix Sigmoid PT2 test to mimic AIMP PT2 and refactor torchbind class register #2185

Closed PaulZhang12 closed 1 week ago

PaulZhang12 commented 1 week ago

Summary: Fix PT2 sigmoid test with updated non-strict export mode and mimic AIMP PT2 path: https://fburl.com/code/y84jsrwo.

Given use of registering torchbind classes in multiple files when fixing this diff, put registration/deregistration in utils file.

Differential Revision: D59117177

facebook-github-bot commented 1 week ago

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

facebook-github-bot commented 1 week ago

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