strisunshinewentingwang/pytorch

Pull Requests Commits

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

Chillee committed 2 years ago

65c54e28

[functorch] Added full_like and refactored factory macros a bit

Chillee committed 2 years ago

6916cc5d

[functorch] Added triu/tril

Chillee committed 2 years ago

b815b5bc

[functorch] finished batching rules for div

Chillee committed 2 years ago

0aedd9e8

[functorch] Added sort batching rules

Chillee committed 2 years ago

345cf3eb

[functorch] finished off where batching rules + added OP_DECOMPOSE macro

Chillee committed 2 years ago

94a55b7e

[functorch] Added where/_s_where, changed flatten (a composite op) to decomposition, and added (var/std).correction

Chillee committed 2 years ago

7b847833

[functorch] Added erf/inverse/isinf/isnan batching rules

Chillee committed 2 years ago

60fbba9d

[functorch] Added isnan + topk batching rules

Chillee committed 2 years ago

58319030

[functorch] Added clamp.tensor batching rule

Chillee committed 2 years ago

7bb63ff2

[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

Newer Older