torch.xpu.can_device_access_peer

torch.xpu.can_device_access_peer(device, peer) [исходный код]

Проверяет, может ли устройство обращаться к памяти устройства-партнера.

Параметры:

device (torch.device or int or str) – выбранное устройство.
peer (torch.device or int or str) – устройство-партнер, доступ к которому проверяется.

Возвращает:

True, если device может обращаться к peer, в противном случае — False.

Тип возвращаемого значения: