warp.device_from_paddle#

warp.device_from_paddle(paddle_device)[source]#

Return the Warp device corresponding to a Paddle device.

Parameters:

paddle_device (Place, CPUPlace, CUDAPinnedPlace, CUDAPlace, or str) – Paddle device identifier

Raises:

RuntimeError – Paddle device does not have a corresponding Warp device

Return type:

warp._src.context.Device