tensorflow::GetMarkForCompilationPassFlags()
      ->tf_xla_disable_strict_signature_checks = true;
  tensorflow::GetMarkForCompilationPassFlags()->tf_xla_persistent_cache_prefix =
      "my_test_prefix";
  ::testing::InitGoogleTest(&argc, argv);
  return RUN_ALL_TESTS();

           "If true, entires loaded into the XLA compile cache will not have "
           "their signatures checked strictly. Defaults to false."),
      Flag("tf_xla_persistent_cache_prefix",
           &mark_for_compilation_flags->tf_xla_persistent_cache_prefix,
           "Specifies the persistance cache prefix. Default is "
           "\"xla_compile_cache\""),
      Flag("tf_xla_sparse_core_disable_table_stacking",

  PjRtDeviceExecutablePersistor::Config persistor_config(
      persistent_cache_directory,
      GetMarkForCompilationPassFlags()->tf_xla_disable_strict_signature_checks,
      GetMarkForCompilationPassFlags()->tf_xla_persistent_cache_prefix,
      GetMarkForCompilationPassFlags()->tf_xla_persistent_cache_read_only);

  return new PjRtDeviceCompiler(
      std::make_unique<PjRtDeviceExecutablePersistor>(

  // for debugging. Defaults to false.
  bool tf_xla_disable_strict_signature_checks;

  // Specifies the persistance cache prefix. Default is "xla_compile_cache"
  string tf_xla_persistent_cache_prefix;
};

// Flags associated with XLA Sparse Core.
struct XlaSparseCoreFlags {
  // Max level of division to split input data into minibatches.
  int tf_xla_sparse_core_minibatch_max_division_level;