strisunshinewentingwang/pytorch

Pull Requests Commits

[aot] disable inference view tracking

Valentine233 committed 2 years ago

072a6763

nn.EmbeddingBag bound check (#96022)

xw285cornell committed 2 years ago

cf732053

[MPS] Fix the failure with ReplicatePad3D (#96988)

razarmehr committed 2 years ago

50beab29

Revert "[PTD][Checkpoint] Add checkpointing support for DTensor submesh (#96802)"

pytorchmergebot committed 2 years ago

417e7bc0

Stop using my channel for 3.11 builds (#96973)

malfet committed 2 years ago

c9135e44

record caller frame instead of function frame (#96882)

avikchaudhuri committed 2 years ago

e4e761b2

Expose Stream Recording Apis in python (#96384)

eellison committed 2 years ago

ea741508

addmv decomp #2 (#96264)

cybershiptrooper committed 2 years ago

b01d6f2c

vmap support for torch.tril and torch.triu (#94287)

isdanni committed 2 years ago

5842e5c1

[PTD][Checkpoint] Add checkpointing support for DTensor submesh (#96802)

wz337 committed 2 years ago

cfa6b52e

Add masked_grad kw argument to to_dense (#96095)

pearu committed 2 years ago

2abcafcf

Retry brew and gem installation in trunk ios workflow (#96970)

huydhn committed 2 years ago

9d80969f

[FSDP] Reduce CPU overhead (#96958)

Andrew Gu committed 2 years ago

b02fd701

[inductor] Refactor memory management code in wrapper codegen (#96768)

desertfire committed 2 years ago

931a4913

Passing LinearPackedParamBase Capsule as a saved_data to backward stage (#96269)

kwanghoon-meta committed 2 years ago

3f409065

[DTensor] Fix DeviceMesh (#96861)

shintaro-iwasaki committed 2 years ago

397fb276

Cache the transformers model used in ONNX test (#96793)

huydhn committed 2 years ago

6718e3ca

Back out "Fixing a bug where allocating a 4GB block results in using 8GB of memory (#95827)" (#96796)

jaewonlee-fb committed 2 years ago

aeb3db8a

Enable TestTorchbind on Windows (#96507)

mantaionut committed 2 years ago

0eb9e01c

[MPS] LSTM grad_y missing fix (#96601)

alexdremov committed 2 years ago

62eb7a2e

Turn off split reductions for dynamic shapes (#96850)

ezyang committed 2 years ago

b249b44b

[primTorch] handle out in `sort` meta function (#96719)

nkaretnikov committed 2 years ago

bf08d138

[CI] Revert https://github.com/pytorch/pytorch/pull/96195 (#96897)

desertfire committed 2 years ago

577d930c

Add xpu device type to torch dispatch overload (#96849)

guangyey committed 2 years ago

8187c0de

Back out "Add _int_mm to expose cuBLAS int8@int8 -> int32 matmul (#94339)" (#96885)

cpuhrsch committed 2 years ago

0a53c962

fix random output issue on index_select when src is scalar and index is empty (#96408)

mingfeima committed 2 years ago

06054d7d

[TP][DTensor Op] Enable Embedding op for DTensor (#96702)

fduwjj committed 2 years ago

3405ac8a

fix flop formulas for sdpa (#96690)

Chillee committed 2 years ago

44c9ecad

Remove pytest block list (#96698)

clee2000 committed 2 years ago

8c2341c1

[memory debugging] Extract frame information from inductor (#95753)

zdevito committed 2 years ago

3162f717

Older