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