[Mlir-commits] [mlir] b3430ed - Revert "[MLIR][SPIRV] Start module combiner"
Mehdi Amini
llvmlistbot at llvm.org
Fri Oct 30 08:09:55 PDT 2020
Author: Mehdi Amini
Date: 2020-10-30T15:09:21Z
New Revision: b3430ed05fea95b89013538b1e30bb3656ee7f57
URL: https://github.com/llvm/llvm-project/commit/b3430ed05fea95b89013538b1e30bb3656ee7f57
DIFF: https://github.com/llvm/llvm-project/commit/b3430ed05fea95b89013538b1e30bb3656ee7f57.diff
LOG: Revert "[MLIR][SPIRV] Start module combiner"
This reverts commit 316593ce839f05af936e705182747743e4638f3c.
Build is broken with:
TestModuleCombiner.cpp:(.text._ZN12_GLOBAL__N_122TestModuleCombinerPass14runOnOperationEv+0x195): undefined reference to `mlir::spirv::combine(llvm::MutableArrayRef<mlir::spirv::ModuleOp>, mlir::OpBuilder&, llvm::function_ref<void (mlir::spirv::ModuleOp, llvm::StringRef, llvm::StringRef)>)'
Added:
Modified:
mlir/lib/Dialect/SPIRV/CMakeLists.txt
mlir/test/lib/Dialect/SPIRV/CMakeLists.txt
mlir/tools/mlir-opt/mlir-opt.cpp
Removed:
mlir/include/mlir/Dialect/SPIRV/ModuleCombiner.h
mlir/lib/Dialect/SPIRV/Linking/CMakeLists.txt
mlir/lib/Dialect/SPIRV/Linking/ModuleCombiner/CMakeLists.txt
mlir/lib/Dialect/SPIRV/Linking/ModuleCombiner/ModuleCombiner.cpp
mlir/test/Dialect/SPIRV/Linking/ModuleCombiner/basic.mlir
mlir/test/Dialect/SPIRV/Linking/ModuleCombiner/conflict_resolution.mlir
mlir/test/lib/Dialect/SPIRV/TestModuleCombiner.cpp
################################################################################
diff --git a/mlir/include/mlir/Dialect/SPIRV/ModuleCombiner.h b/mlir/include/mlir/Dialect/SPIRV/ModuleCombiner.h
deleted file mode 100644
index b7ecd57d103d..000000000000
--- a/mlir/include/mlir/Dialect/SPIRV/ModuleCombiner.h
+++ /dev/null
@@ -1,69 +0,0 @@
-//===- ModuleCombiner.h - MLIR SPIR-V Module Combiner -----------*- C++ -*-===//
-//
-// Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
-// See https://llvm.org/LICENSE.txt for license information.
-// SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
-//
-//===----------------------------------------------------------------------===//
-//
-// This file declares the entry point to the SPIR-V module combiner library.
-//
-//===----------------------------------------------------------------------===//
-
-#ifndef MLIR_DIALECT_SPIRV_MODULECOMBINER_H_
-#define MLIR_DIALECT_SPIRV_MODULECOMBINER_H_
-
-#include "mlir/Dialect/SPIRV/SPIRVModule.h"
-#include "llvm/ADT/ArrayRef.h"
-#include "llvm/ADT/SmallVector.h"
-
-namespace mlir {
-class OpBuilder;
-
-namespace spirv {
-class ModuleOp;
-
-/// To combine a number of MLIR SPIR-V modules, we move all the module-level ops
-/// from all the input modules into one big combined module. To that end, the
-/// combination process proceeds in 2 phases:
-///
-/// (1) resolve conflicts between pairs of ops from
diff erent modules
-/// (2) deduplicate equivalent ops/sub-ops in the merged module. (TODO)
-///
-/// For the conflict resolution phase, the following rules are employed to
-/// resolve such conflicts:
-///
-/// - If 2 spv.func's have the same symbol name, then rename one of the
-/// functions.
-/// - If an spv.func and another op have the same symbol name, then rename the
-/// other symbol.
-/// - If none of the 2 conflicting ops are spv.func, then rename either.
-///
-/// In all cases, the references to the updated symbol are also updated to
-/// reflect the change.
-///
-/// \param modules the list of modules to combine. Input modules are not
-/// modified.
-/// \param combinedMdouleBuilder an OpBuilder to be used for
-/// building up the combined module.
-/// \param symbRenameListener a listener that gets called everytime a symbol in
-/// one of the input modules is renamed. The arguments
-/// passed to the listener are: the input
-/// spirv::ModuleOp that contains the renamed symbol,
-/// a StringRef to the old symbol name, and a
-/// StringRef to the new symbol name. Note that it is
-/// the responsibility of the caller to properly
-/// retain the storage underlying the passed
-/// StringRefs if the listener callback outlives this
-/// function call.
-///
-/// \return the combined module.
-OwningSPIRVModuleRef
-combine(llvm::MutableArrayRef<ModuleOp> modules,
- OpBuilder &combinedModuleBuilder,
- llvm::function_ref<void(ModuleOp, StringRef, StringRef)>
- symbRenameListener);
-} // namespace spirv
-} // namespace mlir
-
-#endif // MLIR_DIALECT_SPIRV_MODULECOMBINER_H_
diff --git a/mlir/lib/Dialect/SPIRV/CMakeLists.txt b/mlir/lib/Dialect/SPIRV/CMakeLists.txt
index f37182121fed..10f06fdb8861 100644
--- a/mlir/lib/Dialect/SPIRV/CMakeLists.txt
+++ b/mlir/lib/Dialect/SPIRV/CMakeLists.txt
@@ -34,6 +34,5 @@ add_mlir_dialect_library(MLIRSPIRV
MLIRTransforms
)
-add_subdirectory(Linking)
add_subdirectory(Serialization)
add_subdirectory(Transforms)
diff --git a/mlir/lib/Dialect/SPIRV/Linking/CMakeLists.txt b/mlir/lib/Dialect/SPIRV/Linking/CMakeLists.txt
deleted file mode 100644
index 4cc016812701..000000000000
--- a/mlir/lib/Dialect/SPIRV/Linking/CMakeLists.txt
+++ /dev/null
@@ -1 +0,0 @@
-add_subdirectory(ModuleCombiner)
diff --git a/mlir/lib/Dialect/SPIRV/Linking/ModuleCombiner/CMakeLists.txt b/mlir/lib/Dialect/SPIRV/Linking/ModuleCombiner/CMakeLists.txt
deleted file mode 100644
index 22756fab23e5..000000000000
--- a/mlir/lib/Dialect/SPIRV/Linking/ModuleCombiner/CMakeLists.txt
+++ /dev/null
@@ -1,6 +0,0 @@
-add_mlir_dialect_library(MLIRSPIRVModuleCombiner
- ModuleCombiner.cpp
-
- ADDITIONAL_HEADER_DIRS
- ${MLIR_MAIN_INCLUDE_DIR}/mlir/Dialect/SPIRV
- )
diff --git a/mlir/lib/Dialect/SPIRV/Linking/ModuleCombiner/ModuleCombiner.cpp b/mlir/lib/Dialect/SPIRV/Linking/ModuleCombiner/ModuleCombiner.cpp
deleted file mode 100644
index ccdf40c68ff5..000000000000
--- a/mlir/lib/Dialect/SPIRV/Linking/ModuleCombiner/ModuleCombiner.cpp
+++ /dev/null
@@ -1,183 +0,0 @@
-//===- ModuleCombiner.cpp - MLIR SPIR-V Module Combiner ---------*- C++ -*-===//
-//
-// Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
-// See https://llvm.org/LICENSE.txt for license information.
-// SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
-//
-//===----------------------------------------------------------------------===//
-//
-// This file implements the the SPIR-V module combiner library.
-//
-//===----------------------------------------------------------------------===//
-
-#include "mlir/Dialect/SPIRV/ModuleCombiner.h"
-
-#include "mlir/Dialect/SPIRV/SPIRVOps.h"
-#include "mlir/IR/Builders.h"
-#include "mlir/IR/SymbolTable.h"
-#include "llvm/ADT/ArrayRef.h"
-#include "llvm/ADT/StringExtras.h"
-
-using namespace mlir;
-
-static constexpr unsigned maxFreeID = 1 << 20;
-
-static SmallString<64> renameSymbol(StringRef oldSymName, unsigned &lastUsedID,
- spirv::ModuleOp combinedModule) {
- SmallString<64> newSymName(oldSymName);
- newSymName.push_back('_');
-
- while (lastUsedID < maxFreeID) {
- std::string possible = (newSymName + llvm::utostr(++lastUsedID)).str();
-
- if (!SymbolTable::lookupSymbolIn(combinedModule, possible)) {
- newSymName += llvm::utostr(lastUsedID);
- break;
- }
- }
-
- return newSymName;
-}
-
-/// Check if a symbol with the same name as op already exists in source. If so,
-/// rename op and update all its references in target.
-static LogicalResult updateSymbolAndAllUses(SymbolOpInterface op,
- spirv::ModuleOp target,
- spirv::ModuleOp source,
- unsigned &lastUsedID) {
- if (!SymbolTable::lookupSymbolIn(source, op.getName()))
- return success();
-
- StringRef oldSymName = op.getName();
- SmallString<64> newSymName = renameSymbol(oldSymName, lastUsedID, target);
-
- if (failed(SymbolTable::replaceAllSymbolUses(op, newSymName, target)))
- return op.emitError("unable to update all symbol uses for ")
- << oldSymName << " to " << newSymName;
-
- SymbolTable::setSymbolName(op, newSymName);
- return success();
-}
-
-namespace mlir {
-namespace spirv {
-
-// TODO Properly test symbol rename listener mechanism.
-
-OwningSPIRVModuleRef
-combine(llvm::MutableArrayRef<spirv::ModuleOp> modules,
- OpBuilder &combinedModuleBuilder,
- llvm::function_ref<void(ModuleOp, StringRef, StringRef)>
- symRenameListener) {
- unsigned lastUsedID = 0;
-
- if (modules.empty())
- return nullptr;
-
- auto addressingModel = modules[0].addressing_model();
- auto memoryModel = modules[0].memory_model();
-
- auto combinedModule = combinedModuleBuilder.create<spirv::ModuleOp>(
- modules[0].getLoc(), addressingModel, memoryModel);
- combinedModuleBuilder.setInsertionPointToStart(&*combinedModule.getBody());
-
- // In some cases, a symbol in the (current state of the) combined module is
- // renamed in order to maintain the conflicting symbol in the input module
- // being merged. For example, if the conflict is between a global variable in
- // the current combined module and a function in the input module, the global
- // varaible is renamed. In order to notify listeners of the symbol updates in
- // such cases, we need to keep track of the module from which the renamed
- // symbol in the combined module originated. This map keeps such information.
- DenseMap<StringRef, spirv::ModuleOp> symNameToModuleMap;
-
- for (auto module : modules) {
- if (module.addressing_model() != addressingModel ||
- module.memory_model() != memoryModel) {
- module.emitError(
- "input modules
diff er in addressing model and/or memory model");
- return nullptr;
- }
-
- spirv::ModuleOp moduleClone = module.clone();
-
- // In the combined module, rename all symbols that conflict with symbols
- // from the current input module. This renmaing applies to all ops except
- // for spv.funcs. This way, if the conflicting op in the input module is
- // non-spv.func, we rename that symbol instead and maintain the spv.func in
- // the combined module name as it is.
- for (auto &op : combinedModule.getBlock().without_terminator()) {
- auto symbolOp = dyn_cast<SymbolOpInterface>(op);
- if (!symbolOp)
- continue;
-
- StringRef oldSymName = symbolOp.getName();
-
- if (!isa<FuncOp>(op) &&
- failed(updateSymbolAndAllUses(symbolOp, combinedModule, moduleClone,
- lastUsedID)))
- return nullptr;
-
- StringRef newSymName = symbolOp.getName();
-
- if (symRenameListener && oldSymName != newSymName) {
- spirv::ModuleOp originalModule = symNameToModuleMap.lookup(oldSymName);
-
- if (!originalModule) {
- module.emitError("unable to find original ModuleOp for symbol ")
- << oldSymName;
- return nullptr;
- }
-
- symRenameListener(originalModule, oldSymName, newSymName);
-
- // Since the symbol name is updated, there is no need to maintain the
- // entry that assocaites the old symbol name with the original module.
- symNameToModuleMap.erase(oldSymName);
- // Instead, add a new entry to map the new symbol name to the original
- // module in case it gets renamed again later.
- symNameToModuleMap[newSymName] = originalModule;
- }
- }
-
- // In the current input module, rename all symbols that conflict with
- // symbols from the combined module. This includes renaming spv.funcs.
- for (auto &op : moduleClone.getBlock().without_terminator()) {
- auto symbolOp = dyn_cast<SymbolOpInterface>(op);
- if (!symbolOp)
- continue;
-
- StringRef oldSymName = symbolOp.getName();
-
- if (failed(updateSymbolAndAllUses(symbolOp, moduleClone, combinedModule,
- lastUsedID)))
- return nullptr;
-
- StringRef newSymName = symbolOp.getName();
-
- if (symRenameListener && oldSymName != newSymName) {
- symRenameListener(module, oldSymName, newSymName);
-
- // Insert the module associated with the symbol name.
- auto emplaceResult = symNameToModuleMap.try_emplace(newSymName, module);
-
- // If an entry with the same symbol name is already present, this must
- // be a problem with the implementation, specially clean-up of the map
- // while iterating over the combined module above.
- if (!emplaceResult.second) {
- module.emitError("did not expect to find an entry for symbol ")
- << newSymName;
- return nullptr;
- }
- }
- }
-
- // Clone all the module's ops to the combined module.
- for (auto &op : moduleClone.getBlock().without_terminator())
- combinedModuleBuilder.insert(op.clone());
- }
-
- return combinedModule;
-}
-
-} // namespace spirv
-} // namespace mlir
diff --git a/mlir/test/Dialect/SPIRV/Linking/ModuleCombiner/basic.mlir b/mlir/test/Dialect/SPIRV/Linking/ModuleCombiner/basic.mlir
deleted file mode 100644
index 07fd41e4fe86..000000000000
--- a/mlir/test/Dialect/SPIRV/Linking/ModuleCombiner/basic.mlir
+++ /dev/null
@@ -1,50 +0,0 @@
-// RUN: mlir-opt -test-spirv-module-combiner -split-input-file -verify-diagnostics %s | FileCheck %s
-
-// CHECK: module {
-// CHECK-NEXT: spv.module Logical GLSL450 {
-// CHECK-NEXT: spv.specConstant @m1_sc
-// CHECK-NEXT: spv.specConstant @m2_sc
-// CHECK-NEXT: spv.func @variable_init_spec_constant
-// CHECK-NEXT: spv._reference_of @m2_sc
-// CHECK-NEXT: spv.Variable init
-// CHECK-NEXT: spv.Return
-// CHECK-NEXT: }
-// CHECK-NEXT: }
-// CHECK-NEXT: }
-
-module {
-spv.module Logical GLSL450 {
- spv.specConstant @m1_sc = 42.42 : f32
-}
-
-spv.module Logical GLSL450 {
- spv.specConstant @m2_sc = 42 : i32
- spv.func @variable_init_spec_constant() -> () "None" {
- %0 = spv._reference_of @m2_sc : i32
- %1 = spv.Variable init(%0) : !spv.ptr<i32, Function>
- spv.Return
- }
-}
-}
-
-// -----
-
-module {
-spv.module Physical64 GLSL450 {
-}
-
-// expected-error @+1 {{input modules
diff er in addressing model and/or memory model}}
-spv.module Logical GLSL450 {
-}
-}
-
-// -----
-
-module {
-spv.module Logical Simple {
-}
-
-// expected-error @+1 {{input modules
diff er in addressing model and/or memory model}}
-spv.module Logical GLSL450 {
-}
-}
diff --git a/mlir/test/Dialect/SPIRV/Linking/ModuleCombiner/conflict_resolution.mlir b/mlir/test/Dialect/SPIRV/Linking/ModuleCombiner/conflict_resolution.mlir
deleted file mode 100644
index f5535c483171..000000000000
--- a/mlir/test/Dialect/SPIRV/Linking/ModuleCombiner/conflict_resolution.mlir
+++ /dev/null
@@ -1,682 +0,0 @@
-// RUN: mlir-opt -test-spirv-module-combiner -split-input-file -verify-diagnostics %s | FileCheck %s
-
-// Test basic renaming of conflicting funcOps.
-
-// CHECK: module {
-// CHECK-NEXT: spv.module Logical GLSL450 {
-// CHECK-NEXT: spv.func @foo
-// CHECK-NEXT: spv.ReturnValue
-// CHECK-NEXT: }
-
-// CHECK-NEXT: spv.func @foo_1
-// CHECK-NEXT: spv.ReturnValue
-// CHECK-NEXT: }
-// CHECK-NEXT: }
-// CHECK-NEXT: }
-
-module {
-spv.module Logical GLSL450 {
- spv.func @foo(%arg0 : i32) -> i32 "None" {
- spv.ReturnValue %arg0 : i32
- }
-}
-
-spv.module Logical GLSL450 {
- spv.func @foo(%arg0 : f32) -> f32 "None" {
- spv.ReturnValue %arg0 : f32
- }
-}
-}
-
-// -----
-
-// Test basic renaming of conflicting funcOps across 3 modules.
-
-// CHECK: module {
-// CHECK-NEXT: spv.module Logical GLSL450 {
-// CHECK-NEXT: spv.func @foo
-// CHECK-NEXT: spv.ReturnValue
-// CHECK-NEXT: }
-
-// CHECK-NEXT: spv.func @foo_1
-// CHECK-NEXT: spv.ReturnValue
-// CHECK-NEXT: }
-
-// CHECK-NEXT: spv.func @foo_2
-// CHECK-NEXT: spv.ReturnValue
-// CHECK-NEXT: }
-// CHECK-NEXT: }
-// CHECK-NEXT: }
-
-module {
-spv.module Logical GLSL450 {
- spv.func @foo(%arg0 : i32) -> i32 "None" {
- spv.ReturnValue %arg0 : i32
- }
-}
-
-spv.module Logical GLSL450 {
- spv.func @foo(%arg0 : f32) -> f32 "None" {
- spv.ReturnValue %arg0 : f32
- }
-}
-
-spv.module Logical GLSL450 {
- spv.func @foo(%arg0 : i32) -> i32 "None" {
- spv.ReturnValue %arg0 : i32
- }
-}
-}
-
-// -----
-
-// Test properly updating references to a renamed funcOp.
-
-// CHECK: module {
-// CHECK-NEXT: spv.module Logical GLSL450 {
-// CHECK-NEXT: spv.func @foo
-// CHECK-NEXT: spv.ReturnValue
-// CHECK-NEXT: }
-
-// CHECK-NEXT: spv.func @foo_1
-// CHECK-NEXT: spv.ReturnValue
-// CHECK-NEXT: }
-
-// CHECK-NEXT: spv.func @bar
-// CHECK-NEXT: spv.FunctionCall @foo_1
-// CHECK-NEXT: spv.ReturnValue
-// CHECK-NEXT: }
-// CHECK-NEXT: }
-// CHECK-NEXT: }
-
-module {
-spv.module Logical GLSL450 {
- spv.func @foo(%arg0 : i32) -> i32 "None" {
- spv.ReturnValue %arg0 : i32
- }
-}
-
-spv.module Logical GLSL450 {
- spv.func @foo(%arg0 : f32) -> f32 "None" {
- spv.ReturnValue %arg0 : f32
- }
-
- spv.func @bar(%arg0 : f32) -> f32 "None" {
- %0 = spv.FunctionCall @foo(%arg0) : (f32) -> (f32)
- spv.ReturnValue %0 : f32
- }
-}
-}
-
-// -----
-
-// Test properly updating references to a renamed funcOp if the functionCallOp
-// preceeds the callee funcOp definition.
-
-// CHECK: module {
-// CHECK-NEXT: spv.module Logical GLSL450 {
-// CHECK-NEXT: spv.func @foo
-// CHECK-NEXT: spv.ReturnValue
-// CHECK-NEXT: }
-
-// CHECK-NEXT: spv.func @bar
-// CHECK-NEXT: spv.FunctionCall @foo_1
-// CHECK-NEXT: spv.ReturnValue
-// CHECK-NEXT: }
-
-// CHECK-NEXT: spv.func @foo_1
-// CHECK-NEXT: spv.ReturnValue
-// CHECK-NEXT: }
-// CHECK-NEXT: }
-// CHECK-NEXT: }
-
-module {
-spv.module Logical GLSL450 {
- spv.func @foo(%arg0 : i32) -> i32 "None" {
- spv.ReturnValue %arg0 : i32
- }
-}
-
-spv.module Logical GLSL450 {
- spv.func @bar(%arg0 : f32) -> f32 "None" {
- %0 = spv.FunctionCall @foo(%arg0) : (f32) -> (f32)
- spv.ReturnValue %0 : f32
- }
-
- spv.func @foo(%arg0 : f32) -> f32 "None" {
- spv.ReturnValue %arg0 : f32
- }
-}
-}
-
-// -----
-
-// Test properly updating entryPointOp and executionModeOp attached to renamed
-// funcOp.
-
-// CHECK: module {
-// CHECK-NEXT: spv.module Logical GLSL450 {
-// CHECK-NEXT: spv.func @foo
-// CHECK-NEXT: spv.ReturnValue
-// CHECK-NEXT: }
-
-// CHECK-NEXT: spv.func @foo_1
-// CHECK-NEXT: spv.ReturnValue
-// CHECK-NEXT: }
-
-// CHECK-NEXT: spv.EntryPoint "GLCompute" @foo_1
-// CHECK-NEXT: spv.ExecutionMode @foo_1 "ContractionOff"
-// CHECK-NEXT: }
-// CHECK-NEXT: }
-
-module {
-spv.module Logical GLSL450 {
- spv.func @foo(%arg0 : i32) -> i32 "None" {
- spv.ReturnValue %arg0 : i32
- }
-}
-
-spv.module Logical GLSL450 {
- spv.func @foo(%arg0 : f32) -> f32 "None" {
- spv.ReturnValue %arg0 : f32
- }
-
- spv.EntryPoint "GLCompute" @foo
- spv.ExecutionMode @foo "ContractionOff"
-}
-}
-
-// -----
-
-// CHECK: module {
-// CHECK-NEXT: spv.module Logical GLSL450 {
-// CHECK-NEXT: spv.func @foo
-// CHECK-NEXT: spv.ReturnValue
-// CHECK-NEXT: }
-
-// CHECK-NEXT: spv.EntryPoint "GLCompute" @fo
-// CHECK-NEXT: spv.ExecutionMode @foo "ContractionOff"
-
-// CHECK-NEXT: spv.func @foo_1
-// CHECK-NEXT: spv.ReturnValue
-// CHECK-NEXT: }
-
-// CHECK-NEXT: spv.EntryPoint "GLCompute" @foo_1
-// CHECK-NEXT: spv.ExecutionMode @foo_1 "ContractionOff"
-// CHECK-NEXT: }
-// CHECK-NEXT: }
-
-module {
-spv.module Logical GLSL450 {
- spv.func @foo(%arg0 : i32) -> i32 "None" {
- spv.ReturnValue %arg0 : i32
- }
-
- spv.EntryPoint "GLCompute" @foo
- spv.ExecutionMode @foo "ContractionOff"
-}
-
-spv.module Logical GLSL450 {
- spv.func @foo(%arg0 : f32) -> f32 "None" {
- spv.ReturnValue %arg0 : f32
- }
-
- spv.EntryPoint "GLCompute" @foo
- spv.ExecutionMode @foo "ContractionOff"
-}
-}
-
-// -----
-
-// Resolve conflicting funcOp and globalVariableOp.
-
-// CHECK: module {
-// CHECK-NEXT: spv.module Logical GLSL450 {
-// CHECK-NEXT: spv.func @foo
-// CHECK-NEXT: spv.ReturnValue
-// CHECK-NEXT: }
-
-// CHECK-NEXT: spv.globalVariable @foo_1
-// CHECK-NEXT: }
-
-module {
-spv.module Logical GLSL450 {
- spv.func @foo(%arg0 : i32) -> i32 "None" {
- spv.ReturnValue %arg0 : i32
- }
-}
-
-spv.module Logical GLSL450 {
- spv.globalVariable @foo bind(1, 0) : !spv.ptr<f32, Input>
-}
-}
-
-// -----
-
-// Resolve conflicting funcOp and globalVariableOp and update the global variable's
-// references.
-
-// CHECK: module {
-// CHECK-NEXT: spv.module Logical GLSL450 {
-// CHECK-NEXT: spv.func @foo
-// CHECK-NEXT: spv.ReturnValue
-// CHECK-NEXT: }
-
-// CHECK-NEXT: spv.globalVariable @foo_1
-// CHECK-NEXT: spv.func @bar
-// CHECK-NEXT: spv._address_of @foo_1
-// CHECK-NEXT: spv.Load
-// CHECK-NEXT: spv.ReturnValue
-// CHECK-NEXT: }
-// CHECK-NEXT: }
-
-module {
-spv.module Logical GLSL450 {
- spv.func @foo(%arg0 : i32) -> i32 "None" {
- spv.ReturnValue %arg0 : i32
- }
-}
-
-spv.module Logical GLSL450 {
- spv.globalVariable @foo bind(1, 0) : !spv.ptr<f32, Input>
-
- spv.func @bar() -> f32 "None" {
- %0 = spv._address_of @foo : !spv.ptr<f32, Input>
- %1 = spv.Load "Input" %0 : f32
- spv.ReturnValue %1 : f32
- }
-}
-}
-
-// -----
-
-// Resolve conflicting globalVariableOp and funcOp and update the global variable's
-// references.
-
-// CHECK: module {
-// CHECK-NEXT: spv.module Logical GLSL450 {
-// CHECK-NEXT: spv.globalVariable @foo_1
-// CHECK-NEXT: spv.func @bar
-// CHECK-NEXT: spv._address_of @foo_1
-// CHECK-NEXT: spv.Load
-// CHECK-NEXT: spv.ReturnValue
-// CHECK-NEXT: }
-
-// CHECK-NEXT: spv.func @foo
-// CHECK-NEXT: spv.ReturnValue
-// CHECK-NEXT: }
-// CHECK-NEXT: }
-
-module {
-spv.module Logical GLSL450 {
- spv.globalVariable @foo bind(1, 0) : !spv.ptr<f32, Input>
-
- spv.func @bar() -> f32 "None" {
- %0 = spv._address_of @foo : !spv.ptr<f32, Input>
- %1 = spv.Load "Input" %0 : f32
- spv.ReturnValue %1 : f32
- }
-}
-
-spv.module Logical GLSL450 {
- spv.func @foo(%arg0 : i32) -> i32 "None" {
- spv.ReturnValue %arg0 : i32
- }
-}
-}
-
-// -----
-
-// Resolve conflicting funcOp and specConstantOp.
-
-// CHECK: module {
-// CHECK-NEXT: spv.module Logical GLSL450 {
-// CHECK-NEXT: spv.func @foo
-// CHECK-NEXT: spv.ReturnValue
-// CHECK-NEXT: }
-
-// CHECK-NEXT: spv.specConstant @foo_1
-// CHECK-NEXT: }
-
-module {
-spv.module Logical GLSL450 {
- spv.func @foo(%arg0 : i32) -> i32 "None" {
- spv.ReturnValue %arg0 : i32
- }
-}
-
-spv.module Logical GLSL450 {
- spv.specConstant @foo = -5 : i32
-}
-}
-
-// -----
-
-// Resolve conflicting funcOp and specConstantOp and update the spec constant's
-// references.
-
-// CHECK: module {
-// CHECK-NEXT: spv.module Logical GLSL450 {
-// CHECK-NEXT: spv.func @foo
-// CHECK-NEXT: spv.ReturnValue
-// CHECK-NEXT: }
-
-// CHECK-NEXT: spv.specConstant @foo_1
-// CHECK-NEXT: spv.func @bar
-// CHECK-NEXT: spv._reference_of @foo_1
-// CHECK-NEXT: spv.ReturnValue
-// CHECK-NEXT: }
-// CHECK-NEXT: }
-
-module {
-spv.module Logical GLSL450 {
- spv.func @foo(%arg0 : i32) -> i32 "None" {
- spv.ReturnValue %arg0 : i32
- }
-}
-
-spv.module Logical GLSL450 {
- spv.specConstant @foo = -5 : i32
-
- spv.func @bar() -> i32 "None" {
- %0 = spv._reference_of @foo : i32
- spv.ReturnValue %0 : i32
- }
-}
-}
-
-// -----
-
-// Resolve conflicting specConstantOp and funcOp and update the spec constant's
-// references.
-
-// CHECK: module {
-// CHECK-NEXT: spv.module Logical GLSL450 {
-// CHECK-NEXT: spv.specConstant @foo_1
-// CHECK-NEXT: spv.func @bar
-// CHECK-NEXT: spv._reference_of @foo_1
-// CHECK-NEXT: spv.ReturnValue
-// CHECK-NEXT: }
-
-// CHECK-NEXT: spv.func @foo
-// CHECK-NEXT: spv.ReturnValue
-// CHECK-NEXT: }
-// CHECK-NEXT: }
-
-module {
-spv.module Logical GLSL450 {
- spv.specConstant @foo = -5 : i32
-
- spv.func @bar() -> i32 "None" {
- %0 = spv._reference_of @foo : i32
- spv.ReturnValue %0 : i32
- }
-}
-
-spv.module Logical GLSL450 {
- spv.func @foo(%arg0 : i32) -> i32 "None" {
- spv.ReturnValue %arg0 : i32
- }
-}
-}
-
-// -----
-
-// Resolve conflicting funcOp and specConstantCompositeOp.
-
-// CHECK: module {
-// CHECK-NEXT: spv.module Logical GLSL450 {
-// CHECK-NEXT: spv.func @foo
-// CHECK-NEXT: spv.ReturnValue
-// CHECK-NEXT: }
-
-// CHECK-NEXT: spv.specConstant @bar
-// CHECK-NEXT: spv.specConstantComposite @foo_1 (@bar, @bar)
-// CHECK-NEXT: }
-
-module {
-spv.module Logical GLSL450 {
- spv.func @foo(%arg0 : i32) -> i32 "None" {
- spv.ReturnValue %arg0 : i32
- }
-}
-
-spv.module Logical GLSL450 {
- spv.specConstant @bar = -5 : i32
- spv.specConstantComposite @foo (@bar, @bar) : !spv.array<2 x i32>
-}
-}
-
-// -----
-
-// Resolve conflicting funcOp and specConstantCompositeOp and update the spec
-// constant's references.
-
-// CHECK: module {
-// CHECK-NEXT: spv.module Logical GLSL450 {
-// CHECK-NEXT: spv.func @foo
-// CHECK-NEXT: spv.ReturnValue
-// CHECK-NEXT: }
-
-// CHECK-NEXT: spv.specConstant @bar
-// CHECK-NEXT: spv.specConstantComposite @foo_1 (@bar, @bar)
-// CHECK-NEXT: spv.func @baz
-// CHECK-NEXT: spv._reference_of @foo_1
-// CHECK-NEXT: spv.CompositeExtract
-// CHECK-NEXT: spv.ReturnValue
-// CHECK-NEXT: }
-// CHECK-NEXT: }
-
-module {
-spv.module Logical GLSL450 {
- spv.func @foo(%arg0 : i32) -> i32 "None" {
- spv.ReturnValue %arg0 : i32
- }
-}
-
-spv.module Logical GLSL450 {
- spv.specConstant @bar = -5 : i32
- spv.specConstantComposite @foo (@bar, @bar) : !spv.array<2 x i32>
-
- spv.func @baz() -> i32 "None" {
- %0 = spv._reference_of @foo : !spv.array<2 x i32>
- %1 = spv.CompositeExtract %0[0 : i32] : !spv.array<2 x i32>
- spv.ReturnValue %1 : i32
- }
-}
-}
-
-// -----
-
-// Resolve conflicting specConstantCompositeOp and funcOp and update the spec
-// constant's references.
-
-// CHECK: module {
-// CHECK-NEXT: spv.module Logical GLSL450 {
-// CHECK-NEXT: spv.specConstant @bar
-// CHECK-NEXT: spv.specConstantComposite @foo_1 (@bar, @bar)
-// CHECK-NEXT: spv.func @baz
-// CHECK-NEXT: spv._reference_of @foo_1
-// CHECK-NEXT: spv.CompositeExtract
-// CHECK-NEXT: spv.ReturnValue
-// CHECK-NEXT: }
-
-// CHECK-NEXT: spv.func @foo
-// CHECK-NEXT: spv.ReturnValue
-// CHECK-NEXT: }
-// CHECK-NEXT: }
-
-module {
-spv.module Logical GLSL450 {
- spv.specConstant @bar = -5 : i32
- spv.specConstantComposite @foo (@bar, @bar) : !spv.array<2 x i32>
-
- spv.func @baz() -> i32 "None" {
- %0 = spv._reference_of @foo : !spv.array<2 x i32>
- %1 = spv.CompositeExtract %0[0 : i32] : !spv.array<2 x i32>
- spv.ReturnValue %1 : i32
- }
-}
-
-spv.module Logical GLSL450 {
- spv.func @foo(%arg0 : i32) -> i32 "None" {
- spv.ReturnValue %arg0 : i32
- }
-}
-}
-
-// -----
-
-// Resolve conflicting spec constants and funcOps and update the spec constant's
-// references.
-
-// CHECK: module {
-// CHECK-NEXT: spv.module Logical GLSL450 {
-// CHECK-NEXT: spv.specConstant @bar_1
-// CHECK-NEXT: spv.specConstantComposite @foo_2 (@bar_1, @bar_1)
-// CHECK-NEXT: spv.func @baz
-// CHECK-NEXT: spv._reference_of @foo_2
-// CHECK-NEXT: spv.CompositeExtract
-// CHECK-NEXT: spv.ReturnValue
-// CHECK-NEXT: }
-
-// CHECK-NEXT: spv.func @foo
-// CHECK-NEXT: spv.ReturnValue
-// CHECK-NEXT: }
-
-// CHECK-NEXT: spv.func @bar
-// CHECK-NEXT: spv.ReturnValue
-// CHECK-NEXT: }
-// CHECK-NEXT: }
-
-module {
-spv.module Logical GLSL450 {
- spv.specConstant @bar = -5 : i32
- spv.specConstantComposite @foo (@bar, @bar) : !spv.array<2 x i32>
-
- spv.func @baz() -> i32 "None" {
- %0 = spv._reference_of @foo : !spv.array<2 x i32>
- %1 = spv.CompositeExtract %0[0 : i32] : !spv.array<2 x i32>
- spv.ReturnValue %1 : i32
- }
-}
-
-spv.module Logical GLSL450 {
- spv.func @foo(%arg0 : i32) -> i32 "None" {
- spv.ReturnValue %arg0 : i32
- }
-
- spv.func @bar(%arg0 : f32) -> f32 "None" {
- spv.ReturnValue %arg0 : f32
- }
-}
-}
-
-// -----
-
-// Resolve conflicting globalVariableOps.
-
-// CHECK: module {
-// CHECK-NEXT: spv.module Logical GLSL450 {
-// CHECK-NEXT: spv.globalVariable @foo_1
-
-// CHECK-NEXT: spv.globalVariable @foo
-// CHECK-NEXT: }
-
-module {
-spv.module Logical GLSL450 {
- spv.globalVariable @foo bind(1, 0) : !spv.ptr<f32, Input>
-}
-
-spv.module Logical GLSL450 {
- spv.globalVariable @foo bind(1, 0) : !spv.ptr<f32, Input>
-}
-}
-
-// -----
-
-// Resolve conflicting globalVariableOp and specConstantOp.
-
-// CHECK: module {
-// CHECK-NEXT: spv.module Logical GLSL450 {
-// CHECK-NEXT: spv.globalVariable @foo_1
-
-// CHECK-NEXT: spv.specConstant @foo
-// CHECK-NEXT: }
-
-module {
-spv.module Logical GLSL450 {
- spv.globalVariable @foo bind(1, 0) : !spv.ptr<f32, Input>
-}
-
-spv.module Logical GLSL450 {
- spv.specConstant @foo = -5 : i32
-}
-}
-
-// -----
-
-// Resolve conflicting specConstantOp and globalVariableOp.
-
-// CHECK: module {
-// CHECK-NEXT: spv.module Logical GLSL450 {
-// CHECK-NEXT: spv.specConstant @foo_1
-
-// CHECK-NEXT: spv.globalVariable @foo
-// CHECK-NEXT: }
-
-module {
-spv.module Logical GLSL450 {
- spv.specConstant @foo = -5 : i32
-}
-
-spv.module Logical GLSL450 {
- spv.globalVariable @foo bind(1, 0) : !spv.ptr<f32, Input>
-}
-}
-
-// -----
-
-// Resolve conflicting globalVariableOp and specConstantCompositeOp.
-
-// CHECK: module {
-// CHECK-NEXT: spv.module Logical GLSL450 {
-// CHECK-NEXT: spv.globalVariable @foo_1
-
-// CHECK-NEXT: spv.specConstant @bar
-// CHECK-NEXT: spv.specConstantComposite @foo (@bar, @bar)
-// CHECK-NEXT: }
-
-module {
-spv.module Logical GLSL450 {
- spv.globalVariable @foo bind(1, 0) : !spv.ptr<f32, Input>
-}
-
-spv.module Logical GLSL450 {
- spv.specConstant @bar = -5 : i32
- spv.specConstantComposite @foo (@bar, @bar) : !spv.array<2 x i32>
-}
-}
-
-// -----
-
-// Resolve conflicting globalVariableOp and specConstantComposite.
-
-// CHECK: module {
-// CHECK-NEXT: spv.module Logical GLSL450 {
-// CHECK-NEXT: spv.specConstant @bar
-// CHECK-NEXT: spv.specConstantComposite @foo_1 (@bar, @bar)
-
-// CHECK-NEXT: spv.globalVariable @foo
-// CHECK-NEXT: }
-
-module {
-spv.module Logical GLSL450 {
- spv.specConstant @bar = -5 : i32
- spv.specConstantComposite @foo (@bar, @bar) : !spv.array<2 x i32>
-}
-
-spv.module Logical GLSL450 {
- spv.globalVariable @foo bind(1, 0) : !spv.ptr<f32, Input>
-}
-}
diff --git a/mlir/test/lib/Dialect/SPIRV/CMakeLists.txt b/mlir/test/lib/Dialect/SPIRV/CMakeLists.txt
index 089e4e1464de..204a63337730 100644
--- a/mlir/test/lib/Dialect/SPIRV/CMakeLists.txt
+++ b/mlir/test/lib/Dialect/SPIRV/CMakeLists.txt
@@ -2,7 +2,6 @@
add_mlir_library(MLIRSPIRVTestPasses
TestAvailability.cpp
TestEntryPointAbi.cpp
- TestModuleCombiner.cpp
EXCLUDE_FROM_LIBMLIR
diff --git a/mlir/test/lib/Dialect/SPIRV/TestModuleCombiner.cpp b/mlir/test/lib/Dialect/SPIRV/TestModuleCombiner.cpp
deleted file mode 100644
index b321954c87f3..000000000000
--- a/mlir/test/lib/Dialect/SPIRV/TestModuleCombiner.cpp
+++ /dev/null
@@ -1,48 +0,0 @@
-//===- TestModuleCombiner.cpp - Pass to test SPIR-V module combiner lib ---===//
-//
-// Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
-// See https://llvm.org/LICENSE.txt for license information.
-// SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
-//
-//===----------------------------------------------------------------------===//
-
-#include "mlir/Dialect/SPIRV/ModuleCombiner.h"
-
-#include "mlir/Dialect/SPIRV/SPIRVOps.h"
-#include "mlir/Dialect/SPIRV/SPIRVTypes.h"
-#include "mlir/IR/Builders.h"
-#include "mlir/IR/Module.h"
-#include "mlir/Pass/Pass.h"
-
-using namespace mlir;
-
-namespace {
-class TestModuleCombinerPass
- : public PassWrapper<TestModuleCombinerPass,
- OperationPass<mlir::ModuleOp>> {
-public:
- TestModuleCombinerPass() = default;
- TestModuleCombinerPass(const TestModuleCombinerPass &) {}
- void runOnOperation() override;
-
-private:
- mlir::spirv::OwningSPIRVModuleRef combinedModule;
-};
-} // namespace
-
-void TestModuleCombinerPass::runOnOperation() {
- auto modules = llvm::to_vector<4>(getOperation().getOps<spirv::ModuleOp>());
-
- OpBuilder combinedModuleBuilder(modules[0]);
- combinedModule = spirv::combine(modules, combinedModuleBuilder, nullptr);
-
- for (spirv::ModuleOp module : modules)
- module.erase();
-}
-
-namespace mlir {
-void registerTestSpirvModuleCombinerPass() {
- PassRegistration<TestModuleCombinerPass> registration(
- "test-spirv-module-combiner", "Tests SPIR-V module combiner library");
-}
-} // namespace mlir
diff --git a/mlir/tools/mlir-opt/mlir-opt.cpp b/mlir/tools/mlir-opt/mlir-opt.cpp
index b5506a5a34a0..196bda69dbaf 100644
--- a/mlir/tools/mlir-opt/mlir-opt.cpp
+++ b/mlir/tools/mlir-opt/mlir-opt.cpp
@@ -79,7 +79,6 @@ void registerTestPrintNestingPass();
void registerTestRecursiveTypesPass();
void registerTestReducer();
void registerTestSpirvEntryPointABIPass();
-void registerTestSpirvModuleCombinerPass();
void registerTestSCFUtilsPass();
void registerTestTraitsPass();
void registerTestVectorConversions();
@@ -141,7 +140,6 @@ void registerTestPasses() {
registerTestReducer();
registerTestGpuParallelLoopMappingPass();
registerTestSpirvEntryPointABIPass();
- registerTestSpirvModuleCombinerPass();
registerTestSCFUtilsPass();
registerTestTraitsPass();
registerTestVectorConversions();
More information about the Mlir-commits
mailing list