Function AsROCmDevicePtr

tensorflow/stream_executor/rocm/rocm_gpu_executor.cc:80–82

Given const GPU memory, returns a librocm device pointer datatype, suitable for passing directly to librocm APIs. N.B. we must lose constness in order to pass a suitable type to the existing librocm APIs, so the caller should take care to only pass the result of const GPU memory conversions to librocm functions which will honor constness.

// librocm APIs, so the caller should take care to only pass the result of const
// GPU memory conversions to librocm functions which will honor constness.
static hipDeviceptr_t AsROCmDevicePtr(const DeviceMemoryBase& gpu_mem) {
  return const_cast<hipDeviceptr_t>(gpu_mem.opaque());
}

// See description on const version above.
static hipDeviceptr_t AsROCmDevicePtr(DeviceMemoryBase* gpu_mem) {