pytorch/benchmark

Pull Requests Commits

Run performance test non-alternately (#131935)

HDCharles committed 2 years ago

bf0e5a96

Skip pruning the CUDA 12.4 docker (#2411)

xuzhao9 committed 2 years ago

9a99b9b0

Mark nn module tensor static for cudagraphs (#132736)

anijain2305 committed 2 years ago

0da28c03

Reland 132308, 132314, 132318, 132334 - Make builtin nn modules attributes static (#132539)

anijain2305 committed 2 years ago

a9b8a9a6

Support dict(obj) on constrained subclasses of dict and OrderedDict (#132558)

anijain2305 committed 2 years ago

f597aa44

Rename _internal/exporter to _exporter_legacy (#132429)

justinchuby committed 2 years ago

47ca8ea9

revert map/zip iterator related changes (#132528)

williamwen42 committed 2 years ago

342857cf

Ban decorator usage of dynamo_timed (#132328)

ezyang committed 2 years ago

38880cad

Add lost huggingface.yaml (#2408)

kit1980 committed 2 years ago

ae2c9e8a

Adding llama31 (#2404)

MaanavD committed 2 years ago

86c8ec6c

Populate submodules of `torch._C` to `sys.modules` recursively (#132216)

XuehaiPan committed 2 years ago

5aed9cda

TMA support for FA (#2379)

manman-ren committed 2 years ago

d8454b98

Rename guard_source NN_MODULE to SPECIALIZED_NN_MODULE (#132302)

anijain2305 committed 2 years ago

f2e18194

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

XuehaiPan committed 2 years ago

a9cc6ee0

Add ThunderKittens support (#2370)

xuzhao9 committed 2 years ago

e05b6fa9

Add Inductor torch.compile benchmark for jagged_layer_norm operator

jananisriram committed 2 years ago

043b07d4

Add Inductor torch.compile benchmark for jagged_softmax operator

jananisriram committed 2 years ago

55f03bbf

Add Inductor torch.compile benchmark for jagged_mean operator

jananisriram committed 2 years ago

c031f41b

Add Inductor torch.compile benchmark for jagged_sum operator

jananisriram committed 2 years ago

d7f6346f

Refactor jagged operator benchmarks in TritonBench with helper functions for readability

jananisriram committed 2 years ago

31ebbaaa

Fix colfax_cutlass flash_attention operator (#2401)

xuzhao9 committed 2 years ago

0a2ff223

add large-k shapes to addmm

karthik-man committed 2 years ago

f4ed185c

Skip memory leak for sam_fast (#2402)

kit1980 committed 2 years ago

d236f4c1

Download from pytorch ci script (#2397)

xuzhao9 committed 2 years ago

67b5e75c

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

xuzhao9 committed 2 years ago

2698ff8b

Print traceback when exception occurs during benchmarking (#2398)

bertmaher committed 2 years ago

89cdb8a6

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

bertmaher committed 2 years ago

974d0c43

Install torchao in nightly docker build (#2395)

xuzhao9 committed 2 years ago

56b9036e

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

kit1980 committed 2 years ago

0dc03849

Fix v2 nightly and bisection workflows (#2366)

xuzhao9 committed 2 years ago

6ca69e64

Newer Older