strisunshinewentingwang/pytorch

Pull Requests Commits

[functorch] Finished most of the rest of the pad batching rules (also added an existing_batch_dim_template)

Chillee committed 2 years ago

308f4775

[functorch] Added batching rules for constant_pad_nd

Chillee committed 2 years ago

a5a7245e

[functorch] updated opinfo db and fixed failing tests

Chillee committed 2 years ago

266d2ced

[functorch] Revert "We actually fixed a lot of the vjp problems"

zou3519 committed 2 years ago

ed65f0a8

[functorch] We actually fixed a lot of the vjp problems

zou3519 committed 2 years ago

22ad9d47

[functorch] Resolved some vjp failures

zou3519 committed 2 years ago

2ceba07e

[functorch] Normalize, cross_entropy OpInfos

zou3519 committed 2 years ago

5cfa5728

[functorch] OpInfo for pad

zou3519 committed 2 years ago

f8caad7f

[functorch] additional OpInfo for Conv2d

zou3519 committed 2 years ago

4028de1d

[functorch] More OpInfos

zou3519 committed 2 years ago

f28e1996

[functorch] Add `functorch_additional_op_db`

zou3519 committed 2 years ago

6b59f1ad

[functorch] Some more batch rules for pointwise ops

zou3519 committed 2 years ago

d5069519

[functorch] batch rules for torch.special unary ops

zou3519 committed 2 years ago

983a43cf

[functorch] Quick grab bag of batch rules

zou3519 committed 2 years ago

1b78cae7

[functorch] Fix CI; for real this time

zou3519 committed 2 years ago

6ca3e96e

[functorch] Fix ci

zou3519 committed 2 years ago

2e7ddb7a

[functorch] Added some make_fx+vjp/jac/vmap tests

Chillee committed 2 years ago

6d39fa33

[functorch] Add make_fx(grad(..)) test

Chillee committed 2 years ago

8e62e271

[functorch] Update README.md

Chillee committed 2 years ago

236d2f20

[functorch] Added a quick benchmark

zou3519 committed 2 years ago

046453c6

[functorch] Added citation section

zou3519 committed 2 years ago

8d127816

[functorch] Fix parametrized testing

zou3519 committed 2 years ago

57f48cc6

[functorch] Fix grad wrapper behavior when wrapping Python tensor

zou3519 committed 2 years ago

fea6254d

[functorch] removed comment that previous commit fixed :)

Chillee committed 2 years ago

e836870e

[functorch] ported random registrations to boxedfallback

Chillee committed 2 years ago

d71bb374

[functorch] Added errors for dynamic ops

Chillee committed 2 years ago

b28268c8

[functorch] Batch rule for resize_ when batch dim is at front

zou3519 committed 2 years ago

899fea5e

[functorch] Added dropout (tested by opinfo to be upstreamed)

Chillee committed 2 years ago

0bba49c8

[functorch] Added softmax batching rule (tested through local OpInfo, to be upstreamed)

Chillee committed 2 years ago

a598ed6b

[functorch] Added dist decomposition

Chillee committed 2 years ago

abd8c710

Newer Older