Skip to content

Pull requests: pytorch/torchrec

New pull request New
227 Open 1,638 Closed
227 Open 1,638 Closed
Author
Filter by author
Label
Filter by label
Use alt + click/return to exclude labels
or ⇧ + click/return for logical OR
Projects
Filter by project
Milestones
Filter by milestone
Reviews
Filter by reviews
No reviews Review required Approved review Changes requested
Assignee
Filter by who’s assigned
Sort
Sort by
Newest Oldest Most commented Least commented Recently updated Least recently updated Best match
Most reactions
👍 👎 😄 🎉 😕 ❤️ 🚀 👀

Pull requests list

Reduce prefetch pipeline IO cost by 1x of input size
CLA Signed This label is managed by the Facebook bot. Authors need to sign the CLA before a PR can be reviewed. fb-exported
#2011 opened May 16, 2024 by levythu Loading…
1
Introduce use_sync_collectives configuration
CLA Signed This label is managed by the Facebook bot. Authors need to sign the CLA before a PR can be reviewed. fb-exported
#2010 opened May 16, 2024 by IvanKobzarev Loading…
6
One table per TBE mode will NOT separate multiple shards within the same table
CLA Signed This label is managed by the Facebook bot. Authors need to sign the CLA before a PR can be reviewed. fb-exported
#2008 opened May 15, 2024 by levythu Loading…
1
Fix OSS 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
#2004 opened May 14, 2024 by sarckk Loading…
2
Remove extremely slow logic from RequestOnlySplitEmbeddingBagCollectionsSparseArch
CLA Signed This label is managed by the Facebook bot. Authors need to sign the CLA before a PR can be reviewed. fb-exported
#2003 opened May 14, 2024 by PaulZhang12 Loading…
2
Support multi-pass prefetch config in CacheParams, and shard estimator will correctly propagate its storage cost
CLA Signed This label is managed by the Facebook bot. Authors need to sign the CLA before a PR can be reviewed. fb-exported
#2000 opened May 14, 2024 by levythu Loading…
2
Convert torchrec to fast by default architecture] [batch:74/79]
CLA Signed This label is managed by the Facebook bot. Authors need to sign the CLA before a PR can be reviewed. fb-exported
#1994 opened May 13, 2024 by MaggieMoss Loading…
3
Convert .pyre_configuration.local to fast by default architecture] [batch:51/51]
CLA Signed This label is managed by the Facebook bot. Authors need to sign the CLA before a PR can be reviewed. fb-exported
#1992 opened May 13, 2024 by connernilsen Loading…
2
Replacing ShardedTensor with DTensor for RW sharding
CLA Signed This label is managed by the Facebook bot. Authors need to sign the CLA before a PR can be reviewed. fb-exported
#1991 opened May 13, 2024 by zainhuda Loading…
2
Fix Pyre-check
CLA Signed This label is managed by the Facebook bot. Authors need to sign the CLA before a PR can be reviewed. fb-exported
#1987 opened May 10, 2024 by PaulZhang12 Loading…
2
Prefetch for ITEP v8
CLA Signed This label is managed by the Facebook bot. Authors need to sign the CLA before a PR can be reviewed. fb-exported
#1983 opened May 10, 2024 by henrylhtsang Loading…
2
Pyre issues
CLA Signed This label is managed by the Facebook bot. Authors need to sign the CLA before a PR can be reviewed. fb-exported fx
#1957 opened May 7, 2024 by PaulZhang12 Loading…
1
Add gpu re test to prepare for train pipeline 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
#1953 opened May 4, 2024 by dstaay-fb Loading…
2
support non contiguous sharding
CLA Signed This label is managed by the Facebook bot. Authors need to sign the CLA before a PR can be reviewed. fb-exported
#1950 opened May 3, 2024 by xunnanxu Loading…
7
Fix dynamic embedding bugs that may lead to segmentation faults
CLA Signed This label is managed by the Facebook bot. Authors need to sign the CLA before a PR can be reviewed.
#1947 opened May 3, 2024 by sherlockkenan Loading…
10
fix is_weighted by using config
CLA Signed This label is managed by the Facebook bot. Authors need to sign the CLA before a PR can be reviewed.
#1940 opened May 1, 2024 by jiayisuse Loading…
Always return .int() conversion from unwrap_kjt
CLA Signed This label is managed by the Facebook bot. Authors need to sign the CLA before a PR can be reviewed.
#1939 opened Apr 30, 2024 by PaulZhang12 Loading…
Create BoundsCheckMode fused_param for inference
CLA Signed This label is managed by the Facebook bot. Authors need to sign the CLA before a PR can be reviewed.
#1938 opened Apr 30, 2024 by PaulZhang12 Loading…
Move eager model processing functions to torchrec/inference
CLA Signed This label is managed by the Facebook bot. Authors need to sign the CLA before a PR can be reviewed.
#1937 opened Apr 29, 2024 by PaulZhang12 Loading…
Benchmarking
CLA Signed This label is managed by the Facebook bot. Authors need to sign the CLA before a PR can be reviewed.
#1936 opened Apr 29, 2024 by dstaay-fb Loading…
Non-determinstic Flag
CLA Signed This label is managed by the Facebook bot. Authors need to sign the CLA before a PR can be reviewed.
#1935 opened Apr 29, 2024 by dstaay-fb Loading…
Fix bug in graph partitioner
CLA Signed This label is managed by the Facebook bot. Authors need to sign the CLA before a PR can be reviewed. fb-exported
#1934 opened Apr 29, 2024 by tugsbayasgalan Loading…
2
refactor mean pooling tests and test for all sharding types
CLA Signed This label is managed by the Facebook bot. Authors need to sign the CLA before a PR can be reviewed.
#1932 opened Apr 26, 2024 by zainhuda Loading…
Pyre Configurationless migration for] [batch:28/29]
CLA Signed This label is managed by the Facebook bot. Authors need to sign the CLA before a PR can be reviewed.
#1931 opened Apr 26, 2024 by connernilsen Loading…
Add num positive and negative examples to torch metrics
CLA Signed This label is managed by the Facebook bot. Authors need to sign the CLA before a PR can be reviewed.
#1930 opened Apr 26, 2024 by drdarshan Loading…
ProTip! What’s not been updated in a month: updated:<2024-04-16.