Implements IPC-enabled memory pools for Linux in DeviceMemoryResource #930

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Merged

leofang merged 26 commits into NVIDIA:main from Andy-Jost:ipc-mempool-linux

Sep 17, 2025

cuda_core/cuda/core/experimental/__init__.py

-Original file line number
+Diff line change
@@ Expand Up / @@ -14,7 +14,13 @@ @@
     from cuda.core.experimental._launch_config import LaunchConfig
     from cuda.core.experimental._launcher import launch
     from cuda.core.experimental._linker import Linker, LinkerOptions
-    from cuda.core.experimental._memory import Buffer, DeviceMemoryResource, LegacyPinnedMemoryResource, MemoryResource
+    from cuda.core.experimental._memory import (
+        Buffer,
+        DeviceMemoryResource,
+        IPCChannel,
+        LegacyPinnedMemoryResource,
+        MemoryResource,
+    )
     from cuda.core.experimental._module import Kernel, ObjectCode
     from cuda.core.experimental._program import Program, ProgramOptions
     from cuda.core.experimental._stream import Stream, StreamOptions
@@ Expand Down @@

Provide feedback