pytorch/benchmark

Pull Requests Commits

adding arc lint opt in for benchmarking (#2448)

Adam Mainz committed 1 year ago

7bb00692

Support for proxying frozen dataclasses (#134846)

mlazos committed 2 years ago

8c5abe8e

Add tma option for FB8 row-wise GEMM.

htyu committed 2 years ago

71b6e37d

Stop adding useless prefix to error message here, you're pushing the important info off the screen. (#133108)

ezyang committed 2 years ago

a95718f5

Add to torchbench relative path search (#134871)

yiming0416 committed 2 years ago

2927a2b8

adding llama inference specific shapes for flash attention

Adam Mainz committed 2 years ago

fdb33106

move loop ordering after fusion (#126254)

shunting314 committed 2 years ago

49bcc381

Switch benchmarking to use export non-strict mode (#130977)

desertfire committed 2 years ago

1f4461fb

Enable flash_v3 backward (#2445)

bertmaher committed 2 years ago

e6251f18

Add FlexAttention (#2443)

bertmaher committed 2 years ago

c0409aaf

Format flash_attention with black (#2444)

bertmaher committed 2 years ago

31d07c69

Add compile time instruction count metric (#133834)

laithsakka committed 2 years ago

d8735143

fix test_functional_call_sequential_params_and_buffers expectation on Windows (#134394)

xuhancn committed 2 years ago

ae902e4c

fixing OSS ci for tritonbench (#2439)

Adam Mainz committed 2 years ago

b0ae45a8

PR#4179 (#2435)

Elliot Gorokhovsky committed 2 years ago

babb1285

Install the benchmark before running. (#2436)

xuzhao9 committed 2 years ago

52103b5e

Enable cublas for FP8 rowwise

htyu committed 2 years ago

ef3ce35e

typing for decorators - fx/_compatibility (part 1) (#134202)

aorenste committed 2 years ago

f4f13306

Always emit end events even on failure, use thread local storage for stack (#2432)

jamesjwu committed 2 years ago

0968f5ed

fix PR #2428 (#2433)

sfzhu93 committed 2 years ago

ea7a2d35

Collect Export pass rate separately (#134076)

desertfire committed 2 years ago

41cf189a

add support for auto-tune TMA grid constant (#2428)

sfzhu93 committed 2 years ago

fa08aeb5

Remove non-existing arg from docstring (#2431)

kit1980 committed 2 years ago

d445d6d7

Increase max total number of dynamo partitions to 15 (#134153)

malfet committed 2 years ago

3d6d97a0

anijain2305 committed 2 years ago

3cd3dd55

Log chromium events to scuba (#2429)

jamesjwu committed 2 years ago

b38b6f33

Correctly get run_id and run_attempt (#2426)

kit1980 committed 2 years ago

96200f70

Add HSTU apply_rope and batched_multihead_index_sum

xuzhao9 committed 2 years ago

5a121b69

Add fbgemm split_embedding_table benchmark

xuzhao9 committed 2 years ago

57de51f8

Support pop torch function mode stack (#133131)

mlazos committed 2 years ago

78d22e56

Newer Older