def get_pending_cmd(self, watch_nodes):
"""Return the watchpoint in proto format."""
# construct SetCMD
condition_id = self._condition.get('id')
set_cmd = SetCMD()
set_cmd.id = self._id
set_cmd.delete = False
set_cmd.watch_condition.condition = WATCHPOINT_CONDITION_MAPPING.get(
condition_id)
condition_mgr = ConditionMgr()
condition = condition_mgr.get_condition(condition_id)
param_dict = {
param.get('name'): param
for param in self._condition.get('params')
}
for param_name in condition.ordered_parameter_names:
param = param_dict.get(param_name)
if param:
param_proto = set_cmd.watch_condition.params.add()
param_proto.name = param.get('name')
param_proto.value = param.get('value')
param_proto.disabled = False
# Only one parameter of condition in old mindspore version.
set_cmd.watch_condition.value = param.get('value')
else:
param_proto = set_cmd.watch_condition.params.add()
param_proto.name = param_name
param_proto.disabled = True
for watch_node in watch_nodes:
event_node = set_cmd.watch_nodes.add()
event_node.node_name = watch_node.full_name
event_node.node_type = watch_node.type
return set_cmd
def delete_watchpoint(self, watch_point_id):
"""
Delete watchpoint.
Args:
watch_point_id (int): The id of watchpoint.
Returns:
dict, empty response.
"""
self.validate_watchpoint_id(watch_point_id)
self._watchpoints.pop(watch_point_id)
set_cmd = SetCMD()
set_cmd.id = watch_point_id
set_cmd.delete = True
self._deleted_watchpoints.append(set_cmd)
log.debug("Delete watchpoint %d in cache.", watch_point_id)
def _delete_single_watchpoint(self, watch_point_id):
"""
Delete single watchpoint.
Args:
watch_point_id (int): The id of watchpoint.
"""
self._watchpoints.pop(watch_point_id)
# if the watchpoint has not been created by MindSpore, clean the relative cache directly
if watch_point_id in self._created_watchpoints:
self._created_watchpoints.remove(watch_point_id)
self._updated_watchpoints.pop(watch_point_id)
log.debug("Cancel create watchpoint %d in cache.", watch_point_id)
return
set_cmd = SetCMD()
set_cmd.id = watch_point_id
set_cmd.delete = True
self._deleted_watchpoints.append(set_cmd)
log.debug("Delete watchpoint %d in cache.", watch_point_id)
def get_set_cmd(self):
"""Return the watchpoint in proto format."""
# get watch nodes.
watch_nodes = []
self.get_watch_node(self._watch_node, watch_nodes)
# construct SetCMD
set_cmd = SetCMD()
set_cmd.id = self._id
set_cmd.delete = False
set_cmd.watch_condition.condition = WATCHPOINT_CONDITION_MAPPING.get(
self._condition.get('condition'))
if self._condition.get('param'):
# at most one param is provided
set_cmd.watch_condition.value = self._condition.get('param')
for watch_node in watch_nodes:
event_node = set_cmd.watch_nodes.add()
event_node.node_name = watch_node.full_name
event_node.node_type = watch_node.node_type
return set_cmd
def get_set_cmd():
"""Get set command"""
event = get_ack_reply()
event.set_cmd.CopyFrom(SetCMD(id=1, watch_condition=1))
return event