strisunshinewentingwang/pytorch

Pull Requests Commits

[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

[functorch] switch from TORCH_INTERNAL_ASSERT to runtime_error

Chillee committed 2 years ago

69e7dccc

[functorch] test/common.py -> test/common_utils.py

zou3519 committed 2 years ago

f7662a21

[functorch] Added aten::diag and aten::rsub.Scalar batching rules

Chillee committed 2 years ago

31a3d45c

[functorch] Switch PythonTensorImpl to using custom contiguity policy

zou3519 committed 2 years ago

2ddf7b0f

[functorch] Switch to overriding is_contiguous_custom

zou3519 committed 2 years ago

45a4dda3

[functorch] Fix clamp_min / clamp_max

zou3519 committed 2 years ago

38eb3119

[functorch] Roll back clamp_min / clamp_max change

zou3519 committed 2 years ago

bb701ef5

[functorch] Switched over to pytorch core's tree_map

Chillee committed 2 years ago

256099c1

[functorch] Removed WrapModule

Chillee committed 2 years ago

66081519

[functorch] Batching rules for: threshold_backward, clamp_min, clamp_max

zou3519 committed 2 years ago

f0f15bc8

[functorch] Update README's torch nightly version

zou3519 committed 2 years ago

d9d5a52a

[functorch] Parameterized testing

zou3519 committed 2 years ago

5e90f1e6

[functorch] removed unnecessary code left in

Chillee committed 2 years ago

647ce0ab

[functorch] fix oversight with reusing the same storage

Chillee committed 2 years ago

69ae2b6d

[functorch] updated README

Chillee committed 2 years ago

26d2ab7e

Newer Older