strisunshinewentingwang/pytorch

Pull Requests Commits

[reland] Early terminate when CUDA assert were thrown (#49799)

Rong Rong (AI Infra) committed 3 years ago

e6a21559

[numpy] `torch.erfinv`: promote integer inputs to float (#49155)

kshitij12345 committed 3 years ago

3f4b98d5

[pt][quant] Make the CUDA fake quantize logic consistent with CPU fake quantize logic (#49808)

jianyuh committed 3 years ago

4d611093

removes more unused THC functions (#49788)

ngimel committed 3 years ago

e1631729

Improve docs for scatter and gather functions (#49679)

rgommers committed 3 years ago

d99a0c3b

Mod lists to neutral+descriptive terms in caffe2/docs (#49803)

r-barnes committed 3 years ago

b3387139

Added linalg.inv (#48261)

IvanYashchuk committed 3 years ago

8554b58f

Preserve memory format in qconv op (#49533)

Akshit Khurana committed 3 years ago

370350c7

[lint doc] how to fix flake errors if pre-commit hook wasn't there (#49345)

stas00 committed 3 years ago

5171bd94

[Gradient Compression] Directly let world_size = group_to_use.size() (#49715)

Yi Wang committed 3 years ago

55b431b1

[Gradient Compression] Explicitly restrict the scope of torch.cuda.synchronize to the current device (#49711)

Yi Wang committed 3 years ago

88c33ff8

torch.utils.checkpoint.checkpoint + torch.cuda.amp (#49757)

mcarilli committed 3 years ago

ee271047

[quant][graphmode][fx] Change standalone module api (#49719)

jerryzh168 committed 3 years ago

f474ffa1

[Gradient Compression] Change wait() to value() in some callbacks of PowerSGD communication hook (#49709)

Yi Wang committed 3 years ago

af1b636b

Add PixelUnshuffle (#49334)

jbschlosser committed 3 years ago

68d438c9

[numpy] `torch.angle`: promote integer inputs to float (#49163)

kshitij12345 committed 3 years ago

461aafe3

Remove unused six code for Python 2/3 compatibility (#48077)

skyline75489 committed 3 years ago

46b83212

Revert D25623219: [pytorch][PR] early terminate when CUDA assert were thrown

ngimel committed 3 years ago

abacf270

Skip None submodule during JIT-tracing (#49765)

Will Feng (DPER) committed 3 years ago

010b9c52

[lint] Apply whitespace linter to all gradle files

dulinriley committed 3 years ago

62f9b03b

add type annotations to torch.nn.parallel._functions (#49687)

guilhermeleobas committed 3 years ago

27f0dd36

fx quant: improve types on convert (#49688)

vkuzo committed 3 years ago

de07d076

fx quant: do not observe bias on F.linear (#49628)

vkuzo committed 3 years ago

19f972b6

fx quant: do not observe bias on F.conv (#49623)

vkuzo committed 3 years ago

c3a7591c

Update `is_floating_point()` docs to mention bfloat16 (#49611)

TylerADavis committed 3 years ago

b4141232

[FX] Try to make it more clear that _update_args_kwargs should not be called (#49745)

James Reed committed 3 years ago

67d0c182

[numpy] Add `torch.xlogy` (#48777)

kshitij12345 committed 3 years ago

27804009

early terminate when CUDA assert were thrown (#49527)

Rong Rong (AI Infra) committed 3 years ago

be091600

Update NNPACK (#49749)

malfet committed 3 years ago

9b6fb856

only upload s3 stats on master, nightly, and release branch (#49645)

Rong Rong (AI Infra) committed 3 years ago

6f9532dd

Newer Older