strisunshinewentingwang/pytorch

Pull Requests Commits

Sparse CUDA: rename files *.cu -> *.cpp (#63894)

IvanYashchuk committed 3 years ago

ff4569ae

[PyTorch] Reduce code size of register_prim_ops.cpp (#61494)

swolchok committed 3 years ago

8fc1064b

Adding alltoall_single collective to collective quantization API (#63154)

marjanfariborz committed 3 years ago

6a76ee04

New TLS to disable forward mode AD (#63117)

albanD committed 3 years ago

04108592

[pruner] add README to repo (#64099)

kazhou committed 3 years ago

6257f5b1

Improve `distributed.get_rank()` API docstring (#63296)

mrshenli committed 3 years ago

101a6263

Modules note v2 (#63963)

jbschlosser committed 3 years ago

196fd3ee

Detect out argument in the schema (#62755)

tugsbayasgalan committed 3 years ago

19c1b45f

[Static Runtime] Add out variant of quantized::embedding_bag_byte_prepack (#64081)

d1jang committed 3 years ago

9f1f22b9

fix resize bug (#61166)

BBuf committed 3 years ago

6ab3a210

[caffe2] fixes to allow stricter compilation flag (#64016)

Pierluigi Taddei committed 3 years ago

538c30a7

Added reference tests to ReductionOpInfo (#62900)

heitorschueroff committed 3 years ago

eca87f72

[JIT] Add aten::slice optimization (#63049)

Mike Iovine committed 3 years ago

babd4499

Add doc for nn.MultiMarginLoss (shape, example) (#63760)

cccntu committed 3 years ago

3abb6060

Refactor structured set_output in Register{DispatchKey}.cpp (#62188)

peterbell10 committed 3 years ago

a9983ac0

[bazel] GPU-support: add @local_config_cuda and @cuda (#63604)

vors committed 3 years ago

f922b58b

[OSS] Enable Metal in PyTorch MacOS nightly builds (#63718)

Hanton Yang committed 3 years ago

22d38bd1

Adds return type annotation for fork_rng function (#63724)

aswinmurali-io committed 3 years ago

a43e7a51

More robust check of whether a class is defined in torch (#64083)

gmagogsfm committed 3 years ago

ad8eddbd

[Static Runtime] Out version for fmod (#64046)

HarutMov committed 3 years ago

f2c47cf4

[Static Runtime] Manage temporary Tensors for aten::layer_norm (#64078)

d1jang committed 3 years ago

c90b3cb1

[Static Runtime] Use F14FastMap/F14FastSet (#63999)

Hao Lu committed 3 years ago

3c3bba41

[static runtime] port c2 argmin kernel (#63632)

ajyu committed 3 years ago

3f1c8094

[quant] Add support for linear_relu fusion for FP16 dynamic quant (#63826)

supriyar committed 3 years ago

294db060

[quant] Add op support for linear_relu_dynamic_fp16 (#63824)

supriyar committed 3 years ago

cec44aa5

[quant] support linear_relu_dynamic for qnnpack backend (#63820)

supriyar committed 3 years ago

975f4cca

[quant][fx] Add support for dynamic linear + relu fusion (INT8) (#63799)

supriyar committed 3 years ago

c7027f19

[torch/deploy] add torch.distributed to build (#63918)

suo committed 3 years ago

63c90ec3

Introduce the torchrun entrypoint (#64049)

cbalioglu committed 3 years ago

65e6194a

Merge script and _script_pdt API (#62420)

nikithamalgi committed 3 years ago

510d2ece

Newer Older