strisunshinewentingwang/pytorch

Pull Requests Commits

[dtensor] move early return check into redistribute autograd function (#121653)

wanchaol committed 191 days ago

a26480a4

Refactor release only changes to two step execution (#121728)

atalman committed 191 days ago

00a53b58

[dynamo] Delete record replay tests as they are not maintained (#121705)

anijain2305 committed 191 days ago

4e63d906

[dynamo] Measure Dynamo cache latency lookup (#121604)

anijain2305 committed 191 days ago

cd1751b1

[dynamo][guards-cpp-refactor] Directly call root guard manager in eval_frame (#121622)

anijain2305 committed 191 days ago

22489bfe

[dynamo][guards-cpp-refactor] Simplify DYNAMIC_INDICES guard (#121614)

anijain2305 committed 191 days ago

2348e8e4

Revert "[DCP] Makes fsspec public (#121508)"

pytorchmergebot committed 191 days ago

0398dc9e

Restore timestamps on C++ logs without glog (#121384)

ezyang committed 191 days ago

b84f94f6

[caffe2] replace refernces to np.asscalar (#121332) (#121545)

igorsugak committed 191 days ago

704e1530

[export] Update error message for set_grad (#121666)

angelayi committed 191 days ago

d1715c3a

[dynamo] Tweak naming for module hook bw_state (#121609)

jansel committed 191 days ago

3c8c7e2a

[DCP][state_dict] Remove the check of FSDP has root (#121544)

fegin committed 191 days ago

7a68e0a3

[DTensor] Moved `Transformer` sharding to staticmethod (#121660)

awgu committed 191 days ago

85dc2543

[ET-VK] Enable Dynamic shape support via tensor virtual and physical resizing (#121598)

SS-JIA committed 191 days ago

cc51e100

Update error message (#121644)

H-Huang committed 191 days ago

2a99e6f2

Modify signature of dequantize ops for decomposed quantized Tensor (#119173) (#121450)

kausikmaiti committed 191 days ago

edf22f3a

Support tt.reduce in Triton kernel analysis pass (#121706)

aakhundov committed 191 days ago

06d23920

[dynamo][compile-time] Caching VTs to reduce compile-time (#121031)

anijain2305 committed 191 days ago

78b4793c

Generate predispatch tests (#121678)

tugsbayasgalan committed 191 days ago

52ad2b68

[ROCm] enable complex128 in test_addmm_sizes_all_sparse_csr for rocm for trivial (k,n,m) cases (#120504)

dnikolaev-amd committed 191 days ago

656134c3

Simplify guards using info from previous guards (#121463)

lezcano committed 191 days ago

86a2d67b

Fix AARCH64 builds (#121700)

malfet committed 191 days ago

703e83e3

[quant][pt2e] Call sub-quantizers' transform_for_annotation in ComposableQuantizer (#121548)

sxu committed 191 days ago

159f3033

Also test predispatch serialization (#121652)

tugsbayasgalan committed 191 days ago

7fc49771

Express y grid > 2^16 in terms of z grid (#121554)

eellison committed 191 days ago

6ca9ae4f

[optim][BE] move complex_2d (last of complex tests) to OptimInfo (#120618)

janeyx99 committed 191 days ago

fb1d7935

Add Flash Attention support on ROCM (#121561)

xinyazhang committed 191 days ago

a37e22de

Port remove_split_ops to PT2 pre-grad passes (#121674)

kflu committed 191 days ago

3a5f48d5

Benchmark templates (#118880)

eellison committed 191 days ago

5b5d423c

[AOTInductor] Reuse generated kernels between constant graph and main graph (#121564)

muchulee8 committed 191 days ago

76764330

Newer Older