Minor changes to address feedback.

Author: Andy-Jost

cuda_core/cuda/core/experimental/__init__.py

@@ -17,6 +17,7 @@
 from cuda.core.experimental._memory import (
     Buffer,
     DeviceMemoryResource,
+    DeviceMemoryResourceOptions,
     LegacyPinnedMemoryResource,
     MemoryResource,
 )

cuda_core/tests/memory_ipc/conftest.py

@@ -2,7 +2,6 @@
 # SPDX-License-Identifier: Apache-2.0
 
 import pytest
-
 from cuda.core.experimental import Device, DeviceMemoryResource
 
 POOL_SIZE = 2097152

cuda_core/tests/memory_ipc/test_leaks.py

@@ -7,7 +7,6 @@
 
 import psutil
 import pytest
-
 from cuda.core.experimental import _memory
 from cuda.core.experimental._utils.cuda_utils import driver
 

cuda_core/tests/memory_ipc/test_memory_ipc.py

@@ -3,9 +3,8 @@
 
 import multiprocessing as mp
 
-from utility import IPCBufferTestHelper
-
 from cuda.core.experimental import Buffer, Device, DeviceMemoryResource
+from utility import IPCBufferTestHelper
 
 CHILD_TIMEOUT_SEC = 4
 NBYTES = 64

cuda_core/tests/memory_ipc/test_send_buffers.py

@@ -4,9 +4,8 @@
 import multiprocessing
 from itertools import cycle
 
-from utility import IPCBufferTestHelper
-
 from cuda.core.experimental import Device, DeviceMemoryResource
+from utility import IPCBufferTestHelper
 
 CHILD_TIMEOUT_SEC = 4
 NBYTES = 64

cuda_core/tests/memory_ipc/test_serialize.py

@@ -5,9 +5,8 @@
 import multiprocessing.reduction
 import os
 
-from utility import IPCBufferTestHelper
-
 from cuda.core.experimental import Buffer, Device, DeviceMemoryResource
+from utility import IPCBufferTestHelper
 
 CHILD_TIMEOUT_SEC = 4
 NBYTES = 64

cuda_core/tests/memory_ipc/test_workerpool.py

@@ -4,9 +4,8 @@
 import multiprocessing
 from itertools import cycle
 
-from utility import IPCBufferTestHelper
-
 from cuda.core.experimental import Buffer, Device, DeviceMemoryResource
+from utility import IPCBufferTestHelper
 
 CHILD_TIMEOUT_SEC = 4
 NBYTES = 64