strisunshinewentingwang/pytorch

Pull Requests Commits

Use ReduceScatter for AllGather.backward only for NCCL backend

SherlockNoMad committed 3 years ago

7ed76522

Use All2All for AllGather.backward in gloo backend

SherlockNoMad committed 3 years ago

21165570

construct a non-contiguous

SherlockNoMad committed 3 years ago

3701619e

AllGather, ReduceScatter, Gather to support noncontiguous inputs

SherlockNoMad committed 3 years ago

833875f5

Fix torch.distributed.nn.functional.all_gather

SherlockNoMad committed 3 years ago

bc05aad4

Remove 11.5 experimental builds now that we have 11.6

janeyx99 committed 3 years ago

5d059d20

[FSDP][Easy] `named_parameters()`, `named_buffers()` refactor

Andrew Gu committed 3 years ago

fa6e3cf0

[pytorch][require export] Don't require the repo to be cloned

bigfootjon committed 3 years ago

01c8ac3b

[SR] Add a copy variant for fused_split_and_squeeze

mikeiovine committed 3 years ago

98b4a410

ns for fx: reenable tests disabled by #62608

vkuzo committed 3 years ago

63c6209d

ns for fx: fix bug to enable again on torchvision models

vkuzo committed 3 years ago

f1f185f6

ns for fx: fix issue with shadowing nodes of unknown dtype

vkuzo committed 3 years ago

ae321042

[ONNX] Support aminmax

BowenBao committed 3 years ago

f78e0fc9

[ONNX] Support dynamic scale & zero_point for fake_quantize_per_tensor_affine

BowenBao committed 3 years ago

6305e572

update codeowner for public API

albanD committed 3 years ago

d4cce305

Revert "remove fp16 support from cpu linalg functions"

pytorchmergebot committed 3 years ago

495c5aeb

remove fp16 support from cpu linalg functions

Natalia Gimelshein committed 3 years ago

de18c28a

Relanding shape cache (75400) (#75710)

Krovatkin committed 3 years ago

ce842f43

Revert "Relanding shape cache (75400)"

pytorchmergebot committed 3 years ago

db180109

[lint] fixes to mypy linter

suo committed 3 years ago

fe1e6de7

Add checks for public and private API

anjali411 committed 3 years ago

d6e6061b

baby steps on patching inf/nan behavior & aten::amin support in nvfuser

jjsjann123 committed 3 years ago

692ebc8d

Fix: Make `nn.init.orthogonal_` no-op for empty input

jaketae committed 3 years ago

b5b296c4

[JIT] nvfuser test - use only major & minor versions

davidberard98 committed 3 years ago

790cc8f2

Add forward AD codegen support for single formula returning multiple outputs

soulitzer committed 3 years ago

d7b29f3e

Add forward AD support for norm, dist, F.pairwise_dist, F.normalize

soulitzer committed 3 years ago

8721abc4

Revert "Remove histogramdd functional wrapper"

pytorchmergebot committed 3 years ago

3471b0eb

Do not ignore lazy/generated/README.me

malfet committed 3 years ago

51666ff1

fix PostLocalSGDOptimizer and ModelAverager average bug

Haijunlv committed 3 years ago

08f3b958

Relanding shape cache (75400)

Krovatkin committed 3 years ago

89486821

Older