(self, arg, typecode, shape=None)
| 423 | dev_type = None |
| 424 | |
| 425 | def __init__(self, arg, typecode, shape=None): |
| 426 | super().__init__(arg, typecode, shape) |
| 427 | self.has_dlpack = hasattr(self.array, "__dlpack_device__") |
| 428 | # TODO(leofang): test CUDA managed memory? |
| 429 | if cupy.cuda.runtime.is_hip: |
| 430 | self.dev_type = dlpack.DLDeviceType.kDLROCM |
| 431 | else: |
| 432 | self.dev_type = dlpack.DLDeviceType.kDLCUDA |
| 433 | |
| 434 | def __dlpack_device__(self): |
| 435 | if self.has_dlpack: |