strisunshinewentingwang/pytorch

Pull Requests Commits

[torch export] replace traceback.extract_stack with CapturedTraceback.extract (#121449)

strisunshinewentingwang committed 2 years ago

90e271dd

Simplify guards using info from previous guards (#121463)

lezcano committed 2 years ago

86a2d67b

Fix AARCH64 builds (#121700)

malfet committed 2 years ago

703e83e3

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

sxu committed 2 years ago

159f3033

Also test predispatch serialization (#121652)

tugsbayasgalan committed 2 years ago

7fc49771

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

eellison committed 2 years ago

6ca9ae4f

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

janeyx99 committed 2 years ago

fb1d7935

Add Flash Attention support on ROCM (#121561)

xinyazhang committed 2 years ago

a37e22de

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

kflu committed 2 years ago

3a5f48d5

Benchmark templates (#118880)

eellison committed 2 years ago

5b5d423c

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

muchulee8 committed 2 years ago

76764330

[FSDP2][BE] Refactored `check_1d_sharded_parity` to use mesh (#121357)

Andrew Gu committed 2 years ago

272cf29e

Delete requirements-flake8.txt (#121657)

kit1980 committed 2 years ago

cd1dc5e4

Revert "Batch Norm Consolidation (#116092)"

pytorchmergebot committed 2 years ago

fd0dbcd8

Don't install torchfix for python<3.9 (#121655)

kit1980 committed 2 years ago

498a94a7

Revert "[compiled autograd] support custom ops backed by c++ autograd::Function (#120681)"

pytorchmergebot committed 2 years ago

b2f09c18

Check for releasing GIL at compiletime (#116695)

Flamefire committed 2 years ago

d1f45a93

Refactor some testing helpers for FX graph cache testing (#121520)

masnesral committed 2 years ago

fd13a56f

[ROCm] Autocast RNN Support (#121539)

alugorey committed 2 years ago

e01b07e1

port fuse_parallel_linear (without changing weights) to PT2 pre-grad (#121617)

kflu committed 2 years ago

fc712311

[pt2 export]fix name collision on constant name (#121145)

zzh142857 committed 2 years ago

34614048

Add a section on release wiki about pytorchbot cherry-pick command (#121648)

huydhn committed 2 years ago

b091a329

fix CMake FindCUDA module for cross-compiling (#121590)

njzjz committed 2 years ago

dd2062c7

Include torch warn in each error in cudnn/Conv_v8.cpp (#120719)

lancerts committed 2 years ago

5fd7f5c4

Slightly faster FX graph iterator (#121611)

jansel committed 2 years ago

9aa3fedb

Update torchbench commit pin, add sam_fast benchmark (#121420)

jamesjwu committed 2 years ago

ae22bdae

[torch] Use __prepare_scriptable__ for closures (#121553)

dherrera-meta committed 2 years ago

dccc1ca8

Clean up macOS x86 binaries build jobs (#116726)

huydhn committed 2 years ago

b4160fd9

Bring torch_xla pin to the latest torch_xla commit (03/08/2024). (#121529)

vanbasten23 committed 2 years ago

8d03c59d

[TEST] Prepare test_cumulative_trapezoid for SciPy 1.12 (#121541)

Aidyn-A committed 2 years ago

39ed038f

Older