strisunshinewentingwang/pytorch

Pull Requests Commits

[functorch] port and fix unfold batching rule (pytorch/functorch#206)

kshitij12345 committed 2 years ago

4e431704

[functorch] Make it so that vmap tests generate with bdim=-1 as well as 0 (pytorch/functorch#204)

zou3519 committed 2 years ago

f220892d

[functorch] Fixed code-snippet with jacrev (pytorch/functorch#197)

vfdev-5 committed 2 years ago

fabbb9b3

[functorch] Fix PointwiseCompiler on CUDA (pytorch/functorch#203)

bertmaher committed 2 years ago

18c37d3b

[functorch] "Scorecard" benchmarks for pointwise op authoring (pytorch/functorch#193)

bertmaher committed 2 years ago

51e5e36f

[functorch] Fix max_pool2d batch rule (pytorch/functorch#202)

zou3519 committed 2 years ago

c3df3e13

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

zou3519 committed 2 years ago

39f05e05

[functorch] Remove xfails

zou3519 committed 2 years ago

543f6b2d

[functorch] max_pool2d_backward batch rule

zou3519 committed 2 years ago

cbd20344

[functorch] Make one of the boxed fallbacks faster

zou3519 committed 2 years ago

768be7a3

[functorch] Fix batch rule for max_pool2d

zou3519 committed 2 years ago

4ed8ddaa

[functorch] resolve_neg, resolve_conj

zou3519 committed 2 years ago

51be1e02

[functorch] add scatter.reduce batching rule (pytorch/functorch#188)

kshitij12345 committed 2 years ago

a475ae8e

[functorch] Support conj bit, neg bit

zou3519 committed 2 years ago

13bd7521

[functorch] Ravel fix, pt3

zou3519 committed 2 years ago

f07936ed

[functorch] fix ravel part 2

zou3519 committed 2 years ago

0e66c857

[functorch] Fix ravel

zou3519 committed 2 years ago

f95bc12c

[functorch] Update script

zou3519 committed 2 years ago

01b84b49

[functorch] temporary xfail for nll_loss

zou3519 committed 2 years ago

fde8d1ed

[functorch] broadcast_to just works thanks to expand (pytorch/functorch#191)

kshitij12345 committed 2 years ago

37810948

[functorch] Add some minor comments to scatter plumbing

Chillee committed 2 years ago

fea84d48

[functorch] removes existing_bdim_boxed (which did nothing sensible...)

Chillee committed 2 years ago

3b2b7367

[functorch] cleaned up some eager compilation stuff

Chillee committed 2 years ago

f92a5f87

[functorch] Batch rule for avg_pool2d_backward

zou3519 committed 2 years ago

206b2364

[functorch] Get rid of new_blah_hacks; diagflat batch rule

zou3519 committed 2 years ago

2e774b12

[functorch] Added index_select batching rule (pytorch/functorch#183)

vfdev-5 committed 2 years ago

d6d36eac

[functorch] Update install.sh (pytorch/functorch#189)

vfdev-5 committed 2 years ago

e0b12716

[functorch] Fix CI by unpinning the binary version (pytorch/functorch#187)

zou3519 committed 2 years ago

e26a33a7

[functorch] add scatter_add batch rule (pytorch/functorch#182)

kshitij12345 committed 2 years ago

0b93c18d

[functorch] Re-land the compile cache (pytorch/functorch#169)

bertmaher committed 2 years ago

51501b4e

Newer Older