Skip to content

Commit 10ed4c6

Browse files
author
Daniel Nichols
committed
clearer naming for driver relpath
1 parent cf6e857 commit 10ed4c6

1 file changed

Lines changed: 4 additions & 4 deletions

File tree

drivers/driver_wrapper.py

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -215,15 +215,15 @@ def test_single_output(self, prompt: str, output: str, test_driver_file: PathLik
215215

216216
def test_all_outputs_in_prompt(self, prompt: dict) -> dict:
217217
""" Run all the generated outputs in the given prompt. """
218-
root = prompt["language"]
218+
lang = prompt["language"]
219219
type = prompt["problem_type"]
220220
name = prompt["name"]
221221
ext = LANGUAGE_EXTENSIONS[prompt["language"]]
222-
if root == "cpp" and self.parallelism_model in ["cuda", "hip"]:
222+
if lang == "cpp" and self.parallelism_model in ["cuda", "hip"]:
223223
ext = ".cu"
224-
driver_root = f"{name}"
224+
driver_dirname = f"{name}"
225225
driver_base = DRIVER_MAP[self.parallelism_model]
226-
test_driver_file = os.path.join(root, "benchmarks", type, driver_root, driver_base + ext)
226+
test_driver_file = os.path.join(lang, "benchmarks", type, driver_dirname, driver_base + ext)
227227
problem_size = self.problem_sizes.get(name, {}).get(self.parallelism_model, "(1<<18)")
228228

229229
outputs = []

0 commit comments

Comments
 (0)