pytorch

pytorch copied to clipboard

Fix stream type to generic in comms default hooks

4

In comms default_hooks - decompress stream is hardcoded to cuda type. fix this to use generic type based on the grad tensor device. cc @mrshenli @pritamdamania87 @zhaojuanmao @satgera @rohan-varma @gqchen...

jeejakp12

Add global configurations to cache key

1

Stack from [ghstack](https://github.com/ezyang/ghstack) (oldest at bottom): * __->__ #126907 * #124745 This adds a bunch of global configurations to the cache key. There's definitely more I haven't added, but this...

jamesjwu

add strobelight cli function profiler

7

This is a meta only tool, this allow users to profile any python function by annotating it with **strobelight** using the strobelight profiler. ex ``` def fn(x, y, z): return...

laithsakka

Use sleef on macOS Apple silicon by default

2

Use sleef for aarch64 by default. cc @voznesenskym @penguinwu @EikanWang @jgong5 @Guobing-Chen @XiaobingSuper @zhuhaozhe @blzheng @wenzhe-nrv @jiayisunx @peterbell10 @ipiszy @yf225 @chenyang78 @kadeng @muchulee8 @ColinPeppler @amjames @desertfire @chauhang

helloguo

Make nn.Module state_dict load_state_dict pre-hook and state_dict post hook public

1

Fixes https://github.com/pytorch/pytorch/issues/75287 and https://github.com/pytorch/pytorch/issues/117437 Stack from [ghstack](https://github.com/ezyang/ghstack) (oldest at bottom): * __->__ #126704 * #126906 cc @mrshenli @pritamdamania87 @zhaojuanmao @satgera @gqchen @aazzolini @osalpekar @jiayisuse @H-Huang @kwen2501 @awgu @penguinwu @fegin @XilunWu...

mikaylagawarecki

[easy] Move state_dict hooks tests to test_module_hooks and annotate tests that call load_state_dict with swap

1

Stack from [ghstack](https://github.com/ezyang/ghstack) (oldest at bottom): * #126704 * __->__ #126906

mikaylagawarecki

[pytree] update treespec dict keys access

4

Stack from [ghstack](https://github.com/ezyang/ghstack) (oldest at bottom): * #114947 * #116375 * __->__ #116372 * #116374 1. Add a new method `treespec.entries()` 2. Change `treespec.context` access to `treespec.entries()` to hide the...

XuehaiPan

Reset grad state across unittests

4

Stack from [ghstack](https://github.com/ezyang/ghstack) (oldest at bottom): * __->__ #126345

williamwen42

Add torchao nightly testing workflow

2

Add and test torchao nightly testing workflow. This workflow will be triggered under the following conditions: 1. If the PR has ciflow/torchao label 2. Manual trigger It will run the...

xuzhao9

[Split Build] Load dependencies from libtorch in __init__.py

1

Stack from [ghstack](https://github.com/ezyang/ghstack) (oldest at bottom): * #126813 * #126699 * __->__ #126826 * #126590 * #126328 This PR makes it such that we search for a libtorch wheel when...

PaliC