strisunshinewentingwang/pytorch

Pull Requests Commits

Add fuse support for batchnorm with affine=False (#45474)

sstsai-adl committed 3 years ago

2596113a

[ONNX] Update embedding_bag export (#44693)

neginraoof committed 3 years ago

6b42ca2d

[FX] Shape propagation example (#45589)

James Reed committed 3 years ago

ac9a708e

SET USE_DISTRIBUTED OFF when libuv is not installed (#45554)

gunandrose4u committed 3 years ago

ffd50b82

[c++] Distance-agnostic triplet margin loss (#45377)

Xinyu Li committed 3 years ago

c9bb9907

Add an option to DDP to take a list of parameters to ignore upfront. (#44826)

rohan-varma committed 3 years ago

181afd52

[quant] Make choose_qparams_optimized return Tensors to preserve dtype (#45530)

supriyar committed 3 years ago

c112e89c

[pytorch] Replace "blacklist" in test/test_mobile_optimizer.py (#45512)

Meghan Lele committed 3 years ago

ce9df084

add dllexport before template specialization functions for windows build (#45477)

mszhanyi committed 3 years ago

a245dd43

[quant][graphmode][fx] Support quantization for custom module (#44074)

jerryzh168 committed 3 years ago

5539066d

Revert D24010742: [pytorch][PR] Add callgrind collection to Timer

Mike Ruberry committed 3 years ago

51d0ae92

Revert D24002415: Some fixes to smooth_l1_loss

bdhirsh committed 3 years ago

6c4aa2a7

type check for torch.quantization.quantize_jit (#45548)

Rong Rong committed 3 years ago

4f392095

Update test_multi_tensor_optimizers test (#45510)

Iurii Zdebskyi committed 3 years ago

939e0389

Add whitelist for complex backward (#45461)

anjali411 committed 3 years ago

415ed434

Upgrade ReadMe document to guide user to install libuv(1.39) in conda env on Windows platform (#45553)

gunandrose4u committed 3 years ago

7e863475

Add ShuffleDataset with buffer (#45290)

ejguan committed 3 years ago

96540e91

Some fixes to smooth_l1_loss (#45532)

bdhirsh committed 3 years ago

fdbed711

Unify Transformer coder Constructors (#45515)

VinodSKumar committed 3 years ago

e02868e1

Enable PE + TE (#45546)

Krovatkin committed 3 years ago

75668237

Add callgrind collection to Timer (#44717)

Taylor Robie committed 3 years ago

9b27e092

Source code level attribution in profiler (#43898)

Ilia Cherniavskii committed 3 years ago

f5c95d5c

Fix docs for kwargs, q-z (#43589)

zasdfgbnm committed 3 years ago

c2c70999

Print tensor shapes and convolution parameters when cuDNN exception is thrown (#45023)

xwang233 committed 3 years ago

b4ba66ae

Move prim::tolist math.log and aten::cpu to lite interpreter for translation model (#45482)

pipibjc committed 3 years ago

93650a82

[TensorExpr] Change API for creating Load and Store expressions. (#45520)

Mikhail Zolotukhin committed 3 years ago

4aca63d3

Fix memory corruption when running torch.svd for complex.doubles (#45486)

malfet committed 3 years ago

772ce9ac

Fix D23995953 import.

Taylor Robie committed 3 years ago

ccad73ab

Enable transposed tensor copy for complex types (#45487)

malfet committed 3 years ago

c87ff2cb

CUDA RTX30 series support (#45489)

zasdfgbnm committed 3 years ago

0a15646e

Newer Older