strisunshinewentingwang/pytorch

Pull Requests Commits

[functorch] Custom Function prototype

zou3519 committed 2 years ago

c6b7ffab

[functorch] fixed test failures

Chillee committed 2 years ago

b1460ea2

[functorch] Added aot_function/aot_module which alias compiled_function/compiled_module

Chillee committed 2 years ago

7f76d5c9

[functorch] layer norm backward batch rule

zou3519 committed 2 years ago

b2d69184

[functorch] Added tests for vmap and autocast (pytorch/functorch#327)

vfdev-5 committed 2 years ago

ca41c5f0

[functorch] fix int64 issue

Chillee committed 2 years ago

ba04c326

[functorch] Update discover_coverage.py

zou3519 committed 2 years ago

5dd39989

[functorch] Fixed typo in docstring code-snippet (pytorch/functorch#316)

vfdev-5 committed 2 years ago

935d96f0

[functorch] Quick hack

zou3519 committed 2 years ago

495fa989

[functorch] modified tanh_backward to avoid rsub

Chillee committed 2 years ago

0c8edf25

[functorch] renamed API

Chillee committed 2 years ago

6a52a485

[functorch] Single cache (pytorch/functorch#319)

anijain2305 committed 2 years ago

30c7c337

[functorch] Added addr op decomposition (pytorch/functorch#323)

vfdev-5 committed 2 years ago

f5ac3641

[functorch] Added cholesky_solve op batch rule (pytorch/functorch#324)

vfdev-5 committed 2 years ago

c4390fb8

[functorch] [batching] diagonal_scatter (pytorch/functorch#315)

kshitij12345 committed 2 years ago

fdafa34f

[functorch] Fix CI

zou3519 committed 2 years ago

affb4d86

[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

Newer Older