Skip to content

Commit 7e3c468

Browse files
authored
apply nit
1 parent 246e8a1 commit 7e3c468

File tree

1 file changed

+1
-7
lines changed

1 file changed

+1
-7
lines changed

cuda_core/tests/test_launcher.py

Lines changed: 1 addition & 7 deletions
Original file line numberDiff line numberDiff line change
@@ -253,13 +253,7 @@ def test_launch_with_buffers_allocated_by_memory_resource(init_cuda, memory_reso
253253

254254
# Create memory resource
255255
if memory_resource_class == "device_memory_resource":
256-
if (
257-
handle_return(
258-
driver.cuDeviceGetAttribute(
259-
driver.CUdevice_attribute.CU_DEVICE_ATTRIBUTE_MEMORY_POOLS_SUPPORTED, dev.device_id
260-
)
261-
)
262-
) == 1:
256+
if dev.properties.memory_pools_supported:
263257
mr = DeviceMemoryResource(dev.device_id)
264258
else:
265259
mr = _SynchronousMemoryResource(dev.device_id)

0 commit comments

Comments
 (0)