forked from huawei/mindspore2022
Add error log if Profiler object initialized before set context in gpu scene
This commit is contained in:
parent
5bf3b7f53e
commit
d376a9704a
|
|
@ -146,6 +146,10 @@ class Profiler:
|
|||
>>> profiler.analyse()
|
||||
"""
|
||||
if self._device_target and self._device_target == "GPU":
|
||||
if context.get_auto_parallel_context('device_num') > 1 and self._dev_id != get_rank():
|
||||
self._dev_id = get_rank()
|
||||
logger.error('Please check the Profiler object initialized after set_auto_parallel_context() '
|
||||
'and init(). Profiler should be initialized after these code. ')
|
||||
self._gpu_profiler.stop()
|
||||
self._generate_timeline()
|
||||
|
||||
|
|
|
|||
Loading…
Reference in New Issue