Skip to content

Commit d6c311a

Browse files
rparolincpcloud
authored andcommitted
formatting change
1 parent 99ff204 commit d6c311a

1 file changed

Lines changed: 2 additions & 6 deletions

File tree

cuda_core/tests/test_tensor_map.py

Lines changed: 2 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -375,9 +375,7 @@ def test_replace_address_accepts_managed_buffer_on_nonzero_device(self, init_cud
375375
data_type=TensorMapDataType.FLOAT32,
376376
)
377377

378-
mr = create_managed_memory_resource_or_skip(
379-
ManagedMemoryResourceOptions(preferred_location=dev1.device_id)
380-
)
378+
mr = create_managed_memory_resource_or_skip(ManagedMemoryResourceOptions(preferred_location=dev1.device_id))
381379
managed_buf = mr.allocate(1024 * 4)
382380

383381
desc.replace_address(managed_buf)
@@ -417,9 +415,7 @@ def test_from_tiled_accepts_managed_buffer_on_nonzero_device(self, init_cuda):
417415
skip_if_managed_memory_unsupported(dev1)
418416

419417
dev1.set_current()
420-
mr = create_managed_memory_resource_or_skip(
421-
ManagedMemoryResourceOptions(preferred_location=dev1.device_id)
422-
)
418+
mr = create_managed_memory_resource_or_skip(ManagedMemoryResourceOptions(preferred_location=dev1.device_id))
423419
managed_buf = mr.allocate(1024 * 4)
424420

425421
desc = TensorMapDescriptor.from_tiled(

0 commit comments

Comments
 (0)