OSSForks/pytorch
1
0
Fork 0
mirror of https://github.com/zebrajr/pytorch.git synced 2025-12-07 12:21:27 +01:00
Code Issues Actions 37 Packages Projects Releases Wiki Activity
49,824 Commits 8,304 Branches 150 Tags 1.5 GiB
97a5d9b703
Commit Graph

82 Commits

Author SHA1 Message Date
Animesh Jain
97a5d9b703 [functorch] Disable autocast (pytorch/functorch#794) 
* Disable autocast

* Add global flag

* Add a test
 2022-07-21 13:41:36 -07:00
Animesh Jain
6050b42388 [functorch] Present Random state (pytorch/functorch#887) 
* Present Random state

* Add tests
 2022-07-21 13:41:36 -07:00
Horace He
6c8874f1a9 [functorch] unify PythonKey (i.e. ProxyTensor) tracer with the one in core (pytorch/functorch#853) 
* unify tracer with the one in core

* modified test

* fix lint issues

* fixed some things
 2022-07-21 13:41:35 -07:00
Samantha Andow
c048d19772 [functorch] fix CI (pytorch/functorch#816) 2022-07-21 13:41:34 -07:00
Samantha Andow
d9b25b1a2a [functorch] fix ci (pytorch/functorch#789) 2022-07-21 13:41:33 -07:00
Horace He
07e000478f [functorch] fix flake 2022-07-21 13:41:33 -07:00
Horace He
1b2f01e712 [functorch] skip unpool tests on python key too 2022-07-21 13:41:33 -07:00
Horace He
fc24004a9f [functorch] Updates to the min-cut partitioner to improve it (pytorch/functorch#653) 
* updated partitioner

* some minor modifications

* stashed changes

* update some lists

* added some more decompositions

* remove clone decompositions

* updated partitioner
 2022-07-21 13:41:30 -07:00
Richard Zou
a46fbe8d79 [functorch] skip flaky test 2022-07-21 13:41:30 -07:00
Richard Zou
244197a2a6 [functorch] Update lagging op db (pytorch/functorch#737) 2022-07-21 13:41:30 -07:00
Richard Zou
7bf45014ce [functorch] Cleanup return_types handling now that it has been upstreamed (pytorch/functorch#730) 
Fixes https://github.com/pytorch/functorch/issues/713
 2022-07-21 13:41:30 -07:00
albanD
0b8752b0f7 [functorch] Fix test following tls update in core (pytorch/functorch#718) 
This test now works!
 2022-07-21 13:41:30 -07:00
Horace He
352c07c2f5 [functorch] fixed some minor issues 2022-07-21 13:41:30 -07:00
Horace He
1a85653108 [functorch] fix issue where default partitioner might recompute things 2022-07-21 13:41:30 -07:00
Animesh Jain
38cbf390e2 [functorch] Skip extracting meta tensor info for sparse tensors (pytorch/functorch#676) 
* Skip extracting meta tensor info for sparse tensors

* Change expected failures
 2022-07-21 13:41:29 -07:00
Animesh Jain
cb931b1649 [functorch] Reduce overhead of AOT Module (pytorch/functorch#660) 
Adding aot_module_simplified and aot_function_simplified

Fallback to aot_module original until we prevent tracing of leaf modules
 2022-07-21 13:41:28 -07:00
Animesh Jain
f4ed73b3d1 [functorch] Trace the backward pass assuming contiguous tensors (pytorch/functorch#536) 2022-07-21 13:41:28 -07:00
Richard Zou
1cc3add0e7 [functorch] Update functorch lagging op db (pytorch/functorch#652) 2022-07-21 13:41:27 -07:00
Horace He
562c91b133 [functorch] removed special.zeta grad 2022-07-21 13:41:27 -07:00
Horace He
0eb5aa5c7d [functorch] fixed some op packet issues and added some more tests for partition optimality 2022-07-21 13:41:26 -07:00
Horace He
3a26772c53 [functorch] Actually made some modifications to fix OpOverload changes (pytorch/functorch#579) 
* actually fixed recent issues

* fixed new issues

* fixed overload issue

* Fix epsilons
 2022-07-21 13:41:26 -07:00
Animesh Jain
1ac79567fc [functorch] Setting tensor_meta attr for inplace ops (pytorch/functorch#565) 
* Setting tensor_meta attr for inplace ops

* Tests

* Linter

* Failures
 2022-07-21 13:41:25 -07:00
Samantha Andow
824eeb96e9 [functorch] remove s_where, use where.self (pytorch/functorch#581) 2022-07-21 13:41:25 -07:00
Edward Z. Yang
57e78c6654 [functorch] Don't unnecessarily wrap the elem in PythonTensor (pytorch/functorch#554) 
* Don't unnecessarily wrap the elem in PythonTensor

Instead of saying that a PythonTensor has a regular (e.g., CPU) tensor
and an FX proxy, a PythonTensor *is a* regular CPU tensor, that also
carries an FX proxy (that updates as we go along).

This should fix https://github.com/pytorch/functorch/issues/465 and
it also fixed some expected failures in the test suite.

This kills the meta variant logic entirely; maybe some other time we'll
try to bring it back.

Signed-off-by: Edward Z. Yang <ezyang@fb.com>
 2022-07-21 13:41:24 -07:00
Edward Z. Yang
6648a05b25 [functorch] Skip networkx if not installed 
Signed-off-by: Edward Z. Yang <ezyang@fb.com>
 2022-07-21 13:41:24 -07:00
Sam Andow
12724711b7 [functorch] skip decomps, fix ci 2022-07-21 13:41:22 -07:00
Horace He
d3406d9cd0 [functorch] Fix some CI fails for AOTAutograd and such 2022-07-21 13:41:21 -07:00
Horace He
7ba1e4303b [functorch] Fixed minor bug with partitioner and other minor changes 2022-07-21 13:41:21 -07:00
Richard Zou
81f3bd4d58 [functorch] Fix CI (pytorch/functorch#461) 2022-07-21 13:41:21 -07:00
Horace He
4b72e62178 [functorch] Added Grad context to AOTAutograd (pytorch/functorch#441) 
* refactored AOTAutograd code a bit

* Added grad context stack

* remove pdb

* fixed flake
 2022-07-21 13:41:20 -07:00
Sam Andow
23ed94181a [functorch] fix unexpected successes 2022-07-21 13:41:19 -07:00
Horace He
fc272929a6 [functorch] Added support for outputs that don't require gradients + fixed some other tests 2022-07-21 13:41:19 -07:00
Horace He
de4b1e2fc1 [functorch] fix CI test for an op I don't have locally yet :) 2022-07-21 13:41:18 -07:00
Horace He
747c2fa2b2 [functorch] fix flake issues 2022-07-21 13:41:18 -07:00
Horace He
02ae449c7b [functorch] refactored decompositions a bit 2022-07-21 13:41:18 -07:00
Sam Andow
761c382550 [functorch] fix ci by disabling certain tests 2022-07-21 13:41:18 -07:00
Richard Zou
67b8a097e7 [functorch] Fix flake8 2022-07-21 13:41:18 -07:00
Horace He
16f34150bd [functorch] port nnc_jit to use AOTAutograd 2022-07-21 13:41:18 -07:00
Horace He
89d30b2987 [functorch] Added better handling for output device inference (i.e. only use when metatracing) 2022-07-21 13:41:17 -07:00
Richard Zou
ae7db8eca8 [functorch] More docs for jacfwd, hessian, jvp (pytorch/functorch#362) 
Also moved them out of the experimental namespace but added warning on
coverage.
 2022-07-21 13:41:17 -07:00
vfdev
c94446a04d [functorch] Fixed python code formatting and added flake8 setup (pytorch/functorch#346) 
* Fixed python code formatting and added flake8 setup

* Fixes config.yaml

* Added missing setup.cfg

* Removed flake8 job from circle ci and setup GHA

* More flake8 fixes

* Fixed test_conv2d

* Fixed failing flake8
 2022-07-21 13:41:17 -07:00
Horace He
00dfb3b2a2 [functorch] Added support for output pytrees in AOTAutograd (pytorch/functorch#332) 
* Added autominifier

* added support for output pytree

* Added some test
 2022-07-21 13:41:16 -07:00
Richard Zou
c952118aeb [functorch] Fix flaky tests 2022-07-21 13:41:15 -07:00
Richard Zou
1f7822f650 [functorch] Update lagging op db (pytorch/functorch#318) 2022-07-21 13:41:14 -07:00
Horace He
73443c14a5 [functorch] added some more decompositions 2022-07-21 13:41:13 -07:00
Horace He
fe9aac72d1 [functorch] Added decomposition testing + display infra (pytorch/functorch#281) 
* Added decomposition testing + display infra

* add a couple more decompositions

* changed some stuff

* made some changes

* Added decomposition testing + display infra

* add a couple more decompositions

* fix some decompositions

* changed some stuff

* updated generation

* fix test failures

* removed extraneous files

* fixed test failures

* fixed tests

* updated

* fixed tests again
 2022-07-21 13:41:12 -07:00
vfdev
1f9c78cce7 [functorch] Removed some xfail for nn functional pad circular (pytorch/functorch#285) 2022-07-21 13:41:12 -07:00
Horace He
8769c9a233 [functorch] changed naming from aot autograd to eager compilation 2022-07-21 13:41:11 -07:00
Richard Zou
752d0ed847 [functorch] Skip testing empty_like instead of xfail (pytorch/functorch#259) 
This test is flaky.
 2022-07-21 13:41:11 -07:00
Richard Zou
41699bb9f5 [functorch] Update functorch lagging op db (pytorch/functorch#261) 2022-07-21 13:41:11 -07:00