warp.dtype_from_paddle

warp.dtype_from_paddle(paddle_dtype)

Return the Warp dtype corresponding to a Paddle dtype.

Parameters:

paddle_dtype – A paddle.dtype that has a corresponding Warp data type. Currently paddle.bfloat16, paddle.complex64, and paddle.complex128 are not supported.

Raises:

TypeError – Unable to find a corresponding Warp data type.