strisunshinewentingwang/pytorch

Pull Requests Commits

nn.EmbeddingBag bound check (#96022)

xw285cornell committed 1 year ago

cf732053

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

razarmehr committed 1 year ago

50beab29

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

pytorchmergebot committed 1 year ago

417e7bc0

Stop using my channel for 3.11 builds (#96973)

malfet committed 1 year ago

c9135e44

record caller frame instead of function frame (#96882)

avikchaudhuri committed 1 year ago

e4e761b2

Expose Stream Recording Apis in python (#96384)

eellison committed 1 year ago

ea741508

addmv decomp #2 (#96264)

cybershiptrooper committed 1 year ago

b01d6f2c

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

isdanni committed 1 year ago

5842e5c1

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

wz337 committed 1 year ago

cfa6b52e

Add masked_grad kw argument to to_dense (#96095)

pearu committed 1 year ago

2abcafcf

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

huydhn committed 1 year ago

9d80969f

[FSDP] Reduce CPU overhead (#96958)

awgu committed 1 year ago

b02fd701

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

desertfire committed 1 year ago

931a4913

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

kwanghoon-meta committed 1 year ago

3f409065

[DTensor] Fix DeviceMesh (#96861)

shintaro-iwasaki committed 1 year ago

397fb276

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

huydhn committed 1 year ago

6718e3ca

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

jaewonlee-fb committed 1 year ago

aeb3db8a

Enable TestTorchbind on Windows (#96507)

mantaionut committed 1 year ago

0eb9e01c

[MPS] LSTM grad_y missing fix (#96601)

alexdremov committed 1 year ago

62eb7a2e

Turn off split reductions for dynamic shapes (#96850)

ezyang committed 1 year ago

b249b44b

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

nkaretnikov committed 1 year ago

bf08d138

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

desertfire committed 1 year ago

577d930c

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

guangyey committed 1 year ago

8187c0de

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

cpuhrsch committed 1 year ago

0a53c962

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

mingfeima committed 1 year ago

06054d7d

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

fduwjj committed 1 year ago

3405ac8a

fix flop formulas for sdpa (#96690)

Chillee committed 1 year ago

44c9ecad

Remove pytest block list (#96698)

clee2000 committed 1 year ago

8c2341c1

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

zdevito committed 1 year ago

3162f717

Revert "Revert "[memory profiling] add a facility to gather combined C++/Python/TorchScript stack traces. (#95541)"" (#96878)

zdevito committed 1 year ago

e74f70d2

Newer Older