strisunshinewentingwang/pytorch

Pull Requests Commits

Update on "[DDP] Support for multiple backwards"

rohan-varma committed 3 years ago

1388135f

Update base for Update on "[DDP] Support for multiple backwards"

rohan-varma committed 3 years ago

d3e9394b

[vulkan] Do not use memcmp to compare structs (#60199)

SS-JIA committed 3 years ago

ef1c107b

Revert D28833086: beef up at::_ops API

bdhirsh committed 3 years ago

6d0fb85a

Correct backend in pipe_with_ddp_test (#60123)

rohan-varma committed 3 years ago

0cbb5e15

Fix Pipe + DDP for unused parameters, static graph (#60118)

rohan-varma committed 3 years ago

acd914f0

[iOS GPU][MaskRCNN] Implement RoIAlign in Metal shaders using Sampler (#56075)

xta0 committed 3 years ago

2062cafa

beef up at::_ops API (#59115)

bdhirsh committed 3 years ago

e2129d1c

Enable GHA sharding on linux (#60124)

janeyx99 committed 3 years ago

462448f0

Run an dummy rpc._all_gather in init_rpc to avoid shutdown timeout (#59801)

mrshenli committed 3 years ago

bbedfd91

Stop warning on .names() access in max_pool2d and max_pool2d_backward (#60059)

zou3519 committed 3 years ago

ebafd2aa

Revert D29104399: Port `all` kernel to structured kernels.

bdhirsh committed 3 years ago

ef094288

Revert D29104395: Port `any` kernel to structured kernels.

bdhirsh committed 3 years ago

3ff5507f

Revert D29104398: Using meta checks for unary `torch.all` and `torch.any`.

bdhirsh committed 3 years ago

81baa7fb

Revert D29104397: Port `argmax` to structured kernels.

bdhirsh committed 3 years ago

873dac4b

Revert D29104396: Port `argmin` kernel to structured kernels.

bdhirsh committed 3 years ago

6b5e7790

[NNC] Handle int64 indices and loop bounds (#59769)

desertfire committed 3 years ago

3dc81121

[NNC] Add a dtypeToCppString virtual method in IRPrinter (#59449)

desertfire committed 3 years ago

96b3537e

PG NCCL cleanup: remove usage of completed_ in WorkNCCL copies (#59899)

agolynski committed 3 years ago

ed1da5be

Add .isort.cfg (#60119)

samestep committed 3 years ago

010f4b6f

Port `argmin` kernel to structured kernels. (#59938)

ysiraichi committed 3 years ago

226d745a

Port `argmax` to structured kernels. (#59937)

ysiraichi committed 3 years ago

6f3da4f4

Using meta checks for unary `torch.all` and `torch.any`. (#59373)

ysiraichi committed 3 years ago

c078cefa

Port `any` kernel to structured kernels. (#59372)

ysiraichi committed 3 years ago

51969836

Port `all` kernel to structured kernels. (#59371)

ysiraichi committed 3 years ago

7809494c

only runs mem leak check on master (#60023)

Rong Rong (AI Infra) committed 3 years ago

b8ab9862

Unifies OpInfo dtype tests (#60157)

Mike Ruberry committed 3 years ago

59b10036

Improved torch.einsum testing and fixed bug (#59731)

heitorschueroff committed 3 years ago

4caca7a1

[TensorExpr] Minor cleanup in TensorExprKernel::computeValue (#60041)

Mikhail Zolotukhin committed 3 years ago

eb36f67d

Revert D29132955: Pass RequestCallback to FaultyPG RPC agent

Mike Ruberry committed 3 years ago

6b171201

Older