// WITH_FIR_TEST_COMPILER_PLUGIN
// DUMP_IR

// MODULE: lib
// MODULE_KIND: LibraryBinary
// FILE: p3/foo.kt
package p3

import org.jetbrains.kotlin.fir.plugin.MyComposable

@MyComposable
fun Scaffold(topBar: @MyComposable () -> Unit, bottomBar: @MyComposable () -> Unit) {
}

// MODULE: main(lib)
// FILE: main.kt
import org.jetbrains.kotlin.fir.plugin.MyComposable
import p3.Scaffold

@MyComposable

// WITH_FIR_TEST_COMPILER_PLUGIN
// DUMP_IR

// FILE: data/PodcastWithExtraInfo.kt
package data

class PodcastWithExtraInfo {
    lateinit var url: String
    var lastEpisodeDate: kotlin.time.TimeSource? = null
    operator fun component1() = url
    operator fun component2() = lastEpisodeDate
}

// FILE: main.kt
package home

import data.PodcastWithExtraInfo

fun preview(featuredPodcast: PodcastWithExtraInfo) {

// WITH_FIR_TEST_COMPILER_PLUGIN
// DUMP_IR
// CHECK_CALLS_WITH_ANNOTATION: org.jetbrains.kotlin.fir.plugin.MyComposable

// MODULE: main
// FILE: main.kt
import org.jetbrains.kotlin.fir.plugin.MyComposable
import p3.foo

@MyComposable
fun Greeting(): String {
    return "Hi $foo!"
}

// FILE: p3/foo.kt
package p3

import org.jetbrains.kotlin.fir.plugin.MyComposable

private var foo_ = 0

fun setFoo(newFoo: Int) {

// WITH_FIR_TEST_COMPILER_PLUGIN
// DUMP_IR

// MODULE: lib
// MODULE_KIND: LibraryBinary
// FILE: p3/foo.kt
package p3

import org.jetbrains.kotlin.fir.plugin.MyComposable

@MyComposable
fun Scaffold(x: @MyComposable () -> (@MyComposable () -> Unit)) {
}

// MODULE: main(lib)
// FILE: main.kt
import org.jetbrains.kotlin.fir.plugin.MyComposable
import p3.Scaffold

@MyComposable
private fun TopAppBar(title: String) {

// WITH_FIR_TEST_COMPILER_PLUGIN
// DUMP_IR

// MODULE: lib
// MODULE_KIND: LibraryBinary
// FILE: p3/foo.kt
package p3

import org.jetbrains.kotlin.fir.plugin.MyComposable

@MyComposable
fun Scaffold(bars: List<@MyComposable () -> Unit>) {
}

// MODULE: main(lib)
// FILE: main.kt
import org.jetbrains.kotlin.fir.plugin.MyComposable
import p3.Scaffold

@MyComposable
private fun TopAppBar(title: String) {
}

  std::unique_ptr<llvm::raw_ostream> os;
  std::string filepath;
  if (tensorflow::CreateFileForDumping(filename, &os, &filepath).ok()) {
    print_callback(*os);
    LOG(INFO) << "Dumped MLIR module to " << filepath;
  }
}

void EnableIrPrinting(mlir::PassManager &pm,
                      absl::string_view file_name_prefix);

// If verbosity level >= 1, this will dump intermediate IRs of passes to a file.
// The dumped mlir files with be under a directory determined by
// the TF_QUANT_MLIR_DUMP_PREFIX env variable. The PassManager will dump to a
// new file for each pass. The file name will have the format

//
// If auto-clustering is enabled this should produce files named
// before_mark_for_compilation_<N>.pbtxt in the temporary directory.  As the
// file name suggests, these are graphdefs that have been dumped right before
// the mark_for_compilation pass.  There should be one
// before_mark_for_compilation_<N>.pbtxt for every TF graph that was
// auto-clustered, out of which usually only one is the "main" graph that's

        <option value="$PROJECT_DIR$/plugins/atomicfu/atomicfu-compiler/testData" />
        <option value="$PROJECT_DIR$/plugins/fir-plugin-prototype/testData" />
        <option value="$PROJECT_DIR$/plugins/imports-dumper/testData" />
        <option value="$PROJECT_DIR$/plugins/jvm-abi-gen/testData" />
        <option value="$PROJECT_DIR$/plugins/kapt3/kapt3-base/testData" />
        <option value="$PROJECT_DIR$/plugins/kapt3/kapt3-cli/testData" />

// The full pipeline of converting jax random include 2 steps.
// 1. Rename the jax random functions to tflite wrapped functions with the aid
//    of "jax.named_call". For example, in the dumped hlo, the
//    jax.random.uniform will have name "tfl_wrapped_jax_random_uniform".
// 2. Replace the body of "tfl_wrapped_jax_random_uniform" and
//    "tfl_wrapped_jax_random_normal" with tfl.CustomOp("RandomUniform") and