strisunshinewentingwang/pytorch

Pull Requests Commits

[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

[functorch] some minor fixes

Chillee committed 2 years ago

afba1985

[functorch] [pytree] register return_types (pytorch/functorch#350)

kshitij12345 committed 2 years ago

a8d54b40

[functorch] xfail failing jvp tests

Sam Andow committed 2 years ago

524002be

[functorch] skip to solve nasty seg fault

Sam Andow committed 2 years ago

21bd4ba5

[functorch] Use num_bytes instead of numel in min-cut partitioning (pytorch/functorch#398)

anijain2305 committed 2 years ago

633d1f02

[functorch] group norm + backwards running (pytorch/functorch#387)

Samantha Andow committed 2 years ago

f0f65be5

[functorch] fix some partitioning issues

Chillee committed 2 years ago

e5980eab

[functorch] Cleanup for memory efficient fusion (pytorch/functorch#388)

anijain2305 committed 2 years ago

183e88cd

Newer Older