// in a `tf_device.launch` op with the same `device` attribute.
std::unique_ptr<OperationPass<func::FuncOp>>
CreateDeviceAttributeToLaunchPass();

// Creates a pass that hoists a `tf_device.launch` body and assigns a `device`
// attribute to each TensorFlow dialect op in the body based on the `device`
// attribute on the `tf_device.launch`.

        "tf_device.launch"() ({
          "tf.opA"() : () -> ()
          tf_device.return
        }) {device = "CORE_0"} : () -> ()
        tf_device.return
      }
      tf_executor.yield
    }
    tf_executor.fetch
  }
  func.return
}

// CHECK: "tf_device.launch"
// CHECK: device = "CORE_0"
// CHECK: "tf.opA"
// CHECK: "tf_device.launch"
// CHECK: device = "CORE_0"
// CHECK: "tf.opA"

        expect:
        runBuild { launcher -> launcher.addArguments() }
    }

    def "Adding null argument throws NPE"() {
        when:
        runBuild { launcher -> launcher.addArguments(null as String) }

        then:
        thrown(NullPointerException)

        when:
        runBuild { launcher -> launcher.addArguments(null as List) }

        then:

        def result = withBuild { BuildLauncher launcher ->
            launcher.setJvmArguments("-Dorg.gradle.workers.max=12")
        }

        then:
        result.standardOutput.contains("max workers: 12")
    }

    def "command-line arguments take precedence over system properties"() {
        when:
        def result = withBuild { BuildLauncher launcher ->

 */
package org.gradle.launcher.daemon.server.exec;

import org.gradle.api.logging.Logger;
import org.gradle.api.logging.Logging;
import org.gradle.launcher.daemon.diagnostics.DaemonDiagnostics;
import org.gradle.launcher.daemon.protocol.Build;
import org.gradle.launcher.daemon.protocol.BuildStarted;
import org.gradle.launcher.daemon.protocol.DaemonUnavailable;
import org.gradle.launcher.daemon.protocol.Failure;

// CHECK:      tf_device.parallel_execute
// CHECK:      tf_device.launch
// CHECK:      <{device = "/TPU:1"}>
// CHECK:      tf.OpA
// CHECK:      tf_device.launch
// CHECK:      <{device = "/TPU:2"}>
// CHECK:      tf.OpB
// CHECK:      _parallel_execution_ids = "r0:0"
// CHECK:      tf_executor.island
// CHECK:      tf_device.parallel_execute
// CHECK:      tf_device.launch
// CHECK:      <{device = "/TPU:1"}>
// CHECK:      tf.OpA

}
```
### `-tf-extract-tpu-copy-with-dynamic-shape-op`

_Extract the TPUCopyWithDynamicShapeOp out of the host launch and place it on device launch_

This pass looks for TPUCopyWithDynamicShapeOp which wraps in a
`tf_device.launch` with host device attribute. It extracts the ops and wrap
them in `tf_device.launch` with tpu device attribute so that ops can be
run on TPU instead of CPU while still being compiled on host.

import org.junit.platform.launcher.Launcher
import org.junit.platform.launcher.LauncherDiscoveryRequest
import org.junit.platform.launcher.PostDiscoveryFilter
import org.junit.platform.launcher.TestExecutionListener
import org.junit.platform.launcher.core.EngineDiscoveryOrchestrator
import org.junit.platform.launcher.core.LauncherConfig
import org.junit.platform.launcher.core.LauncherDiscoveryRequestBuilder

    def executor = Mock(AsyncConsumerActionExecutor)
    def connection = Mock(ConsumerConnection)
    def launcher = new DefaultTestLauncher(executor, Stub(ConnectionParameters))

    def "adding tests does not affect an operation in progress"() {
        given:
        launcher.withJvmTestClasses("test")
        when:
        launcher.run()

        then:

import org.gradle.launcher.daemon.logging.DaemonMessages;
import org.gradle.launcher.daemon.protocol.Build;
import org.gradle.launcher.daemon.server.api.DaemonCommandExecution;
import org.gradle.launcher.daemon.server.stats.DaemonRunningStats;
import org.gradle.launcher.exec.BuildActionExecutor;
import org.gradle.launcher.exec.BuildActionParameters;
import org.gradle.launcher.exec.BuildActionResult;

/**