Skip to content

Commit

Permalink
[Bazel] Fix bazel deps (#3414)
Browse files Browse the repository at this point in the history
#3367 and #3364 introduced new dependencies, causing the [Bazel
workflow](https://github.com/llvm/torch-mlir/actions/workflows/bazelBuildAndTest.yml)
to fail. These need to be fixed in Bazel.
  • Loading branch information
penguin-wwy authored Jun 4, 2024
1 parent 50f7103 commit 89f7d24
Showing 1 changed file with 11 additions and 1 deletion.
12 changes: 11 additions & 1 deletion utils/bazel/torch-mlir-overlay/BUILD.bazel
Original file line number Diff line number Diff line change
Expand Up @@ -329,7 +329,10 @@ gentbl_cc_library(
strip_include_prefix = "include",
tbl_outs = [
(
["-gen-pass-decls"],
[
"-gen-pass-decls",
"-DTORCH_MLIR_ENABLE_STABLEHLO",
],
"include/torch-mlir/Dialect/TorchConversion/Transforms/Passes.h.inc",
),
],
Expand Down Expand Up @@ -496,6 +499,9 @@ cc_library(
"lib/Conversion/TorchToStablehlo/*.cpp",
]),
hdrs = glob(["include/torch-mlir/Conversion/TorchToStablehlo/*.h"]),
defines = [
"TORCH_MLIR_ENABLE_STABLEHLO",
],
strip_include_prefix = "include",
deps = [
":TorchMLIRConversionPassesIncGen",
Expand Down Expand Up @@ -556,6 +562,9 @@ cc_library(
"lib/Dialect/TorchConversion/Transforms/*.h",
]),
hdrs = glob(["include/torch-mlir/Dialect/TorchConversion/Transforms/*.h"]),
defines = [
"TORCH_MLIR_ENABLE_STABLEHLO",
],
strip_include_prefix = "include",
deps = [
":TorchMLIRTorchBackendTypeConversion",
Expand Down Expand Up @@ -891,6 +900,7 @@ cc_library(
"@llvm-project//mlir:Dialect",
"@llvm-project//mlir:DialectUtils",
"@llvm-project//mlir:IR",
"@llvm-project//mlir:TensorInferTypeOpInterfaceImpl",
"@stablehlo//:linalg_passes",
"@stablehlo//:stablehlo_passes",
],
Expand Down

0 comments on commit 89f7d24

Please sign in to comment.