strisunshinewentingwang/pytorch

Pull Requests Commits

[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

[functorch] Added a simple function example

Chillee committed 2 years ago

ac017463

[functorch] Added nnc compilation stuff to functorch

Chillee committed 2 years ago

25453d61

[functorch] Migrate mm and mv batch rules from old style to new style

zou3519 committed 2 years ago

4b0d62a2

[functorch] reshape_dim_into and reshape_dim_outof helpers

zou3519 committed 2 years ago

9bdd9cee

[functorch] Delete some redundant code

zou3519 committed 2 years ago

0c5755fc

[functorch] Added batching rules for some in-place pointwise ops

zou3519 committed 2 years ago

58ee527d

[functorch] Refactory UnaryOps

zou3519 committed 2 years ago

66918c4a

[functorch] Batching rules for comparison ops

zou3519 committed 2 years ago

cb51b16e

[functorch] vjp now supports pytree inputs and outputs

zou3519 committed 2 years ago

f92fbeef

[functorch] Fix vjp and grad for unrelated outputs

zou3519 committed 2 years ago

9490aa0b

[functorch] link colab with install instructions

zou3519 committed 2 years ago

ec99d21d

[functorch] Use functorch._C._set_vmap_fallback_warning_enabled

zou3519 committed 2 years ago

9a812032

[functorch] Beef up make_functional docstring, update some examples

zou3519 committed 2 years ago

15ab42ce

[functorch] fixed some python key tracing issues

Chillee committed 2 years ago

6ecf169a

Newer Older