strisunshinewentingwang/pytorch

Pull Requests Commits

[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

[functorch] Some more batch rules for pointwise ops

zou3519 committed 2 years ago

d5069519

[functorch] batch rules for torch.special unary ops

zou3519 committed 2 years ago

983a43cf

[functorch] Quick grab bag of batch rules

zou3519 committed 2 years ago

1b78cae7

[functorch] Fix CI; for real this time

zou3519 committed 2 years ago

6ca3e96e

[functorch] Fix ci

zou3519 committed 2 years ago

2e7ddb7a

Newer Older