strisunshinewentingwang/pytorch

Pull Requests Commits

[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

[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

Newer Older