strisunshinewentingwang/pytorch

Pull Requests Commits

[functorch] Added support for negative in_dims (pytorch/functorch#136)

vfdev-5 committed 2 years ago

6a8b18b1

[functorch] updated batching rules guide

Chillee committed 2 years ago

7b4451bd

[functorch] Turn -Werror off

zou3519 committed 2 years ago

bafdc5d9

[functorch] refactored eager compilation a bit

Chillee committed 2 years ago

8ea093d3

[functorch] fix master

Chillee committed 2 years ago

59643a59

[functorch] update coverage script

zou3519 committed 2 years ago

03fff794

[functorch] Ported `_reshape_alias` to the new API (pytorch/functorch#130)

vfdev-5 committed 2 years ago

5237b85e

[functorch] Grid sample batch rule registration (pytorch/functorch#128)

Samantha Andow committed 2 years ago

9804cc1e

[functorch] apparently argmax/argmin are covered by our boxed reductions?

Chillee committed 2 years ago

fd82e4ee

[functorch] Support functions with multiple outputs in `compiled_function` (pytorch/functorch#127)

bertmaher committed 2 years ago

2d2d0cae

[functorch] Kill some more dead code; the CI doesn't use gcc?

zou3519 committed 2 years ago

43ed96f6

[functorch] Turn on -Werror

zou3519 committed 2 years ago

051b24d4

[functorch] Update OutOfPlacePlumbing; killed some UB

zou3519 committed 2 years ago

f70f562c

[functorch] Clean up some dead code; regenerate OutOfPlaceBatching

zou3519 committed 2 years ago

beecfbda

[functorch] Added guide to writing batching rules

Chillee committed 2 years ago

70246bdd

[functorch] updated opinfo db

Chillee committed 2 years ago

edf79c4b

[functorch] Added batching rule for index_put_.

Chillee committed 2 years ago

33f05ca5

[functorch] Added Index.Tensor batching rule (pytorch/functorch#123)

Chillee committed 2 years ago

050bf897

[functorch] NNC Compile: use BufHandle instead of Placeholder which is being deleted upstream.

Mikhail Zolotukhin committed 2 years ago

349dc2f3

[functorch] Nested jvp seems to work

zou3519 committed 2 years ago

1c96132a

[functorch] Some view backward batch rules

zou3519 committed 2 years ago

0753d0c4

[functorch] Revert "some view backward batch rules"

zou3519 committed 2 years ago

839517a3

[functorch] Fix README typo. (pytorch/functorch#120)

dan-zheng committed 2 years ago

ae1c2047

[functorch] some view backward batch rules

zou3519 committed 2 years ago

b6412da1

[functorch] Kill SkipInfo

zou3519 committed 2 years ago

0db39623

[functorch] remove scatter expected failure

Sam Andow committed 2 years ago

b47bc8f0

[functorch] move select to new API

Sam Andow committed 2 years ago

975a2df0

[functorch] Added VMAP_SUPPORT for returning vectors of tensors

Chillee committed 2 years ago

9cba6926

[functorch] Added wraps(f) to jacrev

Chillee committed 2 years ago

3a49b986

[functorch] removed some unnecessary batching rules from batchingregistrations.cpp

Chillee committed 2 years ago

dc9cbe1d

Newer Older