// -----

// Test that islands inside a function with `_skip_island_outlining = true` are
// skipped from outlining for both single-core and replicated case (i.e., the
// `_tpu_v1_compat_outlined` module must be empty and no `PartitionedCallOp` is
// created). Also check that `_skip_island_outlining` attribute is removed.

// CHECK-LABEL: @func5
// CHECK-NOT: _skip_island_outlining
// CHECK-NOT: PartitionedCallOp

    tf_executor.fetch
  }
  func.return
}

// Ensures that these islands are not coarsened (due to caller above) and that
// `_skip_island_outlining` is set to true.
// CHECK-LABEL: func @tpu_partitioned_call_reachable() attributes {_skip_island_outlining = true}
func.func @tpu_partitioned_call_reachable() {
// CHECK-COUNT-4: tf_executor.island
// CHECK-NOT: tf_executor.island
  tf_executor.graph {

// island outlining. The attribute is set in
// `TpuV1BridgeExecutorIslandCoarsening` and removed in the subsequent
// `TPUBridgeExecutorIslandOutlining` pass.
inline constexpr llvm::StringRef kSkipIslandOutlining =
    "_skip_island_outlining";
// Function attribute to signal which argument contains bounded dynamic
// dimension.
inline constexpr llvm::StringRef kDynamicArgIndexAttr = "_dynamic_arg_index";

  getOperation().walk([&](IslandOp island_op) {
    auto parent_func = island_op->getParentOfType<func::FuncOp>();
    auto skip_island_outlining =
        parent_func->getAttrOfType<BoolAttr>(mlir::TF::kSkipIslandOutlining);
    if (skip_island_outlining && skip_island_outlining.getValue()) {
      // Island was marked to be skipped.
      return WalkResult::advance();
    }