pytorch/benchmark

Pull Requests Commits

Add layout options to gemm

bertmaher committed 2 years ago

4445aa2b

CSV of extra shapes for gemm benchmarks

bertmaher committed 2 years ago

d512e673

Modernize cutlass call for fp8 blockwise

bertmaher committed 2 years ago

2d9ab0b1

Print TMA benchmark info to stderr

bertmaher committed 2 years ago

0cb1e96d

Skip loading triton.nvidia.cublas if not found

bertmaher committed 2 years ago

0611c41c

Make FA3 work in fbcode

bertmaher committed 2 years ago

b6b67a4f

Fix hardcoded shape in low_mem_dropout benchmark (#2475)

mark14wu committed 2 years ago

252a3b17

Restore FlexAttention and FlashV3 backward (#2473)

bertmaher committed 2 years ago

611bf702

Fix bug #2458 (#2459)

ostrowskimarcin committed 2 years ago

0f050151

Add nsys integration

xuzhao9 committed 2 years ago

2edf80cb

Revert "Trace enter/exit of TorchFunctionModes (#135422)" (#136590)

anijain2305 committed 2 years ago

a31c3fe9

Move FBGEM to 5c3d54f335b1617b5b169061c3b3d59b2a791ebb (#2471)

htyu committed 2 years ago

8a690690

Do not treat user defined nn module attributes static for dynamic shape infra (#136516)

anijain2305 committed 2 years ago

93b58047

Add release tests (#2470)

atalman committed 2 years ago

7d30648b

Add types to _dynamo/testing.py (#136402)

bobrenjc93 committed 2 years ago

deb2988e

Add HSTU ragged attention operator (#2453)

xuzhao9 committed 2 years ago

926fb10a

bf16xint16_gemm operator: add --transpose option (#2466)

davidberard98 committed 2 years ago

0ab0e47e

Move FBGEM to fcf36b445a1929c8a22f07c35d3a29b3697867a4 (#2467)

htyu committed 2 years ago

6a089a45

Add _dynamo.config.suppress_errors logging (#136379)

jovianjaison committed 2 years ago

8d57e930

Run single iteration when collecting ncu traces

xuzhao9 committed 2 years ago

f2f0b30f

make --dump-ir also dump sass (#2460)

davidberard98 committed 2 years ago

46ab2e2b

Log structured logging overhead to dynamo compile (kinda) (#2454)

jamesjwu committed 2 years ago

9c7aacc4

Increase multiplier to 3 for Inductor AMP FP16 benchmark correctness check (#135932)

jiayisunx committed 2 years ago

c1755f53

use npt.NDArray instead of np.ndarray in type annotations

igorsugak committed 2 years ago

c97859eb

Update mypy to 1.11.2 (#133816)

Skylion007 committed 2 years ago

7e1ba8d5

Remove ignored modes workaround (#135502)

mlazos committed 2 years ago

6b1c40bc

Trace enter/exit of TorchFunctionModes (#135422)

mlazos committed 2 years ago

99ea924e

Trace torch function modes entered outside of torch.compile (#133137)

mlazos committed 2 years ago

2536a65b

SKIP llama for dynamic size testing (#135960)

leslie-fang-intel committed 2 years ago

c709128f

Add remote cache time saved to compilation metrics (#2449)

jamesjwu committed 2 years ago

47413ed5

Newer Older