strisunshinewentingwang/pytorch

Pull Requests Commits

[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

[functorch] Quick attempt at hiding functional module init

zou3519 committed 2 years ago

a7f406ce

[functorch] Add batch rule for nll_loss_forward for most common cases

zou3519 committed 2 years ago

7f344c5a

[functorch] Linear batch rule (which is just a decomposition)

zou3519 committed 2 years ago

1cd0dd00

[functorch] fix build failure

zou3519 committed 2 years ago

9496ea3e

[functorch] Update some version numbers"

zou3519 committed 2 years ago

29104230

[functorch] lennard-jones example and test

zou3519 committed 2 years ago

9e6201db

[functorch] Added a way to call the slow fallback from plumbing

zou3519 committed 2 years ago

d65bb48b

[functorch] Inplace +-*/ batch rules

zou3519 committed 2 years ago

03173fad

[functorch] fix issues with passthrough variables

Chillee committed 2 years ago

8e82d5af

[functorch] Added statement if there's an empty NNC compute expression

Chillee committed 2 years ago

7644e62d

[functorch] Added some tests + fix some stupid stuff

Chillee committed 2 years ago

f2204f60

[functorch] Fix some failing tests

zou3519 committed 2 years ago

e1281fca

[functorch] Added conv2d batching rule (pytorch/functorch#10)

Chillee committed 2 years ago

29b90b4f

[functorch] log softmax backward batch rule

zou3519 committed 2 years ago

354e79af

[functorch] Fix some of the more important lint errors

zou3519 committed 2 years ago

8e0e3410

Newer Older