strisunshinewentingwang/pytorch

Pull Requests Commits

[functorch] Add `has_aux=False` to vjp, jacrev (pytorch/functorch#335)

zou3519 committed 2 years ago

f12896b4

[functorch] Kill dead code

zou3519 committed 2 years ago

742d4c9e

[functorch] Added batch rule for LU op (pytorch/functorch#326)

vfdev-5 committed 2 years ago

f30ff855

[functorch] Introduce ForceLocalDispatchKeySet

zou3519 committed 2 years ago

291cbf09

[functorch] Refactor DynamicLayerFront dispatch key selection

zou3519 committed 2 years ago

808306fb

[functorch] Some more xfails

zou3519 committed 2 years ago

9b998052

[functorch] Probably fix CI

zou3519 committed 2 years ago

042f83eb

[functorch] Add some utilities for writing batching rules in python

zou3519 committed 2 years ago

c7c15eed

[functorch] Move DynamicLayer implementation out of header

zou3519 committed 2 years ago

2396dc35

[functorch] DynamicLayerStackHolder -> FuncTorchTLS

zou3519 committed 2 years ago

9ed57423

[functorch] Deduplicate pushDynamicLayer

zou3519 committed 2 years ago

e0ace21b

[functorch] Add setDynamicLayerFrontBackKeysIncluded helper function

zou3519 committed 2 years ago

1e7c7ca5

[functorch] Move all_dynlayer_keyset location

zou3519 committed 2 years ago

ccb277f4

[functorch] Fix CI again

zou3519 committed 2 years ago

99c712e9

[functorch] Revert "Refactor DynamicLayer.cpp"

zou3519 committed 2 years ago

23a11504

[functorch] Fix setup.py, hopefully to work on windows

zou3519 committed 2 years ago

9ad55a21

[functorch] Refactor DynamicLayer.cpp

zou3519 committed 2 years ago

73788801

[functorch] Debugging improvements

zou3519 committed 2 years ago

a9f69501

[functorch] Fix CI

zou3519 committed 2 years ago

0ade178c

[functorch] actually fixed things

Chillee committed 2 years ago

fbe01d27

[functorch] fixed test failures

Chillee committed 2 years ago

a7479002

[functorch] Custom Function prototype

zou3519 committed 2 years ago

c6b7ffab

[functorch] fixed test failures

Chillee committed 2 years ago

b1460ea2

[functorch] Added aot_function/aot_module which alias compiled_function/compiled_module

Chillee committed 2 years ago

7f76d5c9

[functorch] layer norm backward batch rule

zou3519 committed 2 years ago

b2d69184

[functorch] Added tests for vmap and autocast (pytorch/functorch#327)

vfdev-5 committed 2 years ago

ca41c5f0

[functorch] fix int64 issue

Chillee committed 2 years ago

ba04c326

[functorch] Update discover_coverage.py

zou3519 committed 2 years ago

5dd39989

[functorch] Fixed typo in docstring code-snippet (pytorch/functorch#316)

vfdev-5 committed 2 years ago

935d96f0

[functorch] Quick hack

zou3519 committed 2 years ago

495fa989

Newer Older