strisunshinewentingwang/pytorch

Pull Requests Commits

Support tracing native functional collective via python APIs (#119103)

yifuwang committed 220 days ago

cd08dc37

[2/N] Replace std::tie with structural binding (#119879)

cyyever committed 220 days ago

5f9b4324

Fix a bug in kernel analysis with ttir defined args (#119934)

oulgen committed 220 days ago

9ff97987

[torch.compile] Log more compilation time breakdown (#119865)

yanboliang committed 220 days ago

7f5b87c9

[RelEng] Define `BUILD_BUNDLE_PTXAS` (#119750)

malfet committed 220 days ago

516f38a1

[caffe2] Add an avx512 implementation of adagrad_update (#113289)

Orvid committed 220 days ago

a07fd51b

Alternate sharding (#119078)

clee2000 committed 221 days ago

861acda2

Make pattern matcher more robust (#119876)

tugsbayasgalan committed 221 days ago

b4252d73

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

wanchaol committed 221 days ago

daf1050a

Skip test_wrap_bad if run under pytest (#115070)

Flamefire committed 221 days ago

99cb807e

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

malfet committed 221 days ago

d833e2f2

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

awgu committed 221 days ago

096ebcca

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

zhxchen17 committed 221 days ago

8f27fde2

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

soulitzer committed 221 days ago

340b6fa9

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

pytorchmergebot committed 221 days ago

3713103d

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

jbschlosser committed 221 days ago

756cf291

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

jbschlosser committed 221 days ago

0560c193

[C10] Expand half unittest (#119892)

malfet committed 221 days ago

734ae20f

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

LucasLLC committed 221 days ago

3470ab42

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

eqy committed 221 days ago

cd380c79

Reify view_func() closures as ViewFuncs (#118404)

jbschlosser committed 221 days ago

9ec8dd24

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

anijain2305 committed 221 days ago

6b04251b

Handle aliases correctly in foreach (#119508)

mlazos committed 221 days ago

29235c70

Windows Dynamo Error Removal CI Check (#115969)

gs-olive committed 221 days ago

e0f6fa6a

Add pixel_shuffle to core aten decomps (#119899)

angelayi committed 221 days ago

9201d733

Add linux cpu test for 3.12 (#117853)

atalman committed 221 days ago

244b124b

[DTensor] Enable Adamax foreach optimizer (#119850)

wz337 committed 221 days ago

bb67a287

Fix guards for field access through properties (#119719)

aorenste committed 221 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

Newer Older