Skip to content
Snippets Groups Projects
Commit 9ec3aabc authored by Isuru Fernando's avatar Isuru Fernando
Browse files

Check for same argument name when getting kernel arguments

parent 214fc5db
No related branches found
No related tags found
1 merge request!119Fix get_source_args for DirectionalTargetDerivative
......@@ -209,9 +209,10 @@ def gather_arguments(kernel_likes):
result = {}
for knl in kernel_likes:
for arg in knl.get_args():
result[arg.name] = arg
# FIXME: possibly check that arguments match before overwriting
# Check for strict equality until there's a usecase
if result.setdefault(arg.name, arg) != arg:
raise ValueError("Merging two different kernel arguments with"
"the same name {}".format(arg.name))
return sorted(six.itervalues(result), key=lambda arg: arg.name)
......@@ -219,8 +220,10 @@ def gather_source_arguments(kernel_likes):
result = {}
for knl in kernel_likes:
for arg in knl.get_args() + knl.get_source_args():
result[arg.name] = arg
# FIXME: possibly check that arguments match before overwriting
# Check for strict equality until there's a usecase
if result.setdefault(arg.name, arg) != arg:
raise ValueError("Merging two different kernel arguments with"
"the same name {}".format(arg.name))
return sorted(six.itervalues(result), key=lambda arg: arg.name)
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment