pytorch

pytorch copied to clipboard

[Autograd] Cond Higher-Order Operation

1

This is an updated PR to equip cond with the autograd feature and replaces the old [PR](https://github.com/pytorch/pytorch/pull/126007) @ydwu4 I tried to incorporate your requests already. Currently there are two problems...

bohnstingl

[FSDP2] Used `CommDebugMode` for comm. count test

2

Stack from [ghstack](https://github.com/ezyang/ghstack) (oldest at bottom): * #126892 * __->__ #126887 simplify the test :) cc @mrshenli @pritamdamania87 @zhaojuanmao @satgera @gqchen @aazzolini @osalpekar @jiayisuse @H-Huang @kwen2501 @penguinwu @fegin @XilunWu @wanchaol...

awgu

Allow multiple cudagraph recordings per compiled graph

1

Stack from [ghstack](https://github.com/ezyang/ghstack) (oldest at bottom): * __->__ #126822 * #126821 * #126820 cc @voznesenskym @penguinwu @EikanWang @jgong5 @Guobing-Chen @XiaobingSuper @zhuhaozhe @blzheng @wenzhe-nrv @jiayisunx @peterbell10 @ipiszy @yf225 @chenyang78 @kadeng @muchulee8...

mlazos

Add a registry for GraphModuleSerializer

1

This PR adds a registration function and a global registry for GraphModuleSerializer. After this PR, custom serialization methods can be done through registration instead of subclassing for ease of maintenance....

jiashenC

[ONNX] Skip assertion nodes

4

Continue from #125930 The PR addresses the common existence of unused assertion nodes in models. (workaround #112443) Assertion ops does not return anything in the graph, so it's not even...

titaiwangms

[easy] Delete dead code global

1

Stack from [ghstack](https://github.com/ezyang/ghstack) (oldest at bottom): * __->__ #126903 * #126083 cc @voznesenskym @penguinwu @EikanWang @jgong5 @Guobing-Chen @XiaobingSuper @zhuhaozhe @blzheng @wenzhe-nrv @jiayisunx @chenyang78 @kadeng @chauhang

oulgen

[DOCS] Fixed KLDiv example

5

Small import fix to make the example run

elisim

[Do not merge] Test for "Cuda124 arm ci"

1

Will this fixes issues in #126174 ? cc @tinglvv

nWEIdia

[Split Build] Test split build in CI

1

Stack from [ghstack](https://github.com/ezyang/ghstack) (oldest at bottom): * #126813 * __->__ #126699 * #126826 * #126590 * #126328 Specifically this PR tests building the split build on our manywheel builds on...

PaliC

[Split Build] Test split build in pull CI workflow

1

Stack from [ghstack](https://github.com/ezyang/ghstack) (oldest at bottom): * __->__ #126813 * #126699 * #126826 * #126590 * #126328 This PR builds the split build in the pull workflow and runs the...

PaliC