strisunshinewentingwang/pytorch

Pull Requests Commits

Make pattern matcher more robust (#119876)

tugsbayasgalan committed 220 days ago

b4252d73

[dtensor] refactor sharding cost model to count for latency (#119897)

wanchaol committed 220 days ago

daf1050a

Skip test_wrap_bad if run under pytest (#115070)

Flamefire committed 220 days ago

99cb807e

Use ARMV8.2 scalar fp16<->fp32 conversion (#119895)

malfet committed 220 days ago

d833e2f2

[FSDP2] Added gradient accumulation w/o reduction (#118298)

awgu committed 220 days ago

096ebcca

[export] Log private api uses. (#119848)

zhxchen17 committed 220 days ago

8f27fde2

Deduplicate docs between global and non-global full backward hooks (#119708)

soulitzer committed 220 days ago

340b6fa9

Revert "[Inductor] Setting kernel launch and exit callbacks for inductor generated triton kernels (#119450)"

pytorchmergebot committed 220 days ago

3713103d

Fix NJT stride access in SDPA dispatcher logic (#119846)

jbschlosser committed 220 days ago

756cf291

Fix meta registration for _flash_attention_forward() [ROCm forward fix] (#119910)

jbschlosser committed 220 days ago

0560c193

[C10] Expand half unittest (#119892)

malfet committed 220 days ago

734ae20f

[DCP] Automatically set `no_dist` if distributed is unavailable (#119813)

LucasLLC committed 220 days ago

3470ab42

[CUDNN][SDPA] Experimental cuDNN Flash Attention v2 Inference (#115663)

eqy committed 220 days ago

cd380c79

Reify view_func() closures as ViewFuncs (#118404)

jbschlosser committed 220 days ago

9ec8dd24

[inductor][scheduler] Use set for origin (#119861)

anijain2305 committed 220 days ago

6b04251b

Handle aliases correctly in foreach (#119508)

mlazos committed 220 days ago

29235c70

Windows Dynamo Error Removal CI Check (#115969)

gs-olive committed 220 days ago

e0f6fa6a

Add pixel_shuffle to core aten decomps (#119899)

angelayi committed 220 days ago

9201d733

Add linux cpu test for 3.12 (#117853)

atalman committed 220 days ago

244b124b

[DTensor] Enable Adamax foreach optimizer (#119850)

wz337 committed 220 days ago

bb67a287

Fix guards for field access through properties (#119719)

aorenste committed 220 days ago

2aad3f93

[testing][inductor] Allow grad tolerance override (#119844)

amjames committed 221 days ago

7797a8c2

Improve TORCHDYNAMO_EXTENDED_DEBUG for GuardOnDataDependentSymNode (#119412)

ezyang committed 221 days ago

15f1b9f1

[ROCm] TunableOp (#114894)

jeffdaily committed 221 days ago

0e6eee3c

Change default TORCH_LOGS format to match Meta/glog standard (#119869)

ezyang committed 221 days ago

90f785dc

[dtensor] add op support for nll_loss_backward (#119256)

tianyu-l committed 221 days ago

d999222f

Add cpp stack traces to our own reruns (#119408)

albanD committed 221 days ago

47182a8f

[export] Remove references to capture_pre_autograd_graph inside test_export (#119875)

tugsbayasgalan committed 221 days ago

6cf48187

[export] Don't error if nn_module_stack doesn't contain a class (#119753)

angelayi committed 221 days ago

ee3a7bdc

[ROCm] Initial ir.Scan/aten.cumsum lowering support on ROCm (#119369)

jataylo committed 221 days ago

3e21c785

Newer Older