@@ -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