1 parent 756e47b commit 2db18c9Copy full SHA for 2db18c9
1 file changed
cuda_core/tests/graph/test_graph_node_update.py
@@ -67,8 +67,26 @@ def _update_executable_case(graph, case):
67
view.update(replacement[0], user_data=replacement[1])
68
else:
69
view.update(replacement)
70
- elif isinstance(case.node, (MemsetNode, MemcpyNode, KernelNode)):
71
- view.update(**replacement)
+ elif isinstance(case.node, MemsetNode):
+ 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
80
81
+ src=replacement["src"],
82
+ size=replacement["size"],
83
84
+ elif isinstance(case.node, KernelNode):
85
86
+ config=replacement["config"],
87
+ kernel=replacement["kernel"],
88
+ args=replacement["args"],
89
90
elif isinstance(case.node, ChildGraphNode):
91
view.update(replacement["child"])
92
else: # pragma: no cover - fixture cases are exhaustive
0 commit comments