[Feature] Requesting peer2peer access check for PVC for Cupy equivalents

[abagusetty]

Hi,

Just wanted to check if P2P access exists in dpctl. If not, it would be great to ave a feature for sycl_ext_oneapi_peer_access and more importantly copy memory between device/tiles when possible

The idea is to have some access features like:
https://docs.cupy.dev/en/stable/reference/generated/cupy.cuda.runtime.deviceCanAccessPeer.html
https://docs.cupy.dev/en/stable/reference/generated/cupy.cuda.runtime.deviceEnablePeerAccess.html

[ndgrigorian]

@abagusetty
dpctl currently does not support this feature, but this would be a good enhancement. Since it's a supported extension I don't see why we couldn't or shouldn't try to support it.

Thank you for the suggestion!

[ndgrigorian]

@abagusetty
I've opened a draft PR for adding the feature, #2077

It seems to work on PVC and an AMD machine with multiple devices. The compiler tests use q.copy, which we don't currently expose, instead of memcpy. I plan to add that in a follow up. Let me know if you take the branch for a spin.

[abagusetty]

@ndgrigorian Thanks for the super quick dev. Will give it a try and keep you posted.

[abagusetty]

@ndgrigorian P2P discovery worked on my end for Aurora PVCs. Thanks!

[ndgrigorian]

The extension is supported and now merged into master branch, closing this as implemented