strisunshinewentingwang/pytorch

Pull Requests Commits

[functorch] move some test info around

Chillee committed 2 years ago

51408b2b

[functorch] Fixed oversight in vmap tests, fixed bug in sum batching rule, and added var batching rule

Chillee committed 2 years ago

180214dd

[functorch] add lowering for triangular_solve + getitem

Chillee committed 2 years ago

1279ee42

[functorch] Added get_ops and made lowering work for tuple outputs

Chillee committed 2 years ago

bfcaed04

[functorch] Add some citations

zou3519 committed 2 years ago

6877541c

[functorch] Citations

zou3519 committed 2 years ago

b627d351

[functorch] Add vjpvjp tests, fix pytorch/functorch#44

zou3519 committed 2 years ago

c7759445

[functorch] Added code of conduct + contributing

Chillee committed 2 years ago

5ead88c7

[functorch] Added a reference to the license in the README

Chillee committed 2 years ago

06a72286

[functorch] Add LICENSE headers to code files

zou3519 committed 2 years ago

c446037a

[functorch] Refactor vjp testing

zou3519 committed 2 years ago

1a3a2cf1

[functorch] Added error checking to vjp, also added opinfo tests for vjp

zou3519 committed 2 years ago

dd2d217a

[functorch] Fix incorrect vjp semantics

zou3519 committed 2 years ago

a49af32a

[functorch] Beef up grad testing, eliminate some false errors

zou3519 committed 2 years ago

92a4886a

[functorch] Enable colors in build log

zou3519 committed 2 years ago

7ddbbc39

[functorch] Add opinfo based testing for grad transform

zou3519 committed 2 years ago

66ee9e96

[functorch] Kwarg support for grad

zou3519 committed 2 years ago

74c7c736

[functorch] cleaned up logic

Chillee committed 2 years ago

2f3f44c3

[functorch] removed extraneous pdb

Chillee committed 2 years ago

a92f492b

[functorch] made tensors const references - fixes pytorch/functorch#38

Chillee committed 2 years ago

0e3a2b2d

[functorch] Make the tensor wrap/unwrap logic aware of Tensor?[]

zou3519 committed 2 years ago

ea85c20c

[functorch] fix some compile warnings

Chillee committed 2 years ago

fc5c632a

[functorch] Fix being unable to call .backward() after vmap

zou3519 committed 2 years ago

a24930ba

[functorch] ported sumdim batch rule over and added argmax

Chillee committed 2 years ago

bf1df4f3

[functorch] fix vmap tests

Chillee committed 2 years ago

74d02507

[functorch] Add mode dispatch stack (pytorch/functorch#34)

Chillee committed 2 years ago

fbc76eb6

[functorch] Added wraps to grad_and_value

Chillee committed 2 years ago

45d42283

[functorch] nll_loss_backward batch rule for some cases

zou3519 committed 2 years ago

470ecce6

[functorch] Added a MKLDNN decomposition and new_ones overload

Chillee committed 2 years ago

221bdfba

[functorch] added reshape lowerings to nnc

Chillee committed 2 years ago

6b1cc7f4

Newer Older