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

Benchmark batching funcs CLA Signed This label is managed by the Facebook bot. Authors need to sign the CLA before a PR can be reviewed. fb-exported
#187 opened Mar 31, 2022 by s4ayub Loading… updated Mar 31, 2022
Filter out batch_fused from available kernels if fused_params is not explicitly set and contains optimizer CLA Signed This label is managed by the Facebook bot. Authors need to sign the CLA before a PR can be reviewed. fb-exported
#298 opened May 5, 2022 by YLGH Loading… updated May 5, 2022
Update Lazy Extension so in sync with Pytorch CLA Signed This label is managed by the Facebook bot. Authors need to sign the CLA before a PR can be reviewed. fb-exported
#339 opened May 16, 2022 by dstaay-fb Loading… updated May 16, 2022
Repro the ghost processes for TorchAsyncITer CLA Signed This label is managed by the Facebook bot. Authors need to sign the CLA before a PR can be reviewed. fb-exported
#521 opened Jul 12, 2022 by RenfeiChen-FB Loading… updated Jul 12, 2022
Support sparse gradient for embedding bag collection CLA Signed This label is managed by the Facebook bot. Authors need to sign the CLA before a PR can be reviewed. fb-exported
#522 opened Jul 12, 2022 by mmenz Loading… updated Jul 12, 2022
add JaggedTensorMeta CLA Signed This label is managed by the Facebook bot. Authors need to sign the CLA before a PR can be reviewed. fb-exported
#532 opened Jul 21, 2022 by yinbinm Loading… updated Jul 21, 2022
add get_fused_optimizers CLA Signed This label is managed by the Facebook bot. Authors need to sign the CLA before a PR can be reviewed. fb-exported
#558 opened Aug 1, 2022 by YLGH Loading… updated Aug 1, 2022
Modify torchrec code to use reduce_scatter_v and all_gather_v CLA Signed This label is managed by the Facebook bot. Authors need to sign the CLA before a PR can be reviewed. fb-exported
#568 opened Aug 5, 2022 by YazhiGao Loading… updated Aug 5, 2022
Disable sub-group pg on RDMA hosts CLA Signed This label is managed by the Facebook bot. Authors need to sign the CLA before a PR can be reviewed. fb-exported
#573 opened Aug 11, 2022 by xw285cornell Loading… updated Aug 22, 2022
Allow DMP to pass prefix fused params 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 fx
#584 opened Aug 20, 2022 by xing-liu Loading… updated Aug 23, 2022
Move context creation into input_dist(...) CLA Signed This label is managed by the Facebook bot. Authors need to sign the CLA before a PR can be reviewed. fb-exported
#610 opened Sep 8, 2022 by xing-liu Loading… updated Sep 8, 2022
Fix False -> None for apply_overlapped_optimizer in tests CLA Signed This label is managed by the Facebook bot. Authors need to sign the CLA before a PR can be reviewed. fb-exported
#637 opened Sep 16, 2022 by YLGH Loading… updated Sep 16, 2022
Add helper to apply overlapped optimizer to 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
#638 opened Sep 16, 2022 by lequytra Loading… updated Sep 16, 2022
Add a test to ensure feature_processor weights are checkpointed CLA Signed This label is managed by the Facebook bot. Authors need to sign the CLA before a PR can be reviewed. fb-exported
#645 opened Sep 19, 2022 by xush6528 Loading… updated Sep 19, 2022
Decouple compute kernel from 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
#678 opened Sep 30, 2022 by YLGH Loading… updated Sep 30, 2022
Back out "Enable shard storage estimator to consider multiple optimizer classes" CLA Signed This label is managed by the Facebook bot. Authors need to sign the CLA before a PR can be reviewed. fb-exported
#684 opened Oct 4, 2022 by GD06 Loading… updated Oct 4, 2022
support int4 quantization in prospector CLA Signed This label is managed by the Facebook bot. Authors need to sign the CLA before a PR can be reviewed. fb-exported
#688 opened Oct 5, 2022 by nsiddharth Loading… updated Oct 5, 2022
Remove ShardedTensor-specific copy_ from KeyedOptimizer CLA Signed This label is managed by the Facebook bot. Authors need to sign the CLA before a PR can be reviewed. fb-exported
#695 opened Oct 6, 2022 by lequytra Loading… updated Oct 6, 2022
fix state_dict compatibility with ebc family CLA Signed This label is managed by the Facebook bot. Authors need to sign the CLA before a PR can be reviewed. fb-exported
#703 opened Oct 10, 2022 by YazhiGao Loading… updated Oct 10, 2022
create fx argument for sparsefeature types CLA Signed This label is managed by the Facebook bot. Authors need to sign the CLA before a PR can be reviewed. fb-exported
#708 opened Oct 11, 2022 by YazhiGao Loading… updated Oct 11, 2022
Remove test installation CLA Signed This label is managed by the Facebook bot. Authors need to sign the CLA before a PR can be reviewed.
#709 opened Oct 11, 2022 by YLGH Loading… updated Oct 11, 2022
compute kernel derived from user CLA Signed This label is managed by the Facebook bot. Authors need to sign the CLA before a PR can be reviewed.
#716 opened Oct 12, 2022 by YLGH Loading… updated Oct 12, 2022
Allow pass in fused params in default sharders CLA Signed This label is managed by the Facebook bot. Authors need to sign the CLA before a PR can be reviewed. fb-exported
#734 opened Oct 18, 2022 by 842974287 Loading… updated Oct 18, 2022
Upstream apply_optim_in_backward from TorchRec CLA Signed This label is managed by the Facebook bot. Authors need to sign the CLA before a PR can be reviewed. fb-exported
#738 opened Oct 20, 2022 by rohan-varma Loading… updated Oct 20, 2022
ci: Migrate unittest_ci to generic linux_job CLA Signed This label is managed by the Facebook bot. Authors need to sign the CLA before a PR can be reviewed.
#739 opened Oct 21, 2022 by seemethere Draft updated Oct 21, 2022
ProTip! Mix and match filters to narrow down what you’re looking for.