Method LoadHsaco

tensorflow/stream_executor/rocm/rocm_driver.cc:467–492

static */

}

/* static */ bool GpuDriver::LoadHsaco(GpuContext* context,
                                       const char* hsaco_contents,
                                       hipModule_t* module) {
  absl::Notification notification;
  bool ret = true;
  GetDriverExecutor()->Schedule(
      [context, hsaco_contents, module, &ret, &notification]() {
        ScopedActivateContext activation{context};
        void* hsaco_data = const_cast<char*>(hsaco_contents);

        hipError_t res =
            tensorflow::wrap::hipModuleLoadData(module, hsaco_data);

        if (res != hipSuccess) {
          LOG(ERROR) << "failed to load HSACO: " << ToString(res);
          ret = false;
          notification.Notify();
        }

        CHECK(module != nullptr);
        notification.Notify();
      });
  notification.WaitForNotification();

  return ret;
}

/* static */ port::Status GpuDriver::SynchronousMemsetUint8(
    GpuContext* context, hipDeviceptr_t location, uint8 value, size_t size) {