strisunshinewentingwang/pytorch

Pull Requests Commits

Add flop formulas for sdpa (#95854)

Chillee committed 1 year ago

9a781ce3

Improve Discoverability of Inductor Optimizations (#95824)

msaroufim committed 1 year ago

7db5f8c7

Fix typo in RELEASE.md and CONTRIBUTING.md (#95965)

kiszk committed 1 year ago

7d02ecfa

Add export docs, improve asserts (#94961)

voznesenskym committed 1 year ago

ac07de4a

Don't use guardless contiguity/stride-like implementations (#95733)

ezyang committed 1 year ago

027ebca4

[EASY] Unindent some blocks (#95967)

ezyang committed 1 year ago

f8b57ba6

Fix: `validate_input_col` for partial functions (#95067)

BeckerFelix committed 1 year ago

e6f3e16d

memory viz: Add colors for categories and a legend (#90587)

zdevito committed 1 year ago

ee438425

[inductor] do not use `ceil` in `arange` ref (#95773)

nkaretnikov committed 1 year ago

c72fbf2e

[inductor] use FP div in CPP expr printer (#95698)

nkaretnikov committed 1 year ago

feffcafe

[DTensor][Shampoo] add _tenso.zero function (#95863)

minddrummer committed 1 year ago

6c061e51

[MTPG] Improve all_reduce and handle bwd thread support (#95524)

Rodrigo Kumpera committed 1 year ago

1d3c394d

[DCP] Add DCP FSDP sharded_state_dict checkpoint example to DCP .rst file (#95517)

wz337 committed 1 year ago

a7698a82

Revert "Don't use guardless contiguity/stride-like implementations (#95733)"

pytorchmergebot committed 1 year ago

4026c621

Run _nvfuser/test_torchscript serially (#95951)

clee2000 committed 1 year ago

7f5f0b36

Revert "[inductor] Add an AOT compilation mode for Inductor CPP backend (#94822)"

pytorchmergebot committed 1 year ago

879400e4

Run more tests through pytest (#95844)

clee2000 committed 1 year ago

d21577f2

Fix formatting (#95906)

Svetlana Karslioglu committed 1 year ago

004bcffc

[Quant][PT2.0] fix issues for rearranging weight observer for decomposed linear (#94296)

Xia-Weiwen committed 1 year ago

f3c25cd3

Triton kernel for bsr @ dense (#94823)

nikitaved committed 1 year ago

d809020f

Move label check failure to mergebot (#94707)

Goldspear committed 1 year ago

88e554b5

[inductor] Add an AOT compilation mode for Inductor CPP backend (#94822)

desertfire committed 1 year ago

73b66098

Inductor: fix crash when indexing an empty tensor by invalid index (#95046)

blzheng committed 1 year ago

3eb8eaa1

inductor: fix permute_linear_fusion/linear_permute_fusion has nor 'bias' keyError issue (#95930)

XiaobingSuper committed 1 year ago

d4e0d895

Fix "sandcastle_skip_if decorator name is confusing" (#95649)

kit1980 committed 1 year ago

35bf5bac

Refactor inductor collectives around base class (#95920)

wconstab committed 1 year ago

0147a408

Support Inductor collectives with wait or collective outside graph (#95893)

wconstab committed 1 year ago

92a21073

Remove pydispatcher from test since no longer needed (#95890)

wconstab committed 1 year ago

7206b5e9

[dtensor] fix experimental_op slice_scatter (#95894)

wanchaol committed 1 year ago

738beaa6

[MPS] Disallow reshape in slice (#95905)

DenisVieriu97 committed 1 year ago

304a9543

Newer Older