EXPECT_FALSE(UsePjRtForSingleDeviceCompilation(DeviceType(DEVICE_CPU)));
}

TEST(XlaCompileUtilTest, PjRtDeviceCompilerResourceName) {
  EXPECT_EQ(GetPjRtDeviceCompilerResourceName(DeviceType(DEVICE_TPU)),
            "pjrt_device_compiler_TPU");
  EXPECT_EQ(GetPjRtDeviceCompilerResourceName(DeviceType(DEVICE_TPU_NODE)),
            "pjrt_device_compiler_TPU");

  // for CPU/GPU. This is to make sure the DeviceCompiler's lifecycle is
  // maintained appropriately.
  ResourceMgr* rm = nullptr;
  if (device_type == DEVICE_TPU) {
    rm = tfrt_global::GetTFGlobalResourceMgr();
  } else {
    rm = ctx.resource_manager();
  }

  if (!rm) {
    return absl::InternalError("No resource manager found.");
  }
  return rm;
}

  std::vector<xla::XlaOp> returns(1);
  return BuildHloFromGraph(graph, builder, mlir_context, xla_params, returns,
                           use_output_shapes, /*args=*/{},
                           /*control_rets=*/{}, DEVICE_TPU,
                           FunctionLibraryDefinition(OpRegistry::Global()),
                           /*debug_info=*/{},
                           /*custom_legalization_passes=*/{});
}