NASP/cuda-samples
85
0
Fork 0
mirror of https://github.com/NVIDIA/cuda-samples.git synced 2026-10-26 06:38:16 +08:00
Code Issues Packages Projects Releases Wiki Activity

Sync v13.4-public from internal repo

Browse Source
This commit is contained in:
cuda-samples sync
2026-09-09 21:14:33 +00:00
parent b7c5481c55
commit b946e35723
871 changed files with 6843 additions and 321849 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
python/4_DistributedComputing/simpleP2P/requirements.txt
View File

@@ -3,4 +3,4 @@
numpy>=2.3.2
cuda-python>=13.0.0
cuda-core>=1.0.0
cuda-core>=1.2.0

17
python/4_DistributedComputing/simpleP2P/simpleP2P.py
View File

@@ -157,6 +157,15 @@ def run(num_elements=1024 * 1024 * 16):
f"GPU{gpuid[0]}, GPU{gpuid[1]} and CPU Host)..."
)
# DeviceMemoryResource is backed by a device memory pool, which is not
# available on every device.
if not (
dev0.properties.memory_pools_supported
and dev1.properties.memory_pools_supported
):
print("\nDevice memory pools are not supported on this platform.")
return 2
# Allocate on GPU 0 and grant access to GPU 1
dev0.set_current()
mr0 = DeviceMemoryResource(dev0)
@@ -176,6 +185,14 @@ def run(num_elements=1024 * 1024 * 16):
)
# Allocate pinned host memory
# PinnedMemoryResource is backed by a host memory pool, which is not
# available on every device.
if not (
dev0.properties.host_memory_pools_supported
and dev1.properties.host_memory_pools_supported
):
print("\nHost pinned memory pools are not supported on this platform.")
return 2
pinned_mr = PinnedMemoryResource()
h0 = pinned_mr.allocate(buf_size, stream=dev0.default_stream)