strisunshinewentingwang/pytorch

Pull Requests Commits

[functorch] Added static_argnums to nnc_jit

Chillee committed 2 years ago

4ff4501d

[functorch] actually fixed tests?

Chillee committed 2 years ago

dcea4261

[functorch] fix warnings

Chillee committed 2 years ago

66586e95

[functorch] moved a bunch of inplace ops to use boxed fallback

Chillee committed 2 years ago

a528be7d

[functorch] Added special handling for svd correctness

Chillee committed 2 years ago

13fab14f

[functorch] Add svd to test exclusions for now

Chillee committed 2 years ago

b5ae3dcd

[functorch] fix unnecessary thing that breaks custom ops

Chillee committed 2 years ago

b2deb179

[functorch] Added some decompositions and augmented boxed_reduction to support optional single dim args

Chillee committed 2 years ago

17bbac74

[functorch] Fix tests failing since I added batching rules

Chillee committed 2 years ago

6077a0b0

[functorch] Cleaned up some things, added some batching rules, etc. part 2

Chillee committed 2 years ago

1cc47c29

[functorch] Script to list out OpInfo coverage on our @ops tests

zou3519 committed 2 years ago

7671b9f7

[functorch] Batch rule coverage for vmap({grad, vjp}) testing

zou3519 committed 2 years ago

87180080

[functorch] Test for batch rule coverage

zou3519 committed 2 years ago

9707431d

[functorch] Cleaned up some things, added some batching rules, etc.

Chillee committed 2 years ago

a98f43e9

[functorch] Fixed handling for boxed fallback for optional dimarrays + scalar inputs, added linalg_eig, var_mean.correction, and std_mean.correction

Chillee committed 2 years ago

3fa1792c

[functorch] Added linalg_cholesky/inv_ex batching rules and removed STOP_DECOMPOSE entries on non-composite ops

Chillee committed 2 years ago

ad29ecb7

[functorch] Added svd/qr batching rules and cleaned up somew arnings

Chillee committed 2 years ago

19005eb1

[functorch] Added boxed variadic bdims batch rule and some linalg rules (linalg_slogdet, geqrf, logdet, matrix_exp, solve, symeig)

Chillee committed 2 years ago

6ce0325f

[functorch] Added a grab bag of batching rules (bitwise_or/xor, logaddexp, logaddexp2, kthvalue, isposinf, isneginf, nan_to_num, signbit)

Chillee committed 2 years ago

3c47e875

[functorch] Updated gen_data script to use new APIs

Chillee committed 2 years ago

486679ba

[functorch] Added pointwise backward batching rules (elu_backward, hardshrink_backward, logit_backward, log_sigmoid_backward, gelu_backward, softplus_backward)

Chillee committed 2 years ago

fddaff44

[functorch] Added median/nanmedian batching rules

Chillee committed 2 years ago

70d3b160

[functorch] Added some fft batching rules

Chillee committed 2 years ago

d02e227c

[functorch] Added any.dim, count_nonzero.dim_IntList, cummax, cummin, linalg_vector_norm, logcumsumexp, nansum, nansum.dim_IntList, and mvlgamma

Chillee committed 2 years ago

d741c68d

[functorch] fix some errors in reduction batching rules

Chillee committed 2 years ago

854d5380

[functorch] Switched over to boxed implementations for reductions :)))))))

Chillee committed 2 years ago

11bd7e33

[functorch] whups

Chillee committed 2 years ago

2fb4a597

[functorch] Added boxed pointwise batching rule and almost finished pointwise (addcdiv, addcmul, clamp.Tensor, frexp.Tensor, lerp.Scalar, lerp.Tensor, log_sigmoid_forward, polygamma)

Chillee committed 2 years ago

6a7a022f

[functorch] Specialization cache (pytorch/functorch#99)

zou3519 committed 2 years ago

f9144e94

[functorch] Run python key tests in CI (pytorch/functorch#98)

zou3519 committed 2 years ago

38803ae0

Newer Older