Skip to content

Commit cc19515

Browse files
committed
Extended hypertuner with additional kernels, adjusted for benchmark_hub
1 parent c39ac5a commit cc19515

File tree

1 file changed

+15
-3
lines changed

1 file changed

+15
-3
lines changed

kernel_tuner/backends/hypertuner.py

Lines changed: 15 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -64,17 +64,28 @@ def compile(self, kernel_instance):
6464
# gpus = ["RTX_3090", "RTX_2080_Ti"]
6565
# applications = None
6666

67-
gpus = ["A100", "A4000", "MI250X", "W6600"]
67+
gpus = ["A100", "W6600"]
68+
folder = "../../autotuning_methodology/benchmark_hub/kernels"
6869
applications = [
6970
{
7071
"name": "dedispersion_milo",
71-
"folder": "../autotuning_methodology/benchmark_hub/kernels",
72+
"folder": folder,
7273
"input_file": "dedispersion_milo.json"
7374
},
7475
{
7576
"name": "convolution_milo",
76-
"folder": "../autotuning_methodology/benchmark_hub/kernels",
77+
"folder": folder,
7778
"input_file": "convolution_milo.json"
79+
},
80+
{
81+
"name": "hotspot_milo",
82+
"folder": folder,
83+
"input_file": "hotspot_milo.json"
84+
},
85+
{
86+
"name": "gemm_milo",
87+
"folder": folder,
88+
"input_file": "gemm_milo.json"
7889
}
7990
]
8091

@@ -93,6 +104,7 @@ def compile(self, kernel_instance):
93104
# any additional settings
94105
override = {
95106
"experimental_groups_defaults": {
107+
"repeats": 10,
96108
"samples": self.iterations
97109
}
98110
}

0 commit comments

Comments
 (0)