Skip to content

Commit 2db18c9

Browse files
committed
test(core): restore explicit executable update kwargs
The shared replacement fixture carries extra fields that are not update parameters, so spreading it as kwargs breaks memset and kernel cases.
1 parent 756e47b commit 2db18c9

1 file changed

Lines changed: 20 additions & 2 deletions

File tree

‎cuda_core/tests/graph/test_graph_node_update.py‎

Lines changed: 20 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -67,8 +67,26 @@ def _update_executable_case(graph, case):
6767
view.update(replacement[0], user_data=replacement[1])
6868
else:
6969
view.update(replacement)
70-
elif isinstance(case.node, (MemsetNode, MemcpyNode, KernelNode)):
71-
view.update(**replacement)
70+
elif isinstance(case.node, MemsetNode):
71+
view.update(
72+
dst=replacement["dst"],
73+
value=replacement["value"],
74+
width=replacement["width"],
75+
height=replacement["height"],
76+
pitch=replacement["pitch"],
77+
)
78+
elif isinstance(case.node, MemcpyNode):
79+
view.update(
80+
dst=replacement["dst"],
81+
src=replacement["src"],
82+
size=replacement["size"],
83+
)
84+
elif isinstance(case.node, KernelNode):
85+
view.update(
86+
config=replacement["config"],
87+
kernel=replacement["kernel"],
88+
args=replacement["args"],
89+
)
7290
elif isinstance(case.node, ChildGraphNode):
7391
view.update(replacement["child"])
7492
else: # pragma: no cover - fixture cases are exhaustive

0 commit comments

Comments
 (0)