codegen_opts.target_triple = flags.target_triple;
  // Set the XLA Runtime bit if this is an HloLowering.
  if (!flags.mlir_components.empty() && flags.mlir_components != "None") {
    for (auto component : absl::StrSplit(flags.mlir_components, ',')) {
      if (component == "HloLowering") {
        codegen_opts.use_xla_runtime = true;
      }
    }
  }

  if (flags.cpp_class.empty()) {

    tf_library(
        name = bench_name,
        testonly = 1,
        config = config_file,
        cpp_class = "foo::bar::MatMulComp",
        graph = "test_graph_tfmatmul.pb",
        mlir_components = "HloLowering",  # XLA:CPU-Next only.
        tags = [
            "manual",
            "no_mac",  # TODO(b/228273415)
        ],
        xla_flags = xla_flags,
    ),

  const xla::Shape& muladd_result1 =
      ShapeUtil::GetTupleElementShape(muladd_result, 1);
  EXPECT_TRUE(ShapeUtil::Compatible(muladd_result1, f32_2x2));
}

// tf_compile with mlir_component=HloLowering does not currently support
// profiling, so we disable the test case here rather than creating a new test
// target that could allow more divergence.
#if !defined(MHLO_LOWERING_TEST)
TEST(TFCompileTest, HloProfiling) {