strisunshinewentingwang/pytorch

Pull Requests Commits

[functorch] removed no bdim cases from batchruleslinearalgebra

Chillee committed 2 years ago

cb286b9b

[functorch] Added a test for case where we would previously have dispatched to a batching rule despite having no bdims

Chillee committed 2 years ago

8cd80e0b

[functorch] Selectively enable dispatch on kBatchedKey (pytorch/functorch#63)

zou3519 committed 2 years ago

446d0b4e

[functorch] Added einsum batching rule

Chillee committed 2 years ago

bbfdcdbd

[functorch] updated opinfo DB and added meshgrid batching rule

Chillee committed 2 years ago

495112f5

[functorch] Exclude failing tests, add xfail test for torch.tensor

zou3519 committed 2 years ago

f6667347

[functorch] Add flip batching rule

Chillee committed 2 years ago

c02cc07c

[functorch] Added batching rule for logsumexp

Chillee committed 2 years ago

b20e4dec

[functorch] Added norm.Scalar and norm.ScalarOpt_dim

Chillee committed 2 years ago

9c138786

[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

Newer Older