pytorch/benchmark

Pull Requests Commits

Install the benchmark before running.

xuzhao9 committed 1 year ago

8439da56

Enable cublas for FP8 rowwise

htyu committed 1 year ago

ef3ce35e

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

aorenste committed 1 year ago

f4f13306

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

jamesjwu committed 1 year ago

0968f5ed

fix PR #2428 (#2433)

sfzhu93 committed 1 year ago

ea7a2d35

Collect Export pass rate separately (#134076)

desertfire committed 1 year ago

41cf189a

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

sfzhu93 committed 1 year ago

fa08aeb5

Remove non-existing arg from docstring (#2431)

kit1980 committed 1 year ago

d445d6d7

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

malfet committed 1 year ago

3d6d97a0

anijain2305 committed 1 year ago

3cd3dd55

Log chromium events to scuba (#2429)

jamesjwu committed 1 year ago

b38b6f33

Correctly get run_id and run_attempt (#2426)

kit1980 committed 1 year ago

96200f70

Add HSTU apply_rope and batched_multihead_index_sum

xuzhao9 committed 1 year ago

5a121b69

Add fbgemm split_embedding_table benchmark

xuzhao9 committed 1 year ago

57de51f8

Support pop torch function mode stack (#133131)

mlazos committed 1 year ago

78d22e56

Add torch function mode stack guard to dynamo (#133130)

mlazos committed 1 year ago

63e957cc

Fix jagged layer norm torch_compile_nested_tensor_integration

YuqingJ committed 1 year ago

dd547894

Support reconstructing CUDA event object within Dynamo graph (#133635)

yf225 committed 1 year ago

64130f1d

Disable jagged layer norm with torch.compile

xuzhao9 committed 1 year ago

22a574fa

logging hardware specific data in userbench (#2420)

adamomainz committed 1 year ago

fdd7def3

Revert #132806, #132736, #132539, #132487

ezyang committed 1 year ago

6f5d34a6

Bump maxinum num warps (#132458)

eellison committed 1 year ago

c6db4bcb

Revert moving the cupti headers and library files (#2419)

xuzhao9 committed 1 year ago

23fe8251

Fix the broken unit tests

xuzhao9 committed 1 year ago

a2ea728d

Stop using deprecated torch.distributed.reduce_op

alanhdu committed 1 year ago

bbc58210

Make torch.Size can be reconstructed by LOAD_CONST (#133342)

yanboliang committed 1 year ago

d783c63f

Re-sync with internal repository (#2417)

facebook-github-bot committed 1 year ago

Verified 846177e2

Add ChromiumEventLogger, log FXGraphCache and AOTAutogradCache (#132864)

jamesjwu committed 1 year ago

dfa0fe0d

Fix CUPTI and Debugger libraries in the docker image (#2414)

xuzhao9 committed 1 year ago

9f7977bf

Allow omitting the "Bias" column in data table (#2412)

bertmaher committed 1 year ago

cc5eaf7c

Older