-
Notifications
You must be signed in to change notification settings - Fork 379
Pull requests: pytorch/torchrec
Author
Label
Projects
Milestones
Reviews
Assignee
Sort
Pull requests list
fix EmbeddingTrainPipelineTest: test_equal_to_non_pipelined
CLA Signed
This label is managed by the Facebook bot. Authors need to sign the CLA before a PR can be reviewed.
fb-exported
#2175
opened Jun 25, 2024 by
iamzainhuda
Loading…
refactor using encapsulate and decapsulate
CLA Signed
This label is managed by the Facebook bot. Authors need to sign the CLA before a PR can be reviewed.
fb-exported
#2174
opened Jun 25, 2024 by
TroyGarden
Loading…
Metric State Vectorization
CLA Signed
This label is managed by the Facebook bot. Authors need to sign the CLA before a PR can be reviewed.
fb-exported
#2171
opened Jun 25, 2024 by
iamzainhuda
Loading…
copy _permute_tensor_by_segments to avoid packaging import error
CLA Signed
This label is managed by the Facebook bot. Authors need to sign the CLA before a PR can be reviewed.
fb-exported
#2170
opened Jun 25, 2024 by
joshuadeng
Loading…
Grouped table generation for unified embedding
CLA Signed
This label is managed by the Facebook bot. Authors need to sign the CLA before a PR can be reviewed.
fb-exported
#2164
opened Jun 24, 2024 by
Fiery
Loading…
Refactor source code parity check with This label is managed by the Facebook bot. Authors need to sign the CLA before a PR can be reviewed.
ast
CLA Signed
#2160
opened Jun 22, 2024 by
XuehaiPan
Loading…
benchmark of fbgemm op - regroup_keyed_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
#2159
opened Jun 22, 2024 by
TroyGarden
Loading…
benchmark of fbgemm op - permute_multi_embedding
CLA Signed
This label is managed by the Facebook bot. Authors need to sign the CLA before a PR can be reviewed.
fb-exported
#2158
opened Jun 22, 2024 by
TroyGarden
Loading…
Refactor passing over cache params
CLA Signed
This label is managed by the Facebook bot. Authors need to sign the CLA before a PR can be reviewed.
fb-exported
#2155
opened Jun 21, 2024 by
henrylhtsang
Loading…
Prevent excess buffer allocation
CLA Signed
This label is managed by the Facebook bot. Authors need to sign the CLA before a PR can be reviewed.
fb-exported
#2154
opened Jun 21, 2024 by
Damowerko
Loading…
Remove unused exception parameter from titan/yada/tests/messenger/MarkReadApplicatorTest.cpp
CLA Signed
This label is managed by the Facebook bot. Authors need to sign the CLA before a PR can be reviewed.
fb-exported
#2153
opened Jun 21, 2024 by
r-barnes
Loading…
Fix bug in criteo.py that caused NaN issues
CLA Signed
This label is managed by the Facebook bot. Authors need to sign the CLA before a PR can be reviewed.
#2150
opened Jun 21, 2024 by
TomekWei
Loading…
Add ability to specify pipelineable preproc modules to ignore during SDD model rewrite
CLA Signed
This label is managed by the Facebook bot. Authors need to sign the CLA before a PR can be reviewed.
fb-exported
#2149
opened Jun 20, 2024 by
sarckk
Loading…
Make PositionWeightedModuleCollection faster with torch.export
CLA Signed
This label is managed by the Facebook bot. Authors need to sign the CLA before a PR can be reviewed.
fb-exported
#2146
opened Jun 20, 2024 by
PaulZhang12
Loading…
Reland preserve device in empty lookup case
CLA Signed
This label is managed by the Facebook bot. Authors need to sign the CLA before a PR can be reviewed.
fb-exported
#2143
opened Jun 20, 2024 by
PaulZhang12
Loading…
Back out " [tps][api] add configs for 1) client thread num 2) max key per request 3) max local index length 4) ps hosts & ports"
CLA Signed
This label is managed by the Facebook bot. Authors need to sign the CLA before a PR can be reviewed.
fb-exported
#2142
opened Jun 20, 2024 by
henrylhtsang
Loading…
Give user option for multiplier to control preference for rank 0 or rank 1-N for Heteroplanner uneven 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
#2138
opened Jun 20, 2024 by
gnahzg
Loading…
change regroup backend operator for KeyedTensor
CLA Signed
This label is managed by the Facebook bot. Authors need to sign the CLA before a PR can be reviewed.
fb-exported
#2128
opened Jun 17, 2024 by
TroyGarden
Loading…
fix test_utils mismatch in CI
CLA Signed
This label is managed by the Facebook bot. Authors need to sign the CLA before a PR can be reviewed.
fb-exported
#2123
opened Jun 17, 2024 by
joshuadeng
Loading…
FBGEMM kernel for KeyedTensor (PooledEmbedding) permute mapping
CLA Signed
This label is managed by the Facebook bot. Authors need to sign the CLA before a PR can be reviewed.
fb-exported
#2120
opened Jun 17, 2024 by
TroyGarden
Loading…
Overlap comms on backward pass
CLA Signed
This label is managed by the Facebook bot. Authors need to sign the CLA before a PR can be reviewed.
fb-exported
#2117
opened Jun 15, 2024 by
dstaay-fb
Loading…
support serialization of compound module
CLA Signed
This label is managed by the Facebook bot. Authors need to sign the CLA before a PR can be reviewed.
fb-exported
#2082
opened Jun 6, 2024 by
TroyGarden
Loading…
register custom_op for fpEBC
CLA Signed
This label is managed by the Facebook bot. Authors need to sign the CLA before a PR can be reviewed.
fb-exported
#2067
opened Jun 4, 2024 by
TroyGarden
Loading…
Add fbgemm load_library permute_pooled_embedding_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
#2063
opened May 31, 2024 by
IvanKobzarev
Loading…
improved fx graph for inference batching
CLA Signed
This label is managed by the Facebook bot. Authors need to sign the CLA before a PR can be reviewed.
fb-exported
#2052
opened May 28, 2024 by
YazhiGao
Loading…
Previous Next
ProTip!
Add no:assignee to see everything that’s not assigned.