strisunshinewentingwang/pytorch

Pull Requests Commits

[functorch] fix example to work well :)

Chillee committed 2 years ago

e8a5b372

[functorch] Moved to Keops example

Chillee committed 2 years ago

1a2e5385

[functorch] Added eager-mode fusion prototype

Chillee committed 2 years ago

9ca5ae86

[functorch] Added initial NNC example of eager-mode fusion

Chillee committed 2 years ago

e632a46f

[functorch] Fix "vmap: inplace arithmetic(self, *extra_args) is not possible" for linear

zou3519 committed 2 years ago

7c405f44

[functorch] Revert TensorWrapper changes to make things more sane

zou3519 committed 2 years ago

a9327ea8

[functorch] update binary_pointwise_batch_rule

zou3519 committed 2 years ago

fd7e524b

[functorch] Quick grab bag of batching rules

zou3519 committed 2 years ago

4ce294d2

[functorch] Update EXISTING_BDIM_BATCH_RULE

zou3519 committed 2 years ago

0a7954b9

[functorch] Updated variadic_bdims_batch_rule

zou3519 committed 2 years ago

3a1a59a7

[functorch] replace basic_unary_batch_rule with BASIC_UNARY_BATCH_RULE

zou3519 committed 2 years ago

745c6336

[functorch] Updated squeeze and squeeze.dim batching rules to new style and added scalar handling (pytorch/functorch#81)

laurencer committed 2 years ago

7e1d730a

[functorch] remove old python key implementation :)

Chillee committed 2 years ago

53155d3b

[functorch] Added vjpfull

Chillee committed 2 years ago

707c26d3

[functorch] updated nnc_compile to work with new python key

Chillee committed 2 years ago

a92ca843

[functorch] pythonkey refactor

Chillee committed 2 years ago

a11bdcc4

[functorch] Tensor printing

zou3519 committed 2 years ago

8107b13b

[functorch] Fix reshape failure by adding _reshape_alias batch rule

zou3519 committed 2 years ago

b3e39c19

[functorch] Add interpolate/upsample batching rules

Chillee committed 2 years ago

f6bb9acd

[functorch] Finished most of the rest of the pad batching rules (also added an existing_batch_dim_template)

Chillee committed 2 years ago

308f4775

[functorch] Added batching rules for constant_pad_nd

Chillee committed 2 years ago

a5a7245e

[functorch] updated opinfo db and fixed failing tests

Chillee committed 2 years ago

266d2ced

[functorch] Revert "We actually fixed a lot of the vjp problems"

zou3519 committed 2 years ago

ed65f0a8

[functorch] We actually fixed a lot of the vjp problems

zou3519 committed 2 years ago

22ad9d47

[functorch] Resolved some vjp failures

zou3519 committed 2 years ago

2ceba07e

[functorch] Normalize, cross_entropy OpInfos

zou3519 committed 2 years ago

5cfa5728

[functorch] OpInfo for pad

zou3519 committed 2 years ago

f8caad7f

[functorch] additional OpInfo for Conv2d

zou3519 committed 2 years ago

4028de1d

[functorch] More OpInfos

zou3519 committed 2 years ago

f28e1996

[functorch] Add `functorch_additional_op_db`

zou3519 committed 2 years ago

6b59f1ad

Newer Older