strisunshinewentingwang/pytorch

Pull Requests Commits

clee2000 committed 2 years ago

fe5135b5

clee2000 committed 2 years ago

a2fef8bd

Limit world size in test_fsdp_pure_fp16 (#85957)

Flamefire committed 2 years ago

9ba1630b

[1.13] Mention optim_input future BC breakage (#85963)

rohan-varma committed 2 years ago

3a13c849

Clone symint on set_sizes_and_strides (#85878)

wconstab committed 2 years ago

d003757a

Revert "Disallow saved tensor hooks in functorch transforms (#85829)"

pytorchmergebot committed 2 years ago

24adadd4

Revert "Add mechanism to disable the "saved tensors hooks" feature (#85553)"

pytorchmergebot committed 2 years ago

801818f9

Extend collate function that can register collate functions to handle specific types (#85748)

ejguan committed 2 years ago

b13b10a8

Add a way to skip lowering to nvprims (#85811)

IvanYashchuk committed 2 years ago

b00a5359

Implement col2im decomposition and fix im2col and add a few preconditions (#85541)

lezcano committed 2 years ago

787028ca

Adopt ncclRemoteError (#85887)

kwen2501 committed 2 years ago

1f38abb5

[ONNX] Initial version of diagnostics infrastructure. (#85107)

BowenBao committed 2 years ago

8f4edf1e

Update trunk CUDA-10.2 to CUDA-11.7 (#85943)

malfet committed 2 years ago

dab1c7c3

Add reduce_scatter_tensor in place of _reduce_scatter_base (#85867)

kwen2501 committed 2 years ago

ade1c196

[ONNX] Rename 'sarif_om' to 'sarif' (#85918)

BowenBao committed 2 years ago

33401ee8

[ONNX] Add type annotation for SARIF attributes (#85898)

BowenBao committed 2 years ago

6bb0a36d

[ONNX] Migrate SARIF from attr to dataclasses (#85651)

BowenBao committed 2 years ago

e9b254a0

[ONNX] Introduce SARIF (#85428)

BowenBao committed 2 years ago

91667d1d

Refactor distribuetd to use absolute header path (#85780)

minsii committed 2 years ago

1ad0048b

[Profiler] Extend ID assignment to allocations and frees (#85719)

Taylor Robie committed 2 years ago

0b0ce72b

Hotfix for S298125 (#85814)

ezyang committed 2 years ago

95681929

Revert "Refactor distribuetd to use absolute header path (#85780)"

pytorchmergebot committed 2 years ago

a50d8864

Refactor distribuetd to use absolute header path (#85780)

minsii committed 2 years ago

66808271

[MPS] Handle scalar input for scatter and gather (#85842)

abhudev committed 2 years ago

81b366a9

[MPS] Handle output shape for empty input in binary ops (#85836)

abhudev committed 2 years ago

62a4fd79

Make launch check exit code depend on results (#85886)

r-barnes committed 2 years ago

ae93a4dc

[c10d] Reorder macros so they are defined before getting used (#85850)

drej82 committed 2 years ago

dde43d08

[ONNX] Remove protocol dataclass (#85916)

justinchuby committed 2 years ago

9009393f

[MPS] Add support for aten::masked_select on mps (#119) (#85818)

DenisVieriu97 committed 2 years ago

6a14fcb9

Add mask_type=2 to masked_softmax for when mask.size() == input.size() (#85915)

george-qi committed 2 years ago

85258ec1

Older