torchrec
torchrec copied to clipboard
Changing Backend Tensor initialization
Summary: X-link: https://github.com/facebookresearch/FBGEMM/pull/2066
Context: Currently, we are enabling SSD optimizer offloading for the ssd tbe kernel
In this diff: We retrieve the newly added parameters from the tbe config and pass it down to the tbe
Differential Revision: D85353134
@Raahul46 has exported this pull request. If you are a Meta employee, you can view the originating Diff in D85353134.