APIs

PyTorch Native APIs

Most of PyTorch native functions can be used on RBLN NPU.

RBLN Specific APIs

The following functions are RBLN NPU-specific functions defined in the torch.rbln module.

Functions

get_amp_supported_dtype()

Get a list of data types supported by automatic mixed precision (AMP) on RBLN devices.

Returns:

Type	Description
List[dtype]	List[torch.dtype]: A list of data types supported by AMP.

Note

This function currently returns only torch.float16.

is_available()

Check if any RBLN devices are available.

Returns:

Name	Type	Description
bool	bool	True if at least one RBLN device is available, False otherwise.