strisunshinewentingwang/pytorch

Pull Requests Commits

Add a command-line flag for overriding pthreadpool size (#46781)

dreiss committed 3 years ago

ad8c0e57

[JIT] fix documentation typo (#46926)

Elias Ellison committed 3 years ago

a63f391c

[Bootcamp] add CUDA kernel checks to ATen/native/cuda (#47466)

Cameron Burnett committed 3 years ago

ceb16d88

[NNC] Fix an issue with half-scalar vars coerced to float (Take 2) (#47448)

nickgg committed 3 years ago

e985503d

Batched grad for advanced indexing (index) (#47223)

zou3519 committed 3 years ago

9c8f4051

add remove_node in Partition class (#47452)

scottxu0730 committed 3 years ago

65241e36

add get_device_to_partitions_mapping (#47361)

scottxu0730 committed 3 years ago

b4b0fa63

Added CUDA support for complex input for torch.inverse (#45034)

IvanYashchuk committed 3 years ago

33acbeda

Disable unused docker-pytorch-linux-xenial-py3.6-gcc4.8 job (#47446)

janeyx99 committed 3 years ago

c2d4a5b1

[FX] get the correct error message (#47108)

Chillee committed 3 years ago

37324673

Speedup copysign for half and bfloat16 types (#47413)

malfet committed 3 years ago

eed4a57d

Revert D24649817: [pytorch][PR] Fix pickling for Tensor subclasses.

ezyang committed 3 years ago

35491412

[ONNX] Update peephole pass for prim::ListUnpack (#46264)

KsenijaS committed 3 years ago

7a599870

FixedQParamsFakeQuantize: adjust default quant_min and quant_max (#47423)

vkuzo committed 3 years ago

5977d1d8

Revert D24706475: [pytorch][PR] [NNC] Fix an issue in Cuda fusion with fp16 scalar vars coerced to float

zou3519 committed 3 years ago

745899f9

Revert D24659901: Add tests for DDP control flow models.

zou3519 committed 3 years ago

9c8078cd

__noinline__ the top level igamma cuda kernel. (#47414)

gchanan committed 3 years ago

1519c714

Fix sum batching rule, add simple clone batching rule (#47189)

zou3519 committed 3 years ago

e40a5630

Batching rules for complex view functions (#47188)

zou3519 committed 3 years ago

9a9529aa

Move igamma cuda specific code to kernel file. (#47410)

gchanan committed 3 years ago

ae374dc6

Add op benchmark for batch box cox as baseline (#47275)

ShijunK committed 3 years ago

220b3bd6

Add release note scripts (#47360)

albanD committed 3 years ago

68954fe8

Updated docs/test for dot and vdot (#47242)

heitorschueroff committed 3 years ago

a4ba018e

[quant][pyper] Add support for pruned weights in embedding_bag_byte lookup (#47329)

supriyar committed 3 years ago

d8c3b2b1

[quant] Add testing coverage for 4-bit embedding_bag sparse lookup op (#47328)

supriyar committed 3 years ago

433b55bc

Expand the test of torch.addbmm and torch.baddbmm (#47079)

zasdfgbnm committed 3 years ago

f19637e6

[Pytorch] Specialize guts of c10::optional for 32-bit scalars (#47015)

swolchok committed 3 years ago

df5b4696

[NNC] Read/Write Dependency analysis (#46952)

nickgg committed 3 years ago

0edc6a39

Fix pickling for Tensor subclasses. (#47115)

hameerabbasi committed 3 years ago

c4209f11

Add torch.overrides checks for submodules. (#47285)

hameerabbasi committed 3 years ago

60ae8475

Newer Older