Skip to content

Pull requests: pytorch/torchrec

Author
Filter by author
Loading
Label
Filter by label
Loading
Use alt + click/return to exclude labels
or + click/return for logical OR
Projects
Filter by project
Loading
Milestones
Filter by milestone
Loading
Reviews
Assignee
Filter by who’s assigned
Sort

Pull requests list

move module attribute inplace update to leaf function in ManagedCollisionModule CLA Signed This label is managed by the Facebook bot. Authors need to sign the CLA before a PR can be reviewed. fb-exported
#2913 opened Apr 23, 2025 by seanx92 Loading…
Add unsharded module property to sharded modules and EBC CLA Signed This label is managed by the Facebook bot. Authors need to sign the CLA before a PR can be reviewed. fb-exported
#2912 opened Apr 23, 2025 by aporialiao Loading…
Enable Optimizer Storing & Fix incomplete updates to Sharded EBC attributes in resharding CLA Signed This label is managed by the Facebook bot. Authors need to sign the CLA before a PR can be reviewed. fb-exported
#2911 opened Apr 23, 2025 by aporialiao Loading…
use pin_memory for ModelInput and minor refactoring CLA Signed This label is managed by the Facebook bot. Authors need to sign the CLA before a PR can be reviewed. fb-exported
#2910 opened Apr 23, 2025 by TroyGarden Loading…
Move util function for creating VBE KJT to the VbModelInput class CLA Signed This label is managed by the Facebook bot. Authors need to sign the CLA before a PR can be reviewed. fb-exported
#2906 opened Apr 22, 2025 by jd7-tr Loading…
Enable More General Comms Bandwidth Estimates CLA Signed This label is managed by the Facebook bot. Authors need to sign the CLA before a PR can be reviewed. fb-exported
#2905 opened Apr 22, 2025 by gregmacnamara Loading…
Test include stride_per_key to KJT's flatten and unflatten CLA Signed This label is managed by the Facebook bot. Authors need to sign the CLA before a PR can be reviewed. fb-exported
#2903 opened Apr 22, 2025 by jd7-tr Loading…
Track ITEP-enabled models on APS with scuba logging (#2736) CLA Signed This label is managed by the Facebook bot. Authors need to sign the CLA before a PR can be reviewed. fb-exported
#2902 opened Apr 22, 2025 by AKhazane Loading…
Add unsharded module reference to sharded modules CLA Signed This label is managed by the Facebook bot. Authors need to sign the CLA before a PR can be reviewed. fb-exported
#2901 opened Apr 22, 2025 by aporialiao Loading…
Enable custom perf models in MemoryBalancedPartitioner CLA Signed This label is managed by the Facebook bot. Authors need to sign the CLA before a PR can be reviewed. fb-exported
#2897 opened Apr 19, 2025 by micrain Loading…
Perf Model Modeling Critical Path of Sparse Arch CLA Signed This label is managed by the Facebook bot. Authors need to sign the CLA before a PR can be reviewed. fb-exported
#2896 opened Apr 19, 2025 by micrain Loading…
Back out "Fuse "states" level tensors to reduce all gather during metrics compute" CLA Signed This label is managed by the Facebook bot. Authors need to sign the CLA before a PR can be reviewed. fb-exported
#2893 opened Apr 17, 2025 by iamzainhuda Loading…
Bucket offsets and sizes in torchrec shard metadata for bucket wise sharding (#2884) CLA Signed This label is managed by the Facebook bot. Authors need to sign the CLA before a PR can be reviewed. fb-exported
#2885 opened Apr 13, 2025 by faran928 Loading…
support bucket offsets and sizes in shard metadata CLA Signed This label is managed by the Facebook bot. Authors need to sign the CLA before a PR can be reviewed. fb-exported
#2884 opened Apr 12, 2025 by faran928 Loading…
create rfc for flexible collision free embedding table proposal CLA Signed This label is managed by the Facebook bot. Authors need to sign the CLA before a PR can be reviewed. fb-exported
#2883 opened Apr 11, 2025 by emlin Loading…
Resolve Pyre CLA Signed This label is managed by the Facebook bot. Authors need to sign the CLA before a PR can be reviewed. fb-exported
#2880 opened Apr 11, 2025 by aporialiao Loading…
Refactor stride_per_key_per_rank to support torch.Tensor CLA Signed This label is managed by the Facebook bot. Authors need to sign the CLA before a PR can be reviewed. fb-exported
#2872 opened Apr 8, 2025 by jd7-tr Loading…
Update build-wheels-linux runner to Ubuntu-24.04 CLA Signed This label is managed by the Facebook bot. Authors need to sign the CLA before a PR can be reviewed.
#2865 opened Apr 1, 2025 by malfet Loading…
Fix pyre test on OSS remove ignore CLA Signed This label is managed by the Facebook bot. Authors need to sign the CLA before a PR can be reviewed. fb-exported
#2859 opened Mar 31, 2025 by aporialiao Loading…
Re-land torchrec pipeline refactor (with fix) CLA Signed This label is managed by the Facebook bot. Authors need to sign the CLA before a PR can be reviewed. fb-exported
#2853 opened Mar 28, 2025 by che-sh Loading…
Allow undefined libraries to fix CMake errors CLA Signed This label is managed by the Facebook bot. Authors need to sign the CLA before a PR can be reviewed. fb-exported
#2849 opened Mar 26, 2025 by aporialiao Loading…
Force import with post script OSS job CLA Signed This label is managed by the Facebook bot. Authors need to sign the CLA before a PR can be reviewed. fb-exported
#2840 opened Mar 20, 2025 by aporialiao Loading…
Back out "Replace LR access with wrapper" CLA Signed This label is managed by the Facebook bot. Authors need to sign the CLA before a PR can be reviewed. fb-exported
#2839 opened Mar 20, 2025 by spcyppt Loading…
Add SimpleFSDP to leaf node CLA Signed This label is managed by the Facebook bot. Authors need to sign the CLA before a PR can be reviewed. fb-exported
#2835 opened Mar 19, 2025 by Microve Loading…
ProTip! Adding no:label will show everything without a label.