strisunshinewentingwang/pytorch

Pull Requests Commits

[functorch] Adding jacrev numargs argument (pytorch/functorch#111)

Samantha Andow committed 2 years ago

f20150a3

[functorch] removed unnecessary note

Chillee committed 2 years ago

b23c9b90

[functorch] removed stupid line

Chillee committed 2 years ago

fcd08e1a

[functorch] Fix location of file

zou3519 committed 2 years ago

ae5a1290

[functorch] Fix bug in discover_coverage

zou3519 committed 2 years ago

8bd71e0a

[functorch] Handle label smoothing in cross_entropy

zou3519 committed 2 years ago

818487a5

[functorch] Added dlpack conversion to/from for TVM

Chillee committed 2 years ago

7af7eddf

[functorch] Added handling of parameters so that requires_grad is propagated correctly

Chillee committed 2 years ago

9168d834

[functorch] merged

Chillee committed 2 years ago

b1390744

[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

Newer Older