Project import generated by Copybara.

GitOrigin-RevId: 5b23708185311ae39a8605b0c2eff721e7b4939f
2020-08-05 20:27:31 -04:00
parent bdfdaef305
commit 2f86a459b6
115 changed files with 5242 additions and 251 deletions
@@ -575,7 +575,7 @@ CalculatorGraph::PrepareGpu(const std::map<std::string, Packet>& side_packets) {
    // Set up executors.
    for (auto& node : *nodes_) {
      if (node.UsesGpu()) {
-        gpu_resources->PrepareGpuNode(&node);
+        MP_RETURN_IF_ERROR(gpu_resources->PrepareGpuNode(&node));
      }
    }
    for (const auto& name_executor : gpu_resources->GetGpuExecutors()) {