XPUPlace

paddle.device. XPUPlace ( dev_id ) [source]

Return a Baidu Kunlun Place

Parameters

dev_id (int) – Baidu Kunlun device id

Examples

# required: xpu

import paddle
place = paddle.device.XPUPlace(0)