strisunshinewentingwang/pytorch

Pull Requests Commits

[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

[functorch] modified tanh_backward to avoid rsub

Chillee committed 2 years ago

0c8edf25

[functorch] renamed API

Chillee committed 2 years ago

6a52a485

[functorch] Single cache (pytorch/functorch#319)

anijain2305 committed 2 years ago

30c7c337

[functorch] Added addr op decomposition (pytorch/functorch#323)

vfdev-5 committed 2 years ago

f5ac3641

[functorch] Added cholesky_solve op batch rule (pytorch/functorch#324)

vfdev-5 committed 2 years ago

c4390fb8

[functorch] [batching] diagonal_scatter (pytorch/functorch#315)

kshitij12345 committed 2 years ago

fdafa34f

[functorch] Fix CI

zou3519 committed 2 years ago

affb4d86

Newer Older