torch.xpu.can_device_access_peer

torch.xpu.can_device_access_peer(device, peer)

Query whether a device can access a peer device’s memory.

Parameters

• device (torch.device or int or str) – selected device.

• peer (torch.device or int or str) – peer device to query access to.

Returns

True if device can access peer, False otherwise.

Return type

bool