strisunshinewentingwang/pytorch

Pull Requests Commits

Fix issue with _checkpoint_without_reentrant

jamaliki committed 2 years ago

bc3c7a6c

[SR] Use at::DimVector in reshape_copy_out (#76473)

Mike Iovine committed 2 years ago

9e32cded

[ROCm][GHA] split 4 GPU hosts across two runners

jeffdaily committed 2 years ago

61aec161

[Reland take-2] Add JIT graph fuser for oneDNN Graph API (v0.5)

sanchitintel committed 2 years ago

4ee29d60

exclude distributed tests from windows

clee2000 committed 2 years ago

94fc92f2

Adds amax and amin references

ngimel committed 2 years ago

1c776d20

Port clamp_min and clamp_max to structured

ngimel committed 2 years ago

12279891

rohan-varma committed 2 years ago

b30c027a

Minor follow up fixes for python registration

anjali411 committed 2 years ago

3d28ab07

ns for fx: skip shadowing ops if copy subgraph is not implemented (#76663)

vkuzo committed 2 years ago

3a8752db

ns for fx: skip shadowing for torch.cat, and also for nodes with only kwargs (#76561)

vkuzo committed 2 years ago

d3e33893

[FSDP] Include buffers in `ignored_modules`

awgu committed 2 years ago

73b33de9

`functional.max_unpool`: OpInfo tests + simpler backward + forward ad + fwad over backward ad

nikitaved committed 2 years ago

33fabe9a

Go through the dispatcher in matmul_out for the 1D-1D case

lezcano committed 2 years ago

2b3b205f

lezcano committed 2 years ago

7cb7cd58

Improve derivative of QR decomposition

lezcano committed 2 years ago

1a4eea57

Sparse CSR: Fix sampled_addmm for noncontiguous inputs and fix block sparse triangular solve

IvanYashchuk committed 2 years ago

3df0140c

anjali411 committed 2 years ago

ad8386c1

[lint] preoperly suggest lintrunner when lint fails

suo committed 2 years ago

6a44e066

Revert "Add ZT fastpath for remainder.Tensor JVP"

pytorchmergebot committed 2 years ago

da15d76e

[complex32] real and imag (also remove unused real and imag kernels)

kshitij12345 committed 2 years ago

0f1618ef

[PT-D][Sharding] Clean up sharded tensor code by leverage handle_torch_function (#76824)

fduwjj committed 2 years ago

4c504dd3

Add ZT fastpath for remainder.Tensor JVP

Nikita Karetnikov committed 2 years ago

7471f614

Allow creating new libraries and defining new operators from Python

anjali411 committed 2 years ago

07f766df

Allow users to override kernels for existing C++ ops through Python

anjali411 committed 2 years ago

55f55a4c

ci: Add upload credentials for rocm periodic test

seemethere committed 2 years ago

32ae5840

[ONNX] Format ONNX python with black

justinchuby committed 2 years ago

5dd1c677

Return all overloads for an operator in _jit_get_operation

ezyang committed 2 years ago

3a6da16a

Change TensorMeta to use tensor subclass.

ezyang committed 2 years ago

4a116783

Use TorchFunctionMode to implement PrimTorch tracing context

ezyang committed 2 years ago

48eb8d6a

Newer Older