xxq250
/
mindspore
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

!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
parent 962ccc645d 9b978db51e
commit c81e7af041
No known key found for this signature in database
GPG Key ID: 173E9B9CA92EEF8F
1 changed files with 2 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
mindspore/python/mindspore/profiler/profiling.py
View File

@ -1441,7 +1441,8 @@ class Profiler:
key = name if name.startswith("hcom_") else (name, ts) key = name if name.startswith("hcom_") else (name, ts)
launch_op = kernel_map.get(key) launch_op = kernel_map.get(key)
if not launch_op: 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 continue
launch_ops[index] = launch_op.name launch_ops[index] = launch_op.name
return launch_ops return launch_ops