strisunshinewentingwang/pytorch

Pull Requests Commits

[functorch] Add some quick short-term hacks (pytorch/functorch#317)

zou3519 committed 2 years ago

cad8b765

[functorch] diag_backward batch rule

zou3519 committed 2 years ago

d091220a

[functorch] Update lagging op db (pytorch/functorch#318)

zou3519 committed 2 years ago

1f7822f6

[functorch] fixed CI

Chillee committed 2 years ago

d7fb5002

[functorch] made recompute partitioning prune unused saved values

Chillee committed 2 years ago

6812607e

[functorch] Use FuncTorchTLSBase instead of ThreadLocalDebugInfo (pytorch/functorch#314)

zou3519 committed 2 years ago

ba25a7ec

[functorch] Fix CI

zou3519 committed 2 years ago

73a02476

[functorch] fixed draw_graph

Chillee committed 2 years ago

ac4f7961

[functorch] [Benchmark] Layer norm patterns (pytorch/functorch#311)

anijain2305 committed 2 years ago

98bbf3d5

[functorch] update op analysis script

Chillee committed 2 years ago

85576dbb

[functorch] Added a bunch of __foo__ ops

Chillee committed 2 years ago

421aeeb0

[functorch] [Benchmarking] Adding scripts for lightseq benchmarking (pytorch/functorch#310)

anijain2305 committed 2 years ago

60cdeb66

[functorch] Fix CI

zou3519 committed 2 years ago

306eda08

[functorch] Fix CI

zou3519 committed 2 years ago

6f882bb4

[functorch] Fix CI

zou3519 committed 2 years ago

211b2148

[functorch] Fix CI

zou3519 committed 2 years ago

6f4330e6

[functorch] Fix some linting

zou3519 committed 2 years ago

20997b49

[functorch] Added grid_sample backward batch rule (pytorch/functorch#284)

vfdev-5 committed 2 years ago

1d956a0a

[functorch] replace reshape with view (pytorch/functorch#304)

kshitij12345 committed 2 years ago

d9c63e2d

[functorch] Added logit decompositions

Chillee committed 2 years ago

f03ac2ec

[functorch] added some more decompositions

Chillee committed 2 years ago

73443c14

[functorch] Add node about potential exponential blowup

Chillee committed 2 years ago

211e0958

[functorch] fix decomposition tests to work on GPU

Chillee committed 2 years ago

9720b5b1

[functorch] [batching] pixel_un|shuffle (pytorch/functorch#295)

kshitij12345 committed 2 years ago

f6baad47

[functorch] added some more decompositions and rejiggered the testing infra

Chillee committed 2 years ago

2553c1a2

[functorch] modified gitignore

Chillee committed 2 years ago

5eb9a1d1

[functorch] updated with some more decompositions

Chillee committed 2 years ago

ab64065e

[functorch] jacobian / hessian correctness tests (pytorch/functorch#301)

zou3519 committed 2 years ago

c51aceb8

[functorch] fix unexpected scucess with hessian test

Chillee committed 2 years ago

036b0d87

[functorch] moved decompositions to their own file

Chillee committed 2 years ago

d1f81270

Newer Older