mindspore.hal.is_initialized
- mindspore.hal.is_initialized(device_target)[源代码]
返回指定后端是否已被初始化。
说明
CPU、GPU以及Ascend后端,分别为在如下场景被初始化:
分布式任务中,后端会在调用 mindspore.communication.init 后初始化。
单卡任务中,会在执行第一个算子或者调用创建流/事件接口后被初始化。
- 参数:
device_target (str) - 用户指定的后端类型,必须是
"CPU"
,"GPU"
以及"Ascend"
的其中一个。
- 返回:
bool,指定后端是否已被初始化。
样例:
>>> import mindspore as ms >>> import numpy as np >>> from mindspore import Tensor, ops >>> ms.context.set_context(device_target="CPU") >>> assert not ms.hal.is_initialized("CPU") >>> a = Tensor(np.ones([1, 2]), ms.float32) >>> b = Tensor(np.ones([1, 2]), ms.float32) >>> c = ops.add(a, b).asnumpy() >>> print(ms.hal.is_initialized("CPU")) True