Wrapper around a pointer to device memory to provide a uniform interface with the RMM DeviceBuffer (at least in terms of the interface used by the CUTLASS Python interface)
| 57 | |
| 58 | |
| 59 | class DevicePtrWrapper: |
| 60 | """ |
| 61 | Wrapper around a pointer to device memory to provide a uniform interface with the RMM DeviceBuffer |
| 62 | (at least in terms of the interface used by the CUTLASS Python interface) |
| 63 | """ |
| 64 | def __init__(self, dev_ptr): |
| 65 | self.dev_ptr = dev_ptr |
| 66 | |
| 67 | @property |
| 68 | def ptr(self): |
| 69 | return self.dev_ptr |
| 70 | |
| 71 | |
| 72 | def _todevice(host_data): |