newline

Author: ksimpson-work

cuda_core/cuda/core/experimental/_device.py

@@ -65,8 +65,8 @@ def __new__(cls, device_id=None):
                 for dev_id in range(total):
                     dev = super().__new__(cls)
                     dev._id = dev_id
-                    # If the device is in TCC mode, or does not support memory pools for some other reason, use the AsyncMemoryResource which does not use memory pools.
-                    # The DefaultAsyncMempool uses memory pools, which are not always supported. 
+                    # If the device is in TCC mode, or does not support memory pools for some other reason, 
+                    # use the AsyncMemoryResource which does not use memory pools.
                     if (handle_return(cudart.cudaGetDeviceProperties(dev_id))).memoryPoolsSupported == 0:
                         dev._mr = _AsyncMemoryResource(dev_id)
                     else:

cuda_core/cuda/core/experimental/_memory.py

@@ -326,4 +326,5 @@ def is_host_accessible(self) -> bool:
 
     @property
     def device_id(self) -> int:
-        return self._dev_id
+        return self._dev_id
+