strisunshinewentingwang/pytorch

Pull Requests Commits

[functorch] Added mode batching rule

Chillee committed 2 years ago

f787f0be

[functorch] Added max/min/prod (no dim)

Chillee committed 2 years ago

4f87a069

[functorch] Exclude List[Optional[Tensor]] from the batched fallback

zou3519 committed 2 years ago

85587011

[functorch] Added cumprod/cumsum and ported over log_softmax

Chillee committed 2 years ago

9094e757

[functorch] Added maximum/minimum/clamp batching rules

Chillee committed 2 years ago

ec25616e

[functorch] Added bmm batch rule

Chillee committed 2 years ago

9b00f55a

[functorch] Add deg2rad/rad2deg/radian batching rules

Chillee committed 2 years ago

e9c196e4

[functorch] add acosh/atanh/asinh batching rules

Chillee committed 2 years ago

ede0952e

[functorch] fix some bugs with reduction batching rule + add amax/amin

Chillee committed 2 years ago

ed2f5824

[functorch] Added sigmoid_backward and replaced a bunch of typedefs with decltypes

Chillee committed 2 years ago

43de84e0

[functorch] Added atan2 and zeros_ batching rules

Chillee committed 2 years ago

5af36052

[functorch] Added batching rules for convolution, conv1d, conv2d, conv3d, etc.

Chillee committed 2 years ago

56378e98

[functorch] _unsafe_view batch rule

zou3519 committed 2 years ago

1666d901

[functorch] Clarify the vmap fallback warning message

zou3519 committed 2 years ago

bea0df36

[functorch] Added note about reaching out

Chillee committed 2 years ago

cfcd328f

[functorch] Quick optimization

zou3519 committed 2 years ago

0b602921

[functorch] linalg_eigh batch rule

zou3519 committed 2 years ago

5acd1b99

[functorch] Fix some headers

zou3519 committed 2 years ago

75a64bb0

[functorch] Implement more batch rules for resnet18 per-sample-grads

zou3519 committed 2 years ago

79e42fc1

[functorch] Fix build/test

zou3519 committed 2 years ago

a054adeb

[functorch] cudnn_convolution decomposition

zou3519 committed 2 years ago

dc5e0c0f

[functorch] maxpool_2d_with_indices_backward batch rule for specific case

zou3519 committed 2 years ago

eb788ae9

[functorch] Implement per sample grad rule for cudnn_convolution_backward

zou3519 committed 2 years ago

1262034b

[functorch] quick fix

zou3519 committed 2 years ago

cfa9d984

[functorch] Introduce gen_plumbing.py

zou3519 committed 2 years ago

4a045e96

[functorch] Very experimental PyTorch forward-mode AD support

zou3519 committed 2 years ago

03fbb542

[functorch] Setup circleci (pytorch/functorch#53)

zou3519 committed 2 years ago

e8c5f67c

[functorch] Added enough things into denylists so that tests pass

zou3519 committed 2 years ago

4aea8064

[functorch] Create "lagging op database", use it in our OpInfo tests

zou3519 committed 2 years ago

3524505d

[functorch] Fix a lot of warnings; use c10::irange

zou3519 committed 2 years ago

fee49501

Newer Older