strisunshinewentingwang/pytorch

Pull Requests Commits

Add support for NCCL alltoall (#44374)

zasdfgbnm committed 3 years ago

44922f26

ZeroRedundancyOptimizer: an implementation of a standalone sharded optimizer wrapper (#46750)

blefaudeux committed 3 years ago

87fb3707

Add cloud-tpu-client to xla CI. (#50823)

Ailing Zhang committed 3 years ago

c3e3e606

Remove gtest_prod.h from TP agent. (#50766)

pritamdamania committed 3 years ago

be7e9845

quantization: Linear + BatchNorm1d fusion (#50748)

vkuzo committed 3 years ago

ac8e90fa

Fix replication_pad for cuda launch configuration (#50565)

xwang233 committed 3 years ago

db86dd8a

Fix indexing for overrides. (#49324)

hameerabbasi committed 3 years ago

cf1882ad

[ROCm] re-enable tests (#50691)

KyleCZH committed 3 years ago

16faabe7

Update JIT_OPT macro for easier use (#50602)

tugsbayasgalan committed 3 years ago

fbf7eec8

Enable TensorPipe's CMA channel (#50759)

lw committed 3 years ago

112a5834

[metal] Use MPSCNN kernels for binary elementwise ops

SS-JIA committed 3 years ago

c18403a6

[PyTorch] Remove CAFFE2_FB_LIMITED_MOBILE_CAPABILITY (#50385)

swolchok committed 3 years ago

1e0809db

Fix test_dispatch.py when running with TORCH_SHOW_CPP_STACKTRACES=1 (#50509)

zasdfgbnm committed 3 years ago

4f3cdd97

JIT Testing: Improve assertAutodiffNode error message (#50626)

Lilyjjo committed 3 years ago

f1c57859

Add complex autograd support and OpInfo based test for torch.addr (#50667)

anjali411 committed 3 years ago

1cc8f8a7

tools: Move sha check to else statement (#50773)

seemethere committed 3 years ago

66adfcd2

Issue #48724. Only set the CMake IMPORTED_LOCATION property in static… (#49173)

foundry-markf committed 3 years ago

e1bb4769

[WIP] JIT Static Hooks: cpp tests (#49547)

Lilyjjo committed 3 years ago

22902b92

[WIP] JIT Static Hooks: python tests (#49546)

Lilyjjo committed 3 years ago

3b88e1b0

[WIP] JIT Static Hooks: serialization logic (#49545)

Lilyjjo committed 3 years ago

0eb41e67

[WIP] JIT Static Hooks: schema checking logic (#49975)

Lilyjjo committed 3 years ago

9c494572

[WIP] JIT Static Hooks: adding hooks to class type and adding logic for hook running/compilation (#49544)

Lilyjjo committed 3 years ago

a722d28e

Revert D25958987: [pytorch][PR] Add type annotations to torch.overrides

mrshenli committed 3 years ago

1f5c3b3a

Add new backend type for Intel heterogeneous computation platform. (#49786)

chengjunlu committed 3 years ago

4a8ef452

Let TensorPipe detect peer access (#50676)

mrshenli committed 3 years ago

a3b8cbcd

Implement NumPy-like function torch.fmax() & torch.fmin() (#49312)

Kiyosora committed 3 years ago

4803eaf5

Add type annotations to torch.overrides (#48493)

guilhermeleobas committed 3 years ago

2ace4fc0

[JIT] Fix archive file extension in examples and docs (#50649)

Meghan Lele committed 3 years ago

4aea0073

[quant] Add non-fbgemm fallback implementation for embedding lookup ops (#50706)

supriyar committed 3 years ago

e0096650

[FX] Fix NoneType annotation in generated code (#50777)

James Reed committed 3 years ago

5205cc1c

Newer Older