strisunshinewentingwang/pytorch

Pull Requests Commits

[functorch] add atleast_nd decompositions + tests

Chillee committed 2 years ago

11a363e3

[functorch] moved movedim decomposition to batchrulesstopdecomposition

Chillee committed 2 years ago

dc5311cb

[functorch] readd movedim (pytorch/functorch#221)

Padarn committed 2 years ago

2a9538e6

[functorch] stopped decomposing by default (pytorch/functorch#215)

Chillee committed 2 years ago

1cd009ae

[functorch] Update setup.py (pytorch/functorch#223)

vfdev-5 committed 2 years ago

8823ad84

[functorch] cleanup ci (pytorch/functorch#216)

Samantha Andow committed 2 years ago

ffb2fe05

[functorch] Fix NDEBUG build (pytorch/functorch#217)

zou3519 committed 2 years ago

36cc7fb1

[functorch] update discover_coverage

zou3519 committed 2 years ago

feada960

[functorch] Fix CI

zou3519 committed 2 years ago

59427646

[functorch] Implemented nll loss through decomposition (pytorch/functorch#208)

vfdev-5 committed 2 years ago

93416ece

[functorch] [Partitioning] Recompute forward in the backward pass (pytorch/functorch#213)

anijain2305 committed 2 years ago

cbf900b3

[functorch] [Op-Authoring] Adding mapping from torch ops to ExprHandles (pytorch/functorch#205)

anijain2305 committed 2 years ago

f50fee6c

[functorch] convert movedim to new api (pytorch/functorch#211)

Padarn committed 2 years ago

f46adc90

[functorch] fix CI

Chillee committed 2 years ago

4f7269f9

[functorch] Clean up perf scorecard and add barplot generation script (pytorch/functorch#212)

bertmaher committed 2 years ago

4174156a

[functorch] Excise onlyOnCPUAndCUDA

zou3519 committed 2 years ago

7f7e9dc9

[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

Newer Older