switch the attribute retreival

Author: ksimpson-work

cuda_core/cuda/core/experimental/_device.py

@@ -67,7 +67,7 @@ def __new__(cls, device_id=None):
                     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 handle_return(cudart.cudaGetDeviceProperties(dev_id))['CU_DEVICE_ATTRIBUTE_MEMORY_POOLS_SUPPORTED'] == 0:
+                    if (handle_return(cudart.cudaGetDeviceProperties(dev_id))).memoryPoolsSupported == 0:
                         dev._mr = _AsyncMemoryResource(dev_id)
                     else:
                         dev._mr = _DefaultAsyncMempool(dev_id)