Skip to content

Commit 0da80f6

Browse files
extend argument check to multiple files
1 parent 5b3675a commit 0da80f6

File tree

1 file changed

+5
-1
lines changed

1 file changed

+5
-1
lines changed

kernel_tuner/interface.py

Lines changed: 5 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -299,7 +299,11 @@ def tune_kernel(kernel_name, kernel_string, problem_size, arguments,
299299

300300
# see if the kernel arguments have correct type
301301
if not callable(kernel_string):
302-
util.check_argument_list(kernel_name, util.get_kernel_string(kernel_string), arguments)
302+
if isinstance(kernel_string, list):
303+
for file in kernel_string:
304+
util.check_argument_list(kernel_name, util.get_kernel_string(file), arguments)
305+
else:
306+
util.check_argument_list(kernel_name, util.get_kernel_string(kernel_string), arguments)
303307
else:
304308
logging.debug("Checking of arguments list not supported yet for code generators.")
305309

0 commit comments

Comments
 (0)