pytorch/benchmark

Pull Requests Commits

Mark Saroufim committed 2 years ago

5f65e3c2

Mark Saroufim committed 2 years ago

861937d3

Mark Saroufim committed 2 years ago

799fd714

Mark Saroufim committed 2 years ago

849ac962

Mark Saroufim committed 2 years ago

f8666cdc

Merge branch 'main' into msaroufim/cip

Mark Saroufim committed 2 years ago

657e9c09

Mark Saroufim committed 2 years ago

79b917d0

Update LLAMA v2 init (#1781)

msaroufim committed 2 years ago

28c45ec1

Exclude PT2 on most models, add more knobs to script (#1785)

janeyx99 committed 2 years ago

ed895d7e

Print out runnable repros with better time metadata (#1783)

janeyx99 committed 2 years ago

0d166d10

training now works

Mark Saroufim committed 2 years ago

54b2fdec

Mark Saroufim committed 2 years ago

4a83bf60

Mark Saroufim committed 2 years ago

541d5837

Skip CPU benchmarks since we don't observe their stats (#1784)

janeyx99 committed 2 years ago

725b72cc

Mark Saroufim committed 2 years ago

c3d1c44c

fix loss calculation

Mark Saroufim committed 2 years ago

b4f75a41

fix device test

Mark Saroufim committed 2 years ago

fea57c45

ankita feedback

Mark Saroufim committed 2 years ago

3c618131

Update __init__.py

msaroufim committed 2 years ago

Verified 692e68df

add training support

Mark Saroufim committed 2 years ago

911a5ab6

Mark Saroufim committed 2 years ago

3fa02666

Print torch git version (commit) in workflow (#1779)

janeyx99 committed 2 years ago

d9e6653a

fix uninitialized model_flop variable issue (#1778)

snadampal committed 2 years ago

c03597e8

llama v2 (#1775)

Mark Saroufim committed 2 years ago

196b3b8a

Add functionality to TRT backend (#1753)

gs-olive committed 2 years ago

30a3c4a9

Get PT-D benchmark to support being run internally

H-Huang committed 2 years ago

36e1ed1c

Add torch.utils.flop_counter to TorchBench

monicatang committed 2 years ago

1dada69d

Fix fvcore flops counting for torchvision models

xuzhao9 committed 2 years ago

9d84f9e6

Add dependencies in the base docker build (#1776)

xuzhao9 committed 2 years ago

79bc754f

Fix require_grad typo (#1771)

kit1980 committed 2 years ago

ae8f06fe

Older