def check_op_name(self, op_name):
"""
Check whether the operator name exists.
Args:
op_name (str): The operator name or operator name prefix.
Returns:
bool, `True` if the operator name does exist, else `False`.
"""
if not op_name:
raise ProfilerParamValueErrorException('The op_name should exist.')
for op_time_info in self._timeline_meta:
full_op_name = op_time_info['name']
if full_op_name and full_op_name.startswith(op_name):
return True
return False
def validate_ui_proc(proc_name):
"""
Validate proc name in restful request.
Args:
proc_name (str): The proc name to query. Acceptable value is in
[`iteration_interval`, `fp_and_bp`, `tail`].
Raises:
ProfilerParamValueErrorException: If the proc_name is invalid.
"""
accept_names = ['iteration_interval', 'fp_and_bp', 'tail']
if proc_name not in accept_names:
log.error("Invalid proc_name. The proc_name for restful api is in %s",
accept_names)
raise ProfilerParamValueErrorException(
f'proc_name should be in {accept_names}.')
def check_op_name(self, op_name, is_prefix=True):
"""
Check whether the operator name exists.
Args:
op_name (str): The operator name or operator name prefix.
is_prefix (bool): `True` if the op_name is prefix, else `False`.
Default: True.
Returns:
bool, `True` if the operator name does exist in framework file, else
`False`.
"""
if not op_name:
raise ProfilerParamValueErrorException('The op_name should exist.')
for full_op_name in self._task_id_full_op_name_dict.values():
if full_op_name:
if is_prefix and full_op_name.startswith(op_name):
return True
if not is_prefix and op_name == full_op_name:
return True
return False