strisunshinewentingwang/pytorch

Pull Requests Commits

[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

[functorch] Revert "Ported `view` op to the new API (pytorch/functorch#137)" (pytorch/functorch#156)

vfdev-5 committed 2 years ago

b4daef77

[functorch] Ported `view` op to the new API (pytorch/functorch#137)

vfdev-5 committed 2 years ago

2f64f5fa

[functorch] Remove some commented code (pytorch/functorch#146)

bertmaher committed 2 years ago

acf0318f

[functorch] Fixed functorch to build with PyTorch master

Chillee committed 2 years ago

cee92d4e

[functorch] Update run_test.sh (pytorch/functorch#143)

vfdev-5 committed 2 years ago

321ebb58

[functorch] Added rot90 batch rule (pytorch/functorch#138)

vfdev-5 committed 2 years ago

53af441b

[functorch] Fixed unexpected failures with log_softmax (pytorch/functorch#142)

vfdev-5 committed 2 years ago

b2f72d10

[functorch] made compile_module faster

Chillee committed 2 years ago

ab71bc7b

[functorch] merged

Chillee committed 2 years ago

4b80ff5a

[functorch] Introduce compiled_module for eager compilation of modules (pytorch/functorch#133)

bertmaher committed 2 years ago

53c9fdd2

Newer Older