strisunshinewentingwang/pytorch

Pull Requests Commits

[export] Add aot_export 1/N (#101490)

tugsbayasgalan committed 2 years ago

d9f75dde

Try to build the Docker image if it doesn't exist (#102562)

huydhn committed 2 years ago

04c1c2b7

[RFC] Add method to DDP to check for backward finalization. (#100773)

pritamdamania87 committed 2 years ago

9a2df0a5

Allow existing "Python RAII guards" to be used as context managers (#102579)

zou3519 committed 2 years ago

fc31b3a1

[benchmarks] Use train mode for accuracy checks for HF models (#102578)

anijain2305 committed 2 years ago

65631d45

fix bug in trace model when out-operator has more than one output (#101563)

heidongxianhua committed 2 years ago

213e10dc

python_arg_parser to allow fake tensor element in symint_list when in dynamo mode #95424 (#97508)

lantiankaikai committed 2 years ago

17166c25

adjust header inclusions in C10 as sugguested by IWYU (#102467)

cyyever committed 2 years ago

3ae42cb7

[Replicate] Add unit test with replicate param names (#102401)

rohan-varma committed 2 years ago

0ecca122

Run slow gradcheck on the newer G5 runner (#102496)

huydhn committed 2 years ago

9331b7fa

Faketensor hpu device normalization (#102512)

VRSinghHabana committed 2 years ago

c27cefcc

Enable hipSOLVER in ROCm builds (#97370)

alugorey committed 2 years ago

eaffd988

S390x clang fixes for SIMD (#100874)

AlekseiNikiforovIBM committed 2 years ago

46a92579

merge identical branches in cpu_index_kernel (#102601)

cyyever committed 2 years ago

850b37cc

Fix ignored_states when they are passed as generators (#102575)

zhaojuanmao committed 2 years ago

f47ee877

Add integer overflow checks for large compressed tensor dimensions and nnz (#102530)

pearu committed 2 years ago

9f97b7c4

Improve repeat_interleave with scalar repeat value (#102570)

peterbell10 committed 2 years ago

9fd14fcd

[inductor] Revert a CI remedy for Triton compilation error (#102541)

desertfire committed 2 years ago

47b884a7

nn.Linear with BSR inputs: spare the user from explicit Triton kernel registrations (#98403)

nikitaved committed 2 years ago

d80d3b18

[Executorch][codegen] Add ETKernelIndex for aggregating all kernels for kernel (#102565)

larryliu0820 committed 2 years ago

019c3862

apply constexpr and if constexpr when possible (#102471)

cyyever committed 2 years ago

7c2641d5

Quick fix for keep-going + reruns (#102569)

clee2000 committed 2 years ago

a5ddb72a

[inductor] Fix issue with 0D reductions (#102568)

jansel committed 2 years ago

3c0251a1

[inductor][pattern matcher] Retain meta tags (#102462)

anijain2305 committed 2 years ago

46691d43

Add dynamo collections.deque support (#102412)

wconstab committed 2 years ago

e7cc4177

Fix warning couldn't find split args (#102561)

devashishshankar committed 2 years ago

cdca25cd

[ONNX] Reduce exporter memory usage by removing intermediate values (#101148)

ilyasher committed 2 years ago

b4a49124

[profiler] Reintroduce forward-backward links (#102424)

davidberard98 committed 2 years ago

5324124e

add function specializations for the case of parameters in BFloat16 data type (#100233)

CaoE committed 2 years ago

73fd7235

[build] fix compilation error on s390x (#101923)

kiszk committed 2 years ago

9edf65a8

Older