strisunshinewentingwang/pytorch

Pull Requests Commits

[functorch] [port] `view` to new api (pytorch/functorch#164)

kshitij12345 committed 2 years ago

0d3a029b

[functorch] add aminmax batching rule (pytorch/functorch#180)

kshitij12345 committed 2 years ago

14a304a0

[functorch] no_grad fix (pytorch/functorch#179)

zou3519 committed 2 years ago

660b181e

[functorch] add cosine_similairity batching rule (pytorch/functorch#171)

kshitij12345 committed 2 years ago

4c417e1b

[functorch] Ensure WithoutTop doesn't get rid of mode metadata

zou3519 committed 2 years ago

1c0efffc

[functorch] Some test cases for no_grad

zou3519 committed 2 years ago

c5c89b38

[functorch] [port] permute to new api (pytorch/functorch#172)

kshitij12345 committed 2 years ago

4d6abe97

[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

Newer Older