strisunshinewentingwang/pytorch

Pull Requests Commits

[functorch] fixed dot batching rule and added vmap tests using opinfo

Chillee committed 2 years ago

0d94ae66

[functorch] Added dot/add.Scalar/mul.Scalar/etc. batching rules and added functools.wraps to grad

Chillee committed 2 years ago

918ede7a

[functorch] modified example

Chillee committed 2 years ago

ac9be17a

[functorch] More batch rules

zou3519 committed 2 years ago

c24314c0

[functorch] Grab bag of batch rules

zou3519 committed 2 years ago

d7d266f5

[functorch] fix testing warning, update readme

zou3519 committed 2 years ago

0abba43a

[functorch] Add codegen file; beef up testing

zou3519 committed 2 years ago

42843306

[functorch] Run all correctness tests

zou3519 committed 2 years ago

e8abf483

[functorch] testing, version.txt

zou3519 committed 2 years ago

20fac9da

[functorch] Update readme, fix unsqueeze batch rule

zou3519 committed 2 years ago

8277d74e

[functorch] Update README, setup.py

zou3519 committed 2 years ago

7001a2f1

[functorch] Update

zou3519 committed 2 years ago

70925e47

[functorch] misc cleanup

zou3519 committed 2 years ago

b7096ab8

[functorch] Updated some batching rules to use new API

zou3519 committed 2 years ago

ce453d44

[functorch] new_blah hack

zou3519 committed 2 years ago

985b35c2

[functorch] Various bug fixes; new BR api

zou3519 committed 2 years ago

8201dfc2

[functorch] beef testing

zou3519 committed 2 years ago

a39cb7a8

[functorch] a lot of files

zou3519 committed 2 years ago

93888a37

[functorch] Initial commit

zou3519 committed 2 years ago

608a932c

Change get module info to NOT parse flatbuffer as module. (#81819)

qihqi committed 2 years ago

c84791b4

[Vulkan] Implement select.int operator (#81771)

manuelcandales committed 2 years ago

320cc194

Improve autograd custom function docs (#81340)

soulitzer committed 2 years ago

e60f8f4f

normalize cuda device (#81739)

eellison committed 2 years ago

15fef3dc

reland of #80545 w skip rocm (#81738)

eellison committed 2 years ago

745739d8

Move away from deprecated 10.15 runners for ios tests (#81903)

janeyx99 committed 2 years ago

729077b4

Simplify torch.nn.grad by calling into aten::convolution_backward (#81839)

saketh-are committed 2 years ago

445ee562

Added generator check for parametrize and ops (#81263)

vitrioil committed 2 years ago

d4043f0d

bmm and contiguous constraints (#81527)

migeed-z committed 2 years ago

dfb2a1d1

[DataPipe] Fixes various warnings, exceptions, and clean up testing (#81833)

NivekT committed 2 years ago

428e44ff

remove myself from native_functions.yaml codeowners (#81890)

bdhirsh committed 2 years ago

e02f7f42

Newer Older