Skip to content

Commit

Permalink
Add Bazel deps
Browse files Browse the repository at this point in the history
  • Loading branch information
lialan committed Feb 19, 2025
1 parent a87a086 commit 40104fc
Show file tree
Hide file tree
Showing 10 changed files with 13 additions and 1 deletion.
1 change: 1 addition & 0 deletions compiler/plugins/target/VulkanSPIRV/BUILD.bazel
Original file line number Diff line number Diff line change
Expand Up @@ -38,6 +38,7 @@ iree_compiler_cc_library(
"@llvm-project//mlir:AsmParser",
"@llvm-project//mlir:GPUDialect",
"@llvm-project//mlir:IR",
"@llvm-project//mlir:LinalgInterfaces",
"@llvm-project//mlir:SPIRVDialect",
"@llvm-project//mlir:SPIRVModuleCombiner",
"@llvm-project//mlir:SPIRVSerialization",
Expand Down
1 change: 1 addition & 0 deletions compiler/src/iree/compiler/Codegen/Common/BUILD.bazel
Original file line number Diff line number Diff line change
Expand Up @@ -228,6 +228,7 @@ iree_compiler_cc_library(
"@llvm-project//mlir:LLVMCommonConversion",
"@llvm-project//mlir:LLVMDialect",
"@llvm-project//mlir:LinalgDialect",
"@llvm-project//mlir:LinalgInterfaces",
"@llvm-project//mlir:LinalgTransforms",
"@llvm-project//mlir:LinalgUtils",
"@llvm-project//mlir:LoopLikeInterface",
Expand Down
1 change: 1 addition & 0 deletions compiler/src/iree/compiler/Codegen/Common/GPU/BUILD.bazel
Original file line number Diff line number Diff line change
Expand Up @@ -131,6 +131,7 @@ iree_compiler_cc_library(
"@llvm-project//mlir:IR",
"@llvm-project//mlir:LLVMCommonConversion",
"@llvm-project//mlir:LinalgDialect",
"@llvm-project//mlir:LinalgInterfaces",
"@llvm-project//mlir:LinalgTransforms",
"@llvm-project//mlir:LinalgUtils",
"@llvm-project//mlir:LoopLikeInterface",
Expand Down
1 change: 1 addition & 0 deletions compiler/src/iree/compiler/Codegen/LLVMGPU/BUILD.bazel
Original file line number Diff line number Diff line change
Expand Up @@ -188,6 +188,7 @@ iree_compiler_cc_library(
"@llvm-project//mlir:LLVMCommonConversion",
"@llvm-project//mlir:LLVMDialect",
"@llvm-project//mlir:LinalgDialect",
"@llvm-project//mlir:LinalgInterfaces",
"@llvm-project//mlir:LinalgTransforms",
"@llvm-project//mlir:MathDialect",
"@llvm-project//mlir:MathToLLVM",
Expand Down
2 changes: 2 additions & 0 deletions compiler/src/iree/compiler/Codegen/Utils/BUILD.bazel
Original file line number Diff line number Diff line change
Expand Up @@ -55,6 +55,7 @@ iree_compiler_cc_library(
"@llvm-project//mlir:GPUDialect",
"@llvm-project//mlir:IR",
"@llvm-project//mlir:LinalgDialect",
"@llvm-project//mlir:LinalgInterfaces",
"@llvm-project//mlir:LinalgTransforms",
"@llvm-project//mlir:LinalgUtils",
"@llvm-project//mlir:MemRefDialect",
Expand Down Expand Up @@ -84,6 +85,7 @@ iree_compiler_cc_library(
"@llvm-project//llvm:Support",
"@llvm-project//mlir:IR",
"@llvm-project//mlir:LinalgDialect",
"@llvm-project//mlir:LinalgInterfaces",
"@llvm-project//mlir:Support",
"@llvm-project//mlir:VectorDialect",
],
Expand Down
1 change: 1 addition & 0 deletions compiler/src/iree/compiler/Codegen/Utils/LinalgOpInfo.h
Original file line number Diff line number Diff line change
Expand Up @@ -8,6 +8,7 @@
#define IREE_COMPILER_CODEGEN_COMMON_LINALGOPINFO_H_

#include "mlir/IR/AffineMap.h"
#include "mlir/Dialect/Linalg/IR/LinalgInterfaces.h"

namespace mlir::iree_compiler {

Expand Down
4 changes: 3 additions & 1 deletion compiler/src/iree/compiler/Codegen/Utils/VectorOpUtils.h
Original file line number Diff line number Diff line change
Expand Up @@ -7,13 +7,15 @@
#ifndef IREE_COMPILER_CODEGEN_UTILS_VECTORUTILS_H_
#define IREE_COMPILER_CODEGEN_UTILS_VECTORUTILS_H_

#include "mlir/Support/LogicalResult.h"
#include "mlir/Dialect/Linalg/IR/LinalgInterfaces.h"

namespace mlir::iree_compiler {

/// A class for querying information about a contract op.
class VectorContractOpInfo {
public:
static FailureOr<VectorContractOpInfo>
static llvm::FailureOr<VectorContractOpInfo>
inferFromIndexingMaps(ArrayRef<AffineMap> maps);

// Returns the (LHS M, RHS N) dimension index pair.
Expand Down
1 change: 1 addition & 0 deletions compiler/src/iree/compiler/Dialect/LinalgExt/Utils/Utils.h
Original file line number Diff line number Diff line change
Expand Up @@ -7,6 +7,7 @@
#ifndef IREE_COMPILER_DIALECT_LINALGEXT_UTILS_UTILS_H_
#define IREE_COMPILER_DIALECT_LINALGEXT_UTILS_UTILS_H_

#include "mlir/Dialect/Linalg/IR/LinalgInterfaces.h"
#include "mlir/Dialect/Utils/ReshapeOpsUtils.h"
#include "mlir/IR/Attributes.h"
#include "mlir/IR/BuiltinTypes.h"
Expand Down
1 change: 1 addition & 0 deletions compiler/src/iree/compiler/GlobalOptimization/BUILD.bazel
Original file line number Diff line number Diff line change
Expand Up @@ -107,6 +107,7 @@ iree_compiler_cc_library(
"@llvm-project//mlir:FunctionInterfaces",
"@llvm-project//mlir:IR",
"@llvm-project//mlir:LinalgDialect",
"@llvm-project//mlir:LinalgInterfaces",
"@llvm-project//mlir:LinalgTransforms",
"@llvm-project//mlir:LinalgUtils",
"@llvm-project//mlir:MathDialect",
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -13,6 +13,7 @@
#include "iree/compiler/GlobalOptimization/Passes.h"
#include "llvm/ADT/SmallPtrSet.h"
#include "llvm/Support/Debug.h"
#include "mlir/Dialect/Linalg/IR/LinalgInterfaces.h"

using llvm::SmallPtrSet;

Expand Down

0 comments on commit 40104fc

Please sign in to comment.