@@ -20,30 +20,31 @@ def setup(self):
2020
2121 def benchmarks (self ) -> list [Benchmark ]:
2222 bench_configs = [
23- ("Memory Bandwidth" , 2000 , 200 ),
24- ("Latency" , 100 , 20 ),
25- ("Throughput" , 1500 , 150 ),
26- ("FLOPS" , 3000 , 300 ),
27- ("Cache Miss Rate" , 250 , 25 ),
23+ ("Memory Bandwidth" , 2000 , 200 , "Foo Group" ),
24+ ("Latency" , 100 , 20 , "Bar Group" ),
25+ ("Throughput" , 1500 , 150 , "Foo Group" ),
26+ ("FLOPS" , 3000 , 300 , "Foo Group" ),
27+ ("Cache Miss Rate" , 250 , 25 , "Bar Group" ),
2828 ]
2929
3030 result = []
31- for base_name , base_value , base_diff in bench_configs :
31+ for base_name , base_value , base_diff , group in bench_configs :
3232 for variant in range (6 ):
3333 value_multiplier = 1.0 + (variant * 0.2 )
3434 name = f"{ base_name } { variant + 1 } "
3535 value = base_value * value_multiplier
3636 diff = base_diff * value_multiplier
3737
38- result .append (TestBench (name , value , diff ))
38+ result .append (TestBench (name , value , diff , group ))
3939
4040 return result
4141
4242class TestBench (Benchmark ):
43- def __init__ (self , name , value , diff ):
43+ def __init__ (self , name , value , diff , group = '' ):
4444 self .bname = name
4545 self .value = value
4646 self .diff = diff
47+ self .group = group
4748 super ().__init__ ("" )
4849
4950 def name (self ):
@@ -58,7 +59,7 @@ def setup(self):
5859 def run (self , env_vars ) -> list [Result ]:
5960 random_value = self .value + random .uniform (- 1 * (self .diff ), self .diff )
6061 return [
61- Result (label = self .name (), value = random_value , command = "" , env = {"A" : "B" }, stdout = "no output" , unit = "ms" )
62+ Result (label = self .name (), explicit_group = self . group , value = random_value , command = "" , env = {"A" : "B" }, stdout = "no output" , unit = "ms" )
6263 ]
6364
6465 def teardown (self ):
0 commit comments