Luke Boyer <******@****.***> 1715148352 -0700

Luke Boyer <******@****.***> 1714601324 -0700

#include "tensorflow/compiler/mlir/lite/stablehlo/transforms/passes.h.inc"

// Undo the MHLO::BroadcastInDimOp folding pattern on splat tensor.
// TODO(b/295966255): Remove this pass after moving MHLO folders to a separate
// pass and folders are not applied by default.
class UnfoldSplatConstantPass
    : public impl::UnfoldSplatConstantPassBase<UnfoldSplatConstantPass> {
 public:
  void runOnOperation() override {

        val intermediateModelProvider = project.serviceOf<IntermediateToolingModelProvider>()
        val builders = project.serviceOf<ToolingModelBuilderRegistry>()

        builders.register(KotlinBuildScriptModelBuilder)
        builders.register(KotlinBuildScriptTemplateModelBuilder)
        builders.register(IsolatedScriptsModelBuilder)

        if (project.parent == null) {

    }

    def "applies excludes/includes for src folders"() {
        given:
        def patterns = ["**/*.xml"]
        javaTree = dirTree("java", [], patterns)
        resourcesTree = dirTree("resources", patterns, [])
        when:
        def folders = regularSourceFolders()
        then:
        folders.find { it.dir.path.endsWith("java") }.excludes == []
        folders.find { it.dir.path.endsWith("java") }.includes == patterns

    pluginsRuntimeOnly(project(":resources-http"))
    pluginsRuntimeOnly(project(":build-cache-http"))

    pluginsRuntimeOnly(project(":tooling-api-builders"))
    pluginsRuntimeOnly(project(":declarative-dsl-tooling-builders"))
    pluginsRuntimeOnly(project(":kotlin-dsl-tooling-builders"))

    pluginsRuntimeOnly(project(":test-kit"))
    pluginsRuntimeOnly(project(":unit-test-fixtures"))

    @JvmField
    @Rule
    val tempFolder = TestNameTestDirectoryProvider(javaClass)

    val root: File
        get() = tempFolder.testDirectory

    fun withFolders(folders: FoldersDslExpression) =
        root.withFolders(folders)

    fun folder(path: String): File =
        existing(path).apply {
            assert(isDirectory)
        }

    fun file(path: String): File =
        existing(path).apply {

            Map<String, Builder> builders) {
        this.eventCatapult = eventCatapult;
        this.defaultLifeCycles = defaultLifeCycles;
        this.buildListCalculator = buildListCalculator;
        this.lifecycleDebugLogger = lifecycleDebugLogger;
        this.lifecycleTaskSegmentCalculator = lifecycleTaskSegmentCalculator;
        this.builders = builders;
    }

    @Override

    ],
    tblgen = "@llvm-project//mlir:mlir-tblgen",
    td_file = "transforms/shlo_simplify.td",
    deps = ["@stablehlo//:stablehlo_ops_td_files"],
)

cc_library(
    name = "folders",
    srcs = ["folders.cc"],
    hdrs = ["folders.h"],
    compatible_with = get_compatible_with_portable(),
    deps = [
        "@llvm-project//llvm:Support",
        "@llvm-project//mlir:IR",
        "@llvm-project//mlir:Support",

    }

    private <T extends DependencySpecBuilder> T add(T builder) {
        builders.add(builder);
        return builder;
    }

    private Set<DependencySpec> dependencySpecSet() {
        ImmutableSet.Builder<DependencySpec> specs = ImmutableSet.builder();
        for (DependencySpecBuilder specBuilder : builders) {
            specs.add(specBuilder.build());
        }
        return specs.build();
    }