Skip to content

Pull requests: pytorch/torchrec

New pull request New
225 Open 1,667 Closed
225 Open 1,667 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

Reland [torchrec][inference] Fix device propagation, tests for cpu 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
#2038 opened May 23, 2024 by gnahzg Loading…
1
QEC permute order caching
CLA Signed This label is managed by the Facebook bot. Authors need to sign the CLA before a PR can be reviewed. fb-exported
#2037 opened May 23, 2024 by gnahzg Loading…
1
QEBC permute order caching
CLA Signed This label is managed by the Facebook bot. Authors need to sign the CLA before a PR can be reviewed. fb-exported
#2036 opened May 23, 2024 by gnahzg Loading…
1
Import sparse_ops
CLA Signed This label is managed by the Facebook bot. Authors need to sign the CLA before a PR can be reviewed. fb-exported
#2029 opened May 21, 2024 by Microve Loading…
2
Wrap block_bucketize_sparse_features_inference to make unbucketize_permute non optional
CLA Signed This label is managed by the Facebook bot. Authors need to sign the CLA before a PR can be reviewed. fb-exported
#2028 opened May 21, 2024 by yumin829928 Loading…
2
Fix ParameterConstraints docstrings
CLA Signed This label is managed by the Facebook bot. Authors need to sign the CLA before a PR can be reviewed. fb-exported
#2022 opened May 20, 2024 by henrylhtsang Loading…
2
Formatting
CLA Signed This label is managed by the Facebook bot. Authors need to sign the CLA before a PR can be reviewed. fb-exported
#2020 opened May 20, 2024 by PaulZhang12 Loading…
1
test_comm test inductor
CLA Signed This label is managed by the Facebook bot. Authors need to sign the CLA before a PR can be reviewed. fb-exported
#2017 opened May 20, 2024 by IvanKobzarev Loading…
2
add dense kernel 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.
#2012 opened May 17, 2024 by Ye-Tian-Zero 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
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 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…
ProTip! Filter pull requests by the default branch with base:main.