Skip to content

[benchmarks] overhaul benchmarks #11565

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Open
wants to merge 31 commits into
base: main
Choose a base branch
from
Open
Changes from 1 commit
Commits
Show all changes
31 commits
Select commit Hold shift + click to select a range
24a46cc
start overhauling the benchmarking suite.
sayakpaul May 15, 2025
ab7f381
fixes
sayakpaul May 15, 2025
cc0a38a
fixes
sayakpaul May 15, 2025
169f831
checking.
sayakpaul May 15, 2025
ad18983
checking
sayakpaul May 15, 2025
31e34d5
fixes.
sayakpaul May 16, 2025
36afdea
error handling and logging.
sayakpaul May 16, 2025
0d3af90
Merge branch 'main' into benchmarking-overhaul
sayakpaul May 16, 2025
fd85fbc
Merge branch 'main' into benchmarking-overhaul
sayakpaul May 19, 2025
a2c03a4
Merge branch 'main' into benchmarking-overhaul
sayakpaul May 20, 2025
4d83a47
add flops and params.
sayakpaul May 20, 2025
6815cae
add more models.
sayakpaul May 20, 2025
5635bf8
utility to fire execution of all benchmarking scripts.
sayakpaul May 20, 2025
cfbd21e
utility to push to the hub.
sayakpaul May 20, 2025
4ccfad0
push utility improvement
sayakpaul May 20, 2025
dff3144
seems to be working.
sayakpaul May 20, 2025
accd598
okay
sayakpaul May 20, 2025
41f79a0
add torchprofile dep.
sayakpaul May 20, 2025
befdd9e
remove total gpu memory
sayakpaul May 20, 2025
4784b8b
fixes
sayakpaul May 20, 2025
c19dc5b
fix
sayakpaul May 20, 2025
2da4fac
need a big gpu
sayakpaul May 20, 2025
7367bb1
better
sayakpaul May 20, 2025
1cd472f
what's happening.
sayakpaul May 20, 2025
214795d
okay
sayakpaul May 20, 2025
7d4f459
Merge branch 'main' into benchmarking-overhaul
sayakpaul May 21, 2025
2d5b305
Merge branch 'main' into benchmarking-overhaul
sayakpaul May 22, 2025
dd42244
Merge branch 'main' into benchmarking-overhaul
sayakpaul May 27, 2025
9d28606
Merge branch 'main' into benchmarking-overhaul
sayakpaul May 28, 2025
ffed3b3
Merge branch 'main' into benchmarking-overhaul
sayakpaul May 29, 2025
a28c881
Merge branch 'main' into benchmarking-overhaul
sayakpaul May 29, 2025
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
Prev Previous commit
Next Next commit
checking
  • Loading branch information
sayakpaul committed May 15, 2025
commit ad18983985b6bc4823f89f97986473869277ecad
9 changes: 0 additions & 9 deletions benchmarks/benchmarking_utils.py
Original file line number Diff line number Diff line change
Expand Up @@ -37,15 +37,6 @@ class BenchmarkScenario:

@require_torch_gpu
class BenchmarkMixin:
def get_model_init_dict(self):
raise NotImplementedError

def initialize_model(self):
raise NotImplementedError

def get_input_dict(self):
raise NotImplementedError

def pre_benchmark(self):
flush()
torch.compiler.reset()
Expand Down