strisunshinewentingwang/pytorch

Pull Requests Commits

[functorch] Fix arg type of roll_batch_rule

zou3519 committed 2 years ago

78d8ed15

[functorch] Added tests for op aliases (pytorch/functorch#173)

vfdev-5 committed 2 years ago

9a5105de

[functorch] [port] `expand` to new api (pytorch/functorch#161)

kshitij12345 committed 2 years ago

854a0fd6

[functorch] Update README.md (pytorch/functorch#176)

vfdev-5 committed 2 years ago

15f24f0f

[functorch] temporary fix pth version to make CI pass (pytorch/functorch#174)

vfdev-5 committed 2 years ago

a0df7823

[functorch] Revert the compile cache (pytorch/functorch#168)

bertmaher committed 2 years ago

58aaddb2

[functorch] Python pointwise compiler implementation (pytorch/functorch#163)

bertmaher committed 2 years ago

00c28622

[functorch] [port] diagonal to new api (pytorch/functorch#165)

kshitij12345 committed 2 years ago

a80fe5ee

[functorch] Python bindings for compilation cache

bertmaher committed 2 years ago

8d6984b5

[functorch] Complete compile cache, with in-out specialization

bertmaher committed 2 years ago

726e317a

[functorch] Num arg specialized cache

bertmaher committed 2 years ago

ee21ae22

[functorch] Num arg-and-dim specialized cache for generated kernels

bertmaher committed 2 years ago

f4c5322b

[functorch] Proxies for binding compilation results to python objects

bertmaher committed 2 years ago

8f68de2e

[functorch] Class for caching compilation results

bertmaher committed 2 years ago

2305561a

[functorch] Helper to convert SpecializationKey to python object

bertmaher committed 2 years ago

cdea3fb0

[functorch] Shape-specialization key for op caching

bertmaher committed 2 years ago

f4ae9bb3

[functorch] Added argsort batching rule (pytorch/functorch#166)

vfdev-5 committed 2 years ago

a14f16e5

[functorch] {log|softmax}_backward update as per the new signature (pytorch/functorch#160)

kshitij12345 committed 2 years ago

4e3b50e6

[functorch] Support buffers in compiled_module (pytorch/functorch#147)

bertmaher committed 2 years ago

033e7109

[functorch] Added roll batching rule (pytorch/functorch#145)

vfdev-5 committed 2 years ago

8bb907b8

[functorch] made python key turn None into empty tensors

Chillee committed 2 years ago

69db88ab

[functorch] fix multi-output tests

Chillee committed 2 years ago

af5c0c1e

[functorch] fix issue with multiple outputs

Chillee committed 2 years ago

fecaf533

[functorch] plumbed partition_fn through compiled_module

Chillee committed 2 years ago

90ad6784

[functorch] Exposed default partition to users

Chillee committed 2 years ago

aa6b5a39

[functorch] renamed some variables in gen_data

Chillee committed 2 years ago

d9de3842

[functorch] Added cross batching rule (pytorch/functorch#144)

vfdev-5 committed 2 years ago

cdf866da

[functorch] List of method only pytorch operators

zou3519 committed 2 years ago

90331374

[functorch] Check in top operators list to do analysis on

zou3519 committed 2 years ago

92158729

[functorch] Fixes pytorch nightly cpu installation (pytorch/functorch#157)

vfdev-5 committed 2 years ago

ef534dac

Newer Older