strisunshinewentingwang/pytorch

Pull Requests Commits

[Fixed] Enable Half, BFloat16, and Complex dtypes for coo-coo sparse matmul [CUDA] (#59980)

IvanYashchuk committed 3 years ago

1fec9cd7

Reduce test size for max_pool (#65336)

albanD committed 3 years ago

8bab4689

Add functional api for `nn.Module` (#61447)

Emilio Castillo committed 3 years ago

cd813f16

Use higher timeout for TSAN tests. (#65391)

pritamdamania committed 3 years ago

c245632e

OpInfo for `nn.functional.batch_norm` (#63218)

krshrimali committed 3 years ago

28bfdbb0

Add force_on_cpu test to win cuda10.2 on GHA (#65094)

janeyx99 committed 3 years ago

9afdf017

Remove orphan from cuDNN persistent note (#65160)

rodrigoberriel committed 3 years ago

00b732e9

[Static runtime] Micro-optimization pass on GetLivenessMap (#65175)

swolchok committed 3 years ago

c0eb266c

Make new_empty/new_ones/new_zeros/new_full respect subclass (#65169)

ezyang committed 3 years ago

6d7bc34b

[PyTorch] Compare Type pointers before calling operator== in EqualNode (#65352)

swolchok committed 3 years ago

04a5e45a

Fix ENABLE_RECORD_KERNEL_FUNCTION_DTYPE build (#65370)

ezyang committed 3 years ago

88232b4c

THCTensor cleanup (#65369)

ngimel committed 3 years ago

eb4fb1ed

[PT/ShardedTensor]Allow zero size local shard (#65007)

xing-liu committed 3 years ago

600df802

OpInfo: nn.functional.conv2d (#65233)

kshitij12345 committed 3 years ago

7f6580a8

[JIT] Re-land "Add aten::slice optimization" (#65341)

Mike Iovine committed 3 years ago

9324181d

[nn] TripletMarginLoss and PairwiseDistance : no batch dim (#64882)

kshitij12345 committed 3 years ago

9c23f6eb

correlate forward and backward op (#62553)

gaoteng-git committed 3 years ago

d35ee431

[docs] Remove .data from some docs (#65358)

rodrigoberriel committed 3 years ago

f0ada4bd

Adds keyword only args to gradcheck (#65290)

benjaminr committed 3 years ago

daa50f1e

[PyTorch Edge] Backport function for defaults args with out args, flag on (#63651)

cccclai committed 3 years ago

880098a7

[JIT] Delete obsolete message: or if you absolutely have to, use c10::impl::GenericDict(c10::impl::deprecatedUntypedDict()) (#65164)

pbelevich committed 3 years ago

5826d207

[JIT] Support device as Dict key (#65079)

pbelevich committed 3 years ago

19a10638

Reduce PyTorch warnings: Cast fix xplat/caffe2/aten/src/ATen/core/DeprecatedTypeProperties.h (#65031)

Amr Elshennawy committed 3 years ago

512834b6

Basic implementation of ShardedLinear using ShardedTensor. (#64128)

pritamdamania committed 3 years ago

0dc98728

Track peak memory usage (#65157)

driazati committed 3 years ago

257a18d9

Fix logic to determine master vs PR (#65155)

driazati committed 3 years ago

58909395

[quant] Add fp32/fp16 zero_point support for CPU fakeQuant (#65055)

b-koopman committed 3 years ago

60915eb8

[PyPer] copy-free freeze_module (#65118)

Hao Lu committed 3 years ago

ce101fed

Reduce PyTorch warnings: Cast fix xplat/caffe2/c10/core/TensorOptions.h (#65030)

Amr Elshennawy committed 3 years ago

ca649851

[iOS] Zero out NSError to avoid heap corruptions for the OSS builds (#65355)

xta0 committed 3 years ago

2465a103

Newer Older