tensorplay.is_autocast_available

tensorplay.is_autocast_available(device_type: str) → bool[source]

Return a bool indicating if autocast is available on device_type.

Parameters:

device_type (str) – Device type to use. Possible values are: ‘cuda’, ‘cpu’. The type is the same as the type attribute of a tensorplay.device. Thus, you may obtain the device type of a tensor using Tensor.device.type.