TF_ASSERT_OK_AND_ASSIGN(auto stats, profiler->GetCompileStats(function));
    XlaJitCompilationActivity expected_activity;
    expected_activity.set_cluster_name(function.name());
    expected_activity.set_compile_count(stats.compile_count);
    expected_activity.set_compile_time_us(4);
    expected_activity.set_cumulative_compile_time_us(
        stats.cumulative_compile_time_us);
    expected_activity.set_used_persistent_cache(false);

          << ")";

  XlaJitCompilationActivity jit_compilation_activity;
  jit_compilation_activity.set_cluster_name(function_name);
  jit_compilation_activity.set_compile_count(it->second.compile_count);
  jit_compilation_activity.set_compile_time_us(compile_time_us);
  jit_compilation_activity.set_cumulative_compile_time_us(
      it->second.cumulative_compile_time_us);