Skip to content

Commit 91204ff

Browse files
committed
test(cuda.core): use Options dataclasses instead of dicts in MR tests
1 parent 5c0e92d commit 91204ff

1 file changed

Lines changed: 6 additions & 6 deletions

File tree

cuda_core/tests/test_memory.py

Lines changed: 6 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -1981,11 +1981,11 @@ def test_mempool_attributes_repr(memory_resource_factory):
19811981
device.set_current()
19821982

19831983
if MR is DeviceMemoryResource:
1984-
mr = MR(device, options={"max_size": 2048})
1984+
mr = MR(device, options=DeviceMemoryResourceOptions(max_size=2048))
19851985
elif MR is PinnedMemoryResource:
1986-
mr = MR(options={"max_size": 2048})
1986+
mr = MR(options=PinnedMemoryResourceOptions(max_size=2048))
19871987
elif MR is ManagedMemoryResource:
1988-
mr = create_managed_memory_resource_or_skip(options={})
1988+
mr = create_managed_memory_resource_or_skip(options=ManagedMemoryResourceOptions())
19891989

19901990
buffer1 = mr.allocate(64, stream=device.default_stream)
19911991
buffer2 = mr.allocate(64, stream=device.default_stream)
@@ -2018,11 +2018,11 @@ def test_mempool_attributes_ownership(memory_resource_factory):
20182018
device.set_current()
20192019

20202020
if MR is DeviceMemoryResource:
2021-
mr = MR(device, {"max_size": POOL_SIZE})
2021+
mr = MR(device, DeviceMemoryResourceOptions(max_size=POOL_SIZE))
20222022
elif MR is PinnedMemoryResource:
2023-
mr = MR({"max_size": POOL_SIZE})
2023+
mr = MR(PinnedMemoryResourceOptions(max_size=POOL_SIZE))
20242024
elif MR is ManagedMemoryResource:
2025-
mr = create_managed_memory_resource_or_skip({})
2025+
mr = create_managed_memory_resource_or_skip(ManagedMemoryResourceOptions())
20262026

20272027
attributes = mr.attributes
20282028
mr.close()

0 commit comments

Comments
 (0)