Skip to content

Commit 3f42420

Browse files
committed
[benchmarks] fix python formatting with black
1 parent 0a34e0d commit 3f42420

File tree

3 files changed

+30
-11
lines changed

3 files changed

+30
-11
lines changed

devops/scripts/benchmarks/main.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -154,7 +154,7 @@ def main(directory, additional_env_vars, save_name, compare_names, filter):
154154
SyclBench(directory),
155155
LlamaCppBench(directory),
156156
UMFSuite(directory),
157-
TestSuite()
157+
TestSuite(),
158158
]
159159
if not options.dry_run
160160
else []
@@ -442,7 +442,7 @@ def validate_and_parse_env_args(env_args):
442442
type=str,
443443
choices=[p.name for p in Presets],
444444
help="Benchmark preset to run.",
445-
default='FULL',
445+
default="FULL",
446446
)
447447

448448
args = parser.parse_args()

devops/scripts/benchmarks/options.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -2,6 +2,7 @@
22
from enum import Enum
33
from presets import Preset
44

5+
56
class Compare(Enum):
67
LATEST = "latest"
78
AVERAGE = "average"

devops/scripts/benchmarks/presets.py

Lines changed: 27 additions & 9 deletions
Original file line numberDiff line numberDiff line change
@@ -5,46 +5,64 @@
55

66
from enum import Enum
77

8-
class Preset():
8+
9+
class Preset:
910
def description(self):
1011
pass
12+
1113
def suites(self) -> list[str]:
1214
return []
1315

16+
1417
class Full(Preset):
1518
def description(self):
1619
return "All available benchmarks."
20+
1721
def suites(self) -> list[str]:
18-
return ['Compute Benchmarks', 'llama.cpp bench', 'SYCL-Bench', 'Velocity Bench', 'UMF']
22+
return [
23+
"Compute Benchmarks",
24+
"llama.cpp bench",
25+
"SYCL-Bench",
26+
"Velocity Bench",
27+
"UMF",
28+
]
29+
1930

2031
class SYCL(Preset):
2132
def description(self):
2233
return "All available benchmarks related to SYCL."
34+
2335
def suites(self) -> list[str]:
24-
return ['Compute Benchmarks', 'llama.cpp bench', 'SYCL-Bench', 'Velocity Bench']
36+
return ["Compute Benchmarks", "llama.cpp bench", "SYCL-Bench", "Velocity Bench"]
37+
2538

2639
class Minimal(Preset):
2740
def description(self):
2841
return "Short microbenchmarks."
42+
2943
def suites(self) -> list[str]:
30-
return ['Compute Benchmarks']
44+
return ["Compute Benchmarks"]
45+
3146

3247
class Normal(Preset):
3348
def description(self):
3449
return "Comprehensive mix of microbenchmarks and real applications."
50+
3551
def suites(self) -> list[str]:
36-
return ['Compute Benchmarks']
52+
return ["Compute Benchmarks"]
53+
3754

3855
class Test(Preset):
3956
def description(self):
4057
return "Noop benchmarks for framework testing."
58+
4159
def suites(self) -> list[str]:
42-
return ['Test Suite']
60+
return ["Test Suite"]
4361

4462

4563
class Presets(Enum):
4664
FULL = Full
47-
SYCL = SYCL # Nightly
48-
NORMAL = Normal # PR
49-
MINIMAL = Minimal # Quick smoke tests
65+
SYCL = SYCL # Nightly
66+
NORMAL = Normal # PR
67+
MINIMAL = Minimal # Quick smoke tests
5068
TEST = Test

0 commit comments

Comments
 (0)