Edward Z. Yang
e6ec0efaf8
Apply UFMT to all non test/torch files ( #106205 )
...
Signed-off-by: Edward Z. Yang <ezyang@meta.com>
Pull Request resolved: https://github.com/pytorch/pytorch/pull/106205
Approved by: https://github.com/albanD
2023-07-29 02:56:24 +00:00
Animesh Jain
9157be2b10
[functorch] Cleaning up TVM compiler integration ( pytorch/functorch#405 )
...
* Cleaning up TVM compiler integration
* Linters
2022-07-21 13:41:19 -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
Bert Maher
033e7109a4
[functorch] Support buffers in compiled_module ( pytorch/functorch#147 )
...
* Support buffers in compiled_module
* Don't compute gradients for inputs that don't require grad
* Add a unit test for batchnorm
* Fix eager compilation tests that change requires_grad
* Create new args for tests without recompilation
* Enable some eager fusion opinfo tests that now work (because we stopped asking for unimplemented derivatives)
2022-07-21 13:41:06 -07:00
Horace He
7e13e3c6f5
[functorch] added option to skip specialization cache
2022-07-21 13:41:03 -07:00
Horace He
5466ac5da3
[functorch] Update eager_fusion.py
2022-07-21 13:41:00 -07:00
Horace He
e8a5b3725b
[functorch] fix example to work well :)
2022-07-21 13:40:59 -07:00
Horace He
1a2e538580
[functorch] Moved to Keops example
2022-07-21 13:40:59 -07:00
Horace He
9ca5ae86d0
[functorch] Added eager-mode fusion prototype
2022-07-21 13:40:59 -07:00
