!69034 cannel pynative host op time warning

Merge pull request !69034 from zangqx/master_pynative_host_time
This commit is contained in:
i-robot 2024-05-05 09:55:52 +00:00 committed by Gitee
commit c81e7af041
No known key found for this signature in database
GPG Key ID: 173E9B9CA92EEF8F
1 changed files with 2 additions and 1 deletions

View File

@ -1441,7 +1441,8 @@ class Profiler:
key = name if name.startswith("hcom_") else (name, ts)
launch_op = kernel_map.get(key)
if not launch_op:
logger.warning(f"Failed to get launch operator for {name}!")
if context.get_context("mode") == context.GRAPH_MODE or not name.startswith("aclnn"):
logger.warning(f"Failed to get launch operator for {name}!")
continue
launch_ops[index] = launch_op.name
return launch_ops