strisunshinewentingwang/pytorch

Pull Requests Commits

[functorch] added option to skip specialization cache

Chillee committed 2 years ago

7e13e3c6

[functorch] Cleaned up the tests

Chillee committed 2 years ago

c0afd464

[functorch] fix some tests

Chillee committed 2 years ago

0562725c

[functorch] stored some stuff

Chillee committed 2 years ago

d314e2f7

[functorch] moved function to eager_compilation

Chillee committed 2 years ago

cb22fefe

[functorch] Added start of compile_funciton opinfo tests

Chillee committed 2 years ago

09b23abf

[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

Newer Older