strisunshinewentingwang/pytorch

Pull Requests Commits

[functorch] Added support for output pytrees in AOTAutograd (pytorch/functorch#332)

Chillee committed 2 years ago

00dfb3b2

[functorch] More batch rule fixes (pytorch/functorch#348)

zou3519 committed 2 years ago

d7e7fe01

[functorch] fix discover_coverage bug

zou3519 committed 2 years ago

6311baf0

[functorch] Random assortment of batch rules and cleanups (pytorch/functorch#347)

zou3519 committed 2 years ago

152e00f3

[functorch] Fix convolution batch rule in the transpose case (pytorch/functorch#345)

zou3519 committed 2 years ago

476d21bc

[functorch] added syntax highlighting

Chillee committed 2 years ago

6722ee77

[functorch] added compilation readme

Chillee committed 2 years ago

6dd6b061

[functorch] hopefully fixed CI - started skipping conv1d tests

Chillee committed 2 years ago

3d9eee4a

[functorch] Added subprocess launcher for NVFuser minifying

Chillee committed 2 years ago

db07b249

[functorch] added some tests and stuff to minimizer

Chillee committed 2 years ago

967da4b6

[functorch] jvp transform fully composes with vmap (pytorch/functorch#340)

zou3519 committed 2 years ago

2ade2226

[functorch] Another fix

zou3519 committed 2 years ago

695990e7

[functorch] Fix CPU CI

zou3519 committed 2 years ago

953c5d50

[functorch] Fix CI hopefully

zou3519 committed 2 years ago

d6157327

[functorch] add conditional functionalization (pytorch/functorch#235)

bdhirsh committed 2 years ago

9052153f

[functorch] Fix some "unexpected successes"

zou3519 committed 2 years ago

f0301db1

[functorch] Fix undefined bias for conv batching rule

zou3519 committed 2 years ago

24c93d8e

[functorch] Fix CI

zou3519 committed 2 years ago

56961ad1

[functorch] Revert "Fix one composability problem"

zou3519 committed 2 years ago

c0fdee06

[functorch] Fix one composability problem

zou3519 committed 2 years ago

73d5d732

[functorch] Some more utilities

zou3519 committed 2 years ago

3a7e2a98

[functorch] Fix flaky tests

zou3519 committed 2 years ago

c952118a

[functorch] Added autominifier for FX graphs (pytorch/functorch#330)

Chillee committed 2 years ago

a7bf7831

[functorch] Add `has_aux=False` to vjp, jacrev (pytorch/functorch#335)

zou3519 committed 2 years ago

f12896b4

[functorch] Kill dead code

zou3519 committed 2 years ago

742d4c9e

[functorch] Added batch rule for LU op (pytorch/functorch#326)

vfdev-5 committed 2 years ago

f30ff855

[functorch] Introduce ForceLocalDispatchKeySet

zou3519 committed 2 years ago

291cbf09

[functorch] Refactor DynamicLayerFront dispatch key selection

zou3519 committed 2 years ago

808306fb

[functorch] Some more xfails

zou3519 committed 2 years ago

9b998052

[functorch] Probably fix CI

zou3519 committed 2 years ago

042f83eb

Newer Older