strisunshinewentingwang/pytorch

Pull Requests Commits

Use a no-warning registry for TensorPipe backends (#60457)

lw committed 3 years ago

06e6d631

[jit] Added a pass to transform aten::cat ops to prim::Concat op with variable number of inputs (#59881)

navahgar committed 3 years ago

d3a8505e

[jit] Added a new operator for concat that takes in variadic parameters (#59880)

navahgar committed 3 years ago

c35a3dd6

[special] add zeta (#59623)

kshitij12345 committed 3 years ago

dfd2edc0

Support `torch.bitwise_{left/right}_shift` and `__rlshift__`, `__rrshift__` (#59544)

asi1024 committed 3 years ago

26cdec6c

Run dist_autograd backward RPCs on appropriate CUDA streams. (#60606)

pritamdamania committed 3 years ago

b82453cb

[CUDA graphs][BC-breaking] Removes post-backward syncs on default stream (#60421)

mcarilli committed 3 years ago

675cea1a

[caffe2] update db::Transaction::Put() to accept the value by rvalue reference (#60208)

simpkins committed 3 years ago

00896cb9

[caffe2] update the BlobSerializer acceptor to allow moving in the data (#60207)

simpkins committed 3 years ago

b09c0b65

add support for sparse tensors in `torch.testing.assert_close` (#58844)

pmeier committed 3 years ago

6ea22672

[Model Averaging] Periodic model averager (#60320)

Yi Wang committed 3 years ago

80f40b17

DOC Improves input and target docstring for loss functions (#60553)

thomasjpfan committed 3 years ago

4e51503b

DOC Describes parameters/buffers registered as None in load_state_dict (#60549)

thomasjpfan committed 3 years ago

6d1b4642

[Static Runtime] Fix bugs in static_runtime::to_copy (#60503)

Hao Lu committed 3 years ago

1e31d26b

[Static Runtime] Test for dynamic shapes in SR unit tests (#60579)

Hao Lu committed 3 years ago

d200e9de

Migrates nll_loss_forward from TH to Aten (CUDA) (#60097)

thomasjpfan committed 3 years ago

99b64116

Convert floating-point constants to T in Bessel functions (#59416)

xuhdev committed 3 years ago

ef84bcfe

[Testing] Adding reference tests to `OpInfo` class (#59369)

krshrimali committed 3 years ago

08020220

manual revert of 57575 (#60572)

Krovatkin committed 3 years ago

236d3afd

Use `accscalar_t` for CUDA add/sub with Tensor and Scalar (#60454)

crcrpar committed 3 years ago

9e773ea7

[torch][segment_reduce] Add support for sum and min reductions (#60379)

Serhat Yilmaz committed 3 years ago

af66824c

To add Rectified Adam Algorithm to Optimizers (#58968)

iramazanli committed 3 years ago

63219f1f

[torch/distributed.elastic] Fix utils.distributed_test.test_create_store_timeout_on_server to be dual-stack ip compatible (#60558)

Kiuk Chung committed 3 years ago

5a2f41a2

[nnc] Merge inconsistent profiling information (#60510)

bertmaher committed 3 years ago

1a0058f5

[iOS GPU] Add tests for RoIAlign (#60595)

xta0 committed 3 years ago

b5b42d4c

[quant][fx][fix] QAT with object_type in qconfig (#60555)

supriyar committed 3 years ago

1120a1b9

[nnc] Add LoopNest::getLoopAt to retrieve a specified inner For-stmt (#60569)

huiguoo committed 3 years ago

d867340c

[NNC] Add tile transformation in loopnest (fixed #52785) (#57758)

huiguoo committed 3 years ago

c0d08dc1

[Model Averaging] Provide a util function for model averaging (#60303)

Yi Wang committed 3 years ago

aeea5bf4

Fix ZeRO sort to be by numel (#60556)

awgu committed 3 years ago

b770c4b6

Newer Older