llvm
/
torch-mlir
mirror of https://github.com/llvm/torch-mlir
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

[Bazel] Fix bazel deps (#3414) 

#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.
pull/3416/head
penguin_wwy 2024-06-04 15:50:29 +08:00 committed by GitHub
parent 50f7103098
commit 89f7d24fdc
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 11 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
utils/bazel/torch-mlir-overlay/BUILD.bazel
View File

@ -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",
),
],
@ -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",
@ -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",
@ -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",
],