pytorch/benchmark

Pull Requests Commits

Rename guard_source NN_MODULE to SPECIALIZED_NN_MODULE (#132302)

anijain2305 committed 1 year ago

f2e18194

enforce style for empty lines in import segments in `torch/_d*/` (#129767)

XuehaiPan committed 1 year ago

a9cc6ee0

Add ThunderKittens support (#2370)

xuzhao9 committed 1 year ago

e05b6fa9

Add Inductor torch.compile benchmark for jagged_layer_norm operator

jananisriram committed 1 year ago

043b07d4

Add Inductor torch.compile benchmark for jagged_softmax operator

jananisriram committed 1 year ago

55f03bbf

Add Inductor torch.compile benchmark for jagged_mean operator

jananisriram committed 1 year ago

c031f41b

Add Inductor torch.compile benchmark for jagged_sum operator

jananisriram committed 1 year ago

d7f6346f

Refactor jagged operator benchmarks in TritonBench with helper functions for readability

jananisriram committed 1 year ago

31ebbaaa

Fix colfax_cutlass flash_attention operator (#2401)

xuzhao9 committed 1 year ago

0a2ff223

add large-k shapes to addmm

karthik-man committed 1 year ago

f4ed185c

Skip memory leak for sam_fast (#2402)

kit1980 committed 1 year ago

d236f4c1

Download from pytorch ci script (#2397)

xuzhao9 committed 1 year ago

67b5e75c

Add `--gpu-lockdown` option to pin GPU frequency and add power limit

xuzhao9 committed 1 year ago

2698ff8b

Print traceback when exception occurs during benchmarking (#2398)

bertmaher committed 1 year ago

89cdb8a6

Gemm shapes for llama3.1-405B (#2396)

bertmaher committed 1 year ago

974d0c43

Install torchao in nightly docker build (#2395)

xuzhao9 committed 1 year ago

56b9036e

Move benchmarks/dynamo/huggingface configuration to YAML (#131724)

kit1980 committed 1 year ago

0dc03849

Fix v2 nightly and bisection workflows (#2366)

xuzhao9 committed 1 year ago

6ca69e64

add lazy IteratorVariable implementations for map and zip (#131413)

williamwen42 committed 1 year ago

3f8fa499

Move load_yaml_file to common (#131924)

kit1980 committed 1 year ago

dfb15d96

Add hf_T5_generate to inline_inbuilt_nn_modules (#131804)

anijain2305 committed 1 year ago

7797766e

Deploy the H100 flash_attention operator

xuzhao9 committed 1 year ago

5cc1d010

Set dump_exported_program to True in bench (#131670)

justinchuby committed 1 year ago

06cf234e

Updating Types in torch/_dynamo/utils.py (#131001)

adriaorenstein committed 1 year ago

fe362296

Run only a subset for optim benchmarks, attempt to fix (#2392)

janeyx99 committed 1 year ago

9ad6639a

Fix dimension order for FlashAttention-3 (#2391)

bertmaher committed 1 year ago

ff65cb67

Read from dataclass fields

xuzhao9 committed 1 year ago

6af4bd93

Triton Partition K gemm to TritonBench

sijiac committed 1 year ago

53d98e37

Always obtain `model_flops` from the eager mode (#2390)

xuzhao9 committed 1 year ago

43d8a999

Run accuracy check at Diff time (#2388)

xuzhao9 committed 1 year ago

0032c0b4

Newer Older