Repository navigation
[Feature] Requesting peer2peer access check for PVC for Cupy equivalents #2073
Description
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
Activity
ndgrigorian commentedon May 2, 2025
@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 commentedon May 5, 2025
@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 ofmemcpy. I plan to add that in a follow up. Let me know if you take the branch for a spin.abagusetty commentedon May 6, 2025
@ndgrigorian Thanks for the super quick dev. Will give it a try and keep you posted.
abagusetty commentedon May 8, 2025
@ndgrigorian P2P discovery worked on my end for Aurora PVCs. Thanks!
ndgrigorian commentedon May 15, 2025
The extension is supported and now merged into master branch, closing this as implemented