Function CurrentDeviceOrDie

tensorflow/stream_executor/rocm/rocm_driver.cc:148–155

Returns the current device set in HIP. This is done by calling the HIP driver (e.g., this value is not our cached view of the current device).

// Returns the current device set in HIP. This is done by calling the
// HIP driver (e.g., this value is not our cached view of the current device).
static int CurrentDeviceOrDie() {
  int current = -1;
  hipError_t result = tensorflow::wrap::hipGetDevice(&current);
  if (result != hipSuccess) {
    LOG(FATAL) << "failed to query current device: " << ToString(result);
  }
  return current;
}

namespace {