strisunshinewentingwang/pytorch

Pull Requests Commits

[functorch] index_put : batch-rule (pytorch/functorch#426)

kshitij12345 committed 2 years ago

92f552cd

[functorch] some minor fixes to AOTAutograd

Chillee committed 2 years ago

b13db682

[functorch] cleaned up partitioning code a little

Chillee committed 2 years ago

dbd2080d

[functorch] Better error message for static argnums (pytorch/functorch#431)

anijain2305 committed 2 years ago

f5f1dbd3

[functorch] Fix more fixable things in the CI (pytorch/functorch#427)

zou3519 committed 2 years ago

53db0bb6

[functorch] Fix CI (pytorch/functorch#425)

zou3519 committed 2 years ago

13d569ab

[functorch] fix vmap : index_put_ (handle broadcasting of value) (pytorch/functorch#401)

kshitij12345 committed 2 years ago

88c722d8

[functorch] Added static_argnums to nnc_jit

Chillee committed 2 years ago

4c78fd7f

[functorch] Enabled pytree of tensors output for jvp, jacfwd (pytorch/functorch#404)

vfdev-5 committed 2 years ago

5b6b1711

[functorch] Fixed some test cases/added some easy batching rules (pytorch/functorch#416)

Chillee committed 2 years ago

89443f4a

[functorch] Fixes flake8

vfdev-5 committed 2 years ago

d9125de7

[functorch] [requirements] missing networkx (pytorch/functorch#403)

stas00 committed 2 years ago

07679ac6

[functorch] [Compile Cache] Caching the forward and backward compiler ids (pytorch/functorch#413)

anijain2305 committed 2 years ago

5d821c1a

[functorch] Updated README (pytorch/functorch#414)

zou3519 committed 2 years ago

5582fab4

[functorch] fixed some misc things with AOTAutograd

Chillee committed 2 years ago

a052366e

[functorch] fixing test that missed beause jvp != vjp

Sam Andow committed 2 years ago

dbe13adf

[functorch] hopefully last test fix

Sam Andow committed 2 years ago

f4632d94

[functorch] Use empty tensor instead of efficient zeros tensor (pytorch/functorch#406)

zou3519 committed 2 years ago

3c8b626b

[functorch] fix unexpected successes

Sam Andow committed 2 years ago

23ed9418

[functorch] fix failing tests

Sam Andow committed 2 years ago

30355ad6

[functorch] Fixes printing issue with TensorWrapper (pytorch/functorch#397)

vfdev-5 committed 2 years ago

2ffb5a97

[functorch] Cleaning up TVM compiler integration (pytorch/functorch#405)

anijain2305 committed 2 years ago

9157be2b

[functorch] Revert "Minor refactor for default partition. (pytorch/functorch#390)"

Chillee committed 2 years ago

781d932e

[functorch] Minor refactor for default partition. (pytorch/functorch#390)

anijain2305 committed 2 years ago

3ba43c26

[functorch] Fixed unexpected success for cartesian_prod (pytorch/functorch#402)

vfdev-5 committed 2 years ago

3cdbf2c4

[functorch] tragically, yet another flake8 fix (I thought I'd got them all already this time :( )

Chillee committed 2 years ago

faf642a8

[functorch] Added support for outputs that don't require gradients + fixed some other tests

Chillee committed 2 years ago

fc272929

[functorch] Added cartesian_prod decomposition

Chillee committed 2 years ago

9ac203cd

[functorch] Fix failing tests and added comment about contiguity assumptions

Chillee committed 2 years ago

bc6f6d69

[functorch] fix another minor thing

Chillee committed 2 years ago

20377928

Newer Older