strisunshinewentingwang/pytorch

Pull Requests Commits

Implement topk with sort for some cases (#68632)

zasdfgbnm committed 2 years ago

95f4cd0b

Fix retry on connect failure decorator (#68600)

rohan-varma committed 2 years ago

e554d8b8

Make AOT compiler generic (#68637)

priyaramani committed 2 years ago

8e51381b

Rename shard_lengths to shard_sizes to be more inline with Tensor sizes. (#66464)

pritamdamania committed 2 years ago

c41d8290

Strided masked log_softmax. (#68461)

pearu committed 2 years ago

af564e73

Fix nanmedian result using more CUDA memory than necessary (#68591)

peterbell10 committed 2 years ago

578507cb

[fx2trt][easy] Replace all network.add_activation() call with helper function (#68676)

842974287 committed 2 years ago

6cca14d0

[torchelastic][1/n] Fix `caffe2.test.distributed.launcher.api_test` flaky tests (#68624)

aivanou committed 2 years ago

37edb748

[quant][graphmode][fx] Support input_quantized_idxs and output_quantized_idxs in the new convert (#68042)

jerryzh168 committed 2 years ago

a545a409

Remove doubly nested anonymous namespace (#68555)

peterbell10 committed 2 years ago

993b7a20

Add vectorized Jacobian and Hessian computation with forward AD (#67041)

soulitzer committed 2 years ago

5456d8c8

Add forward AD support for miscellanous operators (#67820)

soulitzer committed 2 years ago

7bb401a4

Add OpInfo test and fix a couple cases (#66294)

soulitzer committed 2 years ago

e358c49a

Add internal assert for tangent layout mismatch for view ops (#66293)

soulitzer committed 2 years ago

21d203b5

Address case when layout of tangent is not same as base (#66292)

soulitzer committed 2 years ago

2455cc2a

Support cuda 11.5: install magma for cuda in conda (#68665)

atalman committed 2 years ago

bbe2aae8

[reland] Fix flaky test_nccl_timeout (#68544)

rohan-varma committed 2 years ago

183dcdf5

[quant][trt] Add support for torch.addmm in TensorRT (#67537)

jerryzh168 committed 2 years ago

875ba3dd

[SR] Add utility class to determine tensor ranges (#68284)

Mike Iovine committed 2 years ago

ee4cfaa2

[quant][graphmode][fx] Add support for weight and bias dtype in backend_config_dict (#68602)

jerryzh168 committed 2 years ago

a6d862c5

[ROCm] Use hipCUB/rocPRIM scan algorithms for large index support (#68487)

jaglinux committed 2 years ago

da4a95c7

Allow fuse unsqueeze cat sum with multiple input (#68650)

Shirong Wu committed 2 years ago

5880a2f1

Masked normalization infrastructure and strided masked softmax (#68333)

pearu committed 2 years ago

2cab77f8

add support for None in assert_close (#67795)

pmeier committed 2 years ago

f99f5ee0

refactor assert_close to be more modular (#67794)

pmeier committed 2 years ago

0809553c

[android] Lite interpreter naming for android nightly publishing (#68651)

IvanKobzarev committed 2 years ago

f74779e4

Add OpInfos for parcel Elementwise Binary II (#68085)

saketh-are committed 2 years ago

4bcff473

[quant][embedding qat] Add benchmarks for QAT Embedding+EmbeddingBag (#66560)

b-koopman committed 2 years ago

c2c859bd

[libkineto] Refactor 4/n: Simplify activity logger step 2/3 (#68329)

gdankel committed 2 years ago

f82f14de

[Profiler] Add missing guards (#65812)

gdankel committed 2 years ago

18312313

Newer Older