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

[Bazel] Mirror project reorganization (#2547) 

Updates Bazel build to mirror project reorganization from
https://github.com/llvm/torch-mlir/pull/2542.

GHA:
https://github.com/sjain-stanford/torch-mlir/actions/runs/6744430250/job/18334262852
pull/2544/head
Sambhav Jain 2023-11-03 08:45:51 -07:00 committed by GitHub
parent 3caddd2dd9
commit 1b9fb1b51d
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
2 changed files with 45 additions and 45 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -544,16 +544,15 @@ cc_library(
], ],
) )
# External dialects
td_library( td_library(
name = "TorchMLIRTMTensorOpsTdFiles", name = "TorchMLIRTMTensorOpsTdFiles",
srcs = [ srcs = [
"externals/llvm-external-projects/torch-mlir-dialects/include/torch-mlir-dialects/Dialect/TMTensor/IR/ScalarLoopOpInterface.td", "include/torch-mlir-dialects/Dialect/TMTensor/IR/ScalarLoopOpInterface.td",
"externals/llvm-external-projects/torch-mlir-dialects/include/torch-mlir-dialects/Dialect/TMTensor/IR/TMTensorBase.td", "include/torch-mlir-dialects/Dialect/TMTensor/IR/TMTensorBase.td",
"externals/llvm-external-projects/torch-mlir-dialects/include/torch-mlir-dialects/Dialect/TMTensor/IR/TMTensorInterfaces.td", "include/torch-mlir-dialects/Dialect/TMTensor/IR/TMTensorInterfaces.td",
"externals/llvm-external-projects/torch-mlir-dialects/include/torch-mlir-dialects/Dialect/TMTensor/IR/TMTensorOps.td", "include/torch-mlir-dialects/Dialect/TMTensor/IR/TMTensorOps.td",
], ],
includes = ["externals/llvm-external-projects/torch-mlir-dialects/include"], includes = ["include"],
deps = [ deps = [
"@llvm-project//mlir:ControlFlowInterfacesTdFiles", "@llvm-project//mlir:ControlFlowInterfacesTdFiles",
"@llvm-project//mlir:OpBaseTdFiles", "@llvm-project//mlir:OpBaseTdFiles",
@ -563,27 +562,27 @@ td_library(
gentbl_cc_library( gentbl_cc_library(
name = "TorchMLIRTMTensorInterfacesIncGen", name = "TorchMLIRTMTensorInterfacesIncGen",
strip_include_prefix = "externals/llvm-external-projects/torch-mlir-dialects/include", strip_include_prefix = "include",
tbl_outs = [ tbl_outs = [
( (
["-gen-op-interface-decls"], ["-gen-op-interface-decls"],
"externals/llvm-external-projects/torch-mlir-dialects/include/torch-mlir-dialects/Dialect/TMTensor/IR/TMTensorOpInterfaces.h.inc", "include/torch-mlir-dialects/Dialect/TMTensor/IR/TMTensorOpInterfaces.h.inc",
), ),
( (
["-gen-op-interface-defs"], ["-gen-op-interface-defs"],
"externals/llvm-external-projects/torch-mlir-dialects/include/torch-mlir-dialects/Dialect/TMTensor/IR/TMTensorOpInterfaces.cpp.inc", "include/torch-mlir-dialects/Dialect/TMTensor/IR/TMTensorOpInterfaces.cpp.inc",
), ),
( (
["-gen-type-interface-decls"], ["-gen-type-interface-decls"],
"externals/llvm-external-projects/torch-mlir-dialects/include/torch-mlir-dialects/Dialect/TMTensor/IR/TMTensorTypeInterfaces.h.inc", "include/torch-mlir-dialects/Dialect/TMTensor/IR/TMTensorTypeInterfaces.h.inc",
), ),
( (
["-gen-type-interface-defs"], ["-gen-type-interface-defs"],
"externals/llvm-external-projects/torch-mlir-dialects/include/torch-mlir-dialects/Dialect/TMTensor/IR/TMTensorTypeInterfaces.cpp.inc", "include/torch-mlir-dialects/Dialect/TMTensor/IR/TMTensorTypeInterfaces.cpp.inc",
), ),
], ],
tblgen = "@llvm-project//mlir:mlir-tblgen", tblgen = "@llvm-project//mlir:mlir-tblgen",
td_file = "externals/llvm-external-projects/torch-mlir-dialects/include/torch-mlir-dialects/Dialect/TMTensor/IR/TMTensorInterfaces.td", td_file = "include/torch-mlir-dialects/Dialect/TMTensor/IR/TMTensorInterfaces.td",
deps = [ deps = [
":TorchMLIRTMTensorOpsTdFiles", ":TorchMLIRTMTensorOpsTdFiles",
], ],
@ -591,19 +590,19 @@ gentbl_cc_library(
gentbl_cc_library( gentbl_cc_library(
name = "TorchMLIRTMTensorScalarLoopOpInterfaceIncGen", name = "TorchMLIRTMTensorScalarLoopOpInterfaceIncGen",
strip_include_prefix = "externals/llvm-external-projects/torch-mlir-dialects/include", strip_include_prefix = "include",
tbl_outs = [ tbl_outs = [
( (
["-gen-op-interface-decls"], ["-gen-op-interface-decls"],
"externals/llvm-external-projects/torch-mlir-dialects/include/torch-mlir-dialects/Dialect/TMTensor/IR/ScalarLoopOpInterface.h.inc", "include/torch-mlir-dialects/Dialect/TMTensor/IR/ScalarLoopOpInterface.h.inc",
), ),
( (
["-gen-op-interface-defs"], ["-gen-op-interface-defs"],
"externals/llvm-external-projects/torch-mlir-dialects/include/torch-mlir-dialects/Dialect/TMTensor/IR/ScalarLoopOpInterface.cpp.inc", "include/torch-mlir-dialects/Dialect/TMTensor/IR/ScalarLoopOpInterface.cpp.inc",
), ),
], ],
tblgen = "@llvm-project//mlir:mlir-tblgen", tblgen = "@llvm-project//mlir:mlir-tblgen",
td_file = "externals/llvm-external-projects/torch-mlir-dialects/include/torch-mlir-dialects/Dialect/TMTensor/IR/ScalarLoopOpInterface.td", td_file = "include/torch-mlir-dialects/Dialect/TMTensor/IR/ScalarLoopOpInterface.td",
deps = [ deps = [
":TorchMLIRTMTensorOpsTdFiles", ":TorchMLIRTMTensorOpsTdFiles",
], ],
@ -611,37 +610,37 @@ gentbl_cc_library(
gentbl_cc_library( gentbl_cc_library(
name = "TorchMLIRTMTensorOpsIncGen", name = "TorchMLIRTMTensorOpsIncGen",
strip_include_prefix = "externals/llvm-external-projects/torch-mlir-dialects/include", strip_include_prefix = "include",
tbl_outs = [ tbl_outs = [
( (
["-gen-op-decls"], ["-gen-op-decls"],
"externals/llvm-external-projects/torch-mlir-dialects/include/torch-mlir-dialects/Dialect/TMTensor/IR/TMTensorOps.h.inc", "include/torch-mlir-dialects/Dialect/TMTensor/IR/TMTensorOps.h.inc",
), ),
( (
["-gen-op-defs"], ["-gen-op-defs"],
"externals/llvm-external-projects/torch-mlir-dialects/include/torch-mlir-dialects/Dialect/TMTensor/IR/TMTensorOps.cpp.inc", "include/torch-mlir-dialects/Dialect/TMTensor/IR/TMTensorOps.cpp.inc",
), ),
( (
["-gen-typedef-decls"], ["-gen-typedef-decls"],
"externals/llvm-external-projects/torch-mlir-dialects/include/torch-mlir-dialects/Dialect/TMTensor/IR/TMTensorTypes.h.inc", "include/torch-mlir-dialects/Dialect/TMTensor/IR/TMTensorTypes.h.inc",
), ),
( (
[ [
"-gen-dialect-decls", "-gen-dialect-decls",
"-dialect=tm_tensor", "-dialect=tm_tensor",
], ],
"externals/llvm-external-projects/torch-mlir-dialects/include/torch-mlir-dialects/Dialect/TMTensor/IR/TMTensorDialect.h.inc", "include/torch-mlir-dialects/Dialect/TMTensor/IR/TMTensorDialect.h.inc",
), ),
( (
[ [
"-gen-dialect-defs", "-gen-dialect-defs",
"-dialect=tm_tensor", "-dialect=tm_tensor",
], ],
"externals/llvm-external-projects/torch-mlir-dialects/include/torch-mlir-dialects/Dialect/TMTensor/IR/TMTensorDialect.cpp.inc", "include/torch-mlir-dialects/Dialect/TMTensor/IR/TMTensorDialect.cpp.inc",
), ),
], ],
tblgen = "@llvm-project//mlir:mlir-tblgen", tblgen = "@llvm-project//mlir:mlir-tblgen",
td_file = "externals/llvm-external-projects/torch-mlir-dialects/include/torch-mlir-dialects/Dialect/TMTensor/IR/TMTensorOps.td", td_file = "include/torch-mlir-dialects/Dialect/TMTensor/IR/TMTensorOps.td",
deps = [ deps = [
":TorchMLIRTMTensorOpsTdFiles", ":TorchMLIRTMTensorOpsTdFiles",
], ],
@ -650,18 +649,18 @@ gentbl_cc_library(
cc_library( cc_library(
name = "TorchMLIRTMTensorDialect", name = "TorchMLIRTMTensorDialect",
srcs = [ srcs = [
"externals/llvm-external-projects/torch-mlir-dialects/lib/Dialect/TMTensor/IR/ScalarLoopOpInterface.cpp", "lib/Dialect/TMTensor/IR/ScalarLoopOpInterface.cpp",
"externals/llvm-external-projects/torch-mlir-dialects/lib/Dialect/TMTensor/IR/TMTensorDialect.cpp", "lib/Dialect/TMTensor/IR/TMTensorDialect.cpp",
"externals/llvm-external-projects/torch-mlir-dialects/lib/Dialect/TMTensor/IR/TMTensorInterfaces.cpp", "lib/Dialect/TMTensor/IR/TMTensorInterfaces.cpp",
"externals/llvm-external-projects/torch-mlir-dialects/lib/Dialect/TMTensor/IR/TMTensorOps.cpp", "lib/Dialect/TMTensor/IR/TMTensorOps.cpp",
], ],
hdrs = [ hdrs = [
"externals/llvm-external-projects/torch-mlir-dialects/include/torch-mlir-dialects/Dialect/TMTensor/IR/ScalarLoopOpInterface.h", "include/torch-mlir-dialects/Dialect/TMTensor/IR/ScalarLoopOpInterface.h",
"externals/llvm-external-projects/torch-mlir-dialects/include/torch-mlir-dialects/Dialect/TMTensor/IR/TMTensorDialect.h", "include/torch-mlir-dialects/Dialect/TMTensor/IR/TMTensorDialect.h",
"externals/llvm-external-projects/torch-mlir-dialects/include/torch-mlir-dialects/Dialect/TMTensor/IR/TMTensorInterfaces.h", "include/torch-mlir-dialects/Dialect/TMTensor/IR/TMTensorInterfaces.h",
"externals/llvm-external-projects/torch-mlir-dialects/include/torch-mlir-dialects/Dialect/TMTensor/IR/TMTensorOps.h", "include/torch-mlir-dialects/Dialect/TMTensor/IR/TMTensorOps.h",
], ],
strip_include_prefix = "externals/llvm-external-projects/torch-mlir-dialects/include", strip_include_prefix = "include",
deps = [ deps = [
":TorchMLIRTMTensorInterfacesIncGen", ":TorchMLIRTMTensorInterfacesIncGen",
":TorchMLIRTMTensorOpsIncGen", ":TorchMLIRTMTensorOpsIncGen",
@ -678,7 +677,7 @@ cc_library(
td_library( td_library(
name = "TorchMLIRTMTensorTransformsPassesTdFiles", name = "TorchMLIRTMTensorTransformsPassesTdFiles",
srcs = [ srcs = [
"externals/llvm-external-projects/torch-mlir-dialects/include/torch-mlir-dialects/Dialect/TMTensor/Transforms/Passes.td", "include/torch-mlir-dialects/Dialect/TMTensor/Transforms/Passes.td",
], ],
deps = [ deps = [
"@llvm-project//mlir:OpBaseTdFiles", "@llvm-project//mlir:OpBaseTdFiles",
@ -688,23 +687,23 @@ td_library(
gentbl_cc_library( gentbl_cc_library(
name = "TorchMLIRTMTensorTransformsPassesIncGen", name = "TorchMLIRTMTensorTransformsPassesIncGen",
strip_include_prefix = "externals/llvm-external-projects/torch-mlir-dialects/include", strip_include_prefix = "include",
tbl_outs = [ tbl_outs = [
( (
["-gen-pass-decls"], ["-gen-pass-decls"],
"externals/llvm-external-projects/torch-mlir-dialects/include/torch-mlir-dialects/Dialect/TMTensor/Transforms/Passes.h.inc", "include/torch-mlir-dialects/Dialect/TMTensor/Transforms/Passes.h.inc",
), ),
( (
["-gen-pass-capi-header"], ["-gen-pass-capi-header"],
"externals/llvm-external-projects/torch-mlir-dialects/include/torch-mlir-dialects/Dialect/TMTensor/Transforms/Passes.h.cpi.inc", "include/torch-mlir-dialects/Dialect/TMTensor/Transforms/Passes.h.cpi.inc",
), ),
( (
["-gen-pass-capi-impl"], ["-gen-pass-capi-impl"],
"externals/llvm-external-projects/torch-mlir-dialects/include/torch-mlir-dialects/Dialect/TMTensor/Transforms/Passes.cpi.cpp.inc", "include/torch-mlir-dialects/Dialect/TMTensor/Transforms/Passes.cpi.cpp.inc",
), ),
], ],
tblgen = "@llvm-project//mlir:mlir-tblgen", tblgen = "@llvm-project//mlir:mlir-tblgen",
td_file = "externals/llvm-external-projects/torch-mlir-dialects/include/torch-mlir-dialects/Dialect/TMTensor/Transforms/Passes.td", td_file = "include/torch-mlir-dialects/Dialect/TMTensor/Transforms/Passes.td",
deps = [ deps = [
":TorchMLIRTMTensorTransformsPassesTdFiles", ":TorchMLIRTMTensorTransformsPassesTdFiles",
], ],
@ -713,15 +712,15 @@ gentbl_cc_library(
cc_library( cc_library(
name = "TorchMLIRTMTensorPasses", name = "TorchMLIRTMTensorPasses",
srcs = [ srcs = [
"externals/llvm-external-projects/torch-mlir-dialects/lib/Dialect/TMTensor/Transforms/Bufferize.cpp", "lib/Dialect/TMTensor/Transforms/Bufferize.cpp",
"externals/llvm-external-projects/torch-mlir-dialects/lib/Dialect/TMTensor/Transforms/ConvertToLoops.cpp", "lib/Dialect/TMTensor/Transforms/ConvertToLoops.cpp",
"externals/llvm-external-projects/torch-mlir-dialects/lib/Dialect/TMTensor/Transforms/Passes.cpp", "lib/Dialect/TMTensor/Transforms/Passes.cpp",
], ],
hdrs = [ hdrs = [
"externals/llvm-external-projects/torch-mlir-dialects/include/torch-mlir-dialects/Dialect/TMTensor/Transforms/PassDetail.h", "include/torch-mlir-dialects/Dialect/TMTensor/Transforms/PassDetail.h",
"externals/llvm-external-projects/torch-mlir-dialects/include/torch-mlir-dialects/Dialect/TMTensor/Transforms/Passes.h", "include/torch-mlir-dialects/Dialect/TMTensor/Transforms/Passes.h",
], ],
strip_include_prefix = "externals/llvm-external-projects/torch-mlir-dialects/include", strip_include_prefix = "include",
deps = [ deps = [
":TorchMLIRTMTensorDialect", ":TorchMLIRTMTensorDialect",
":TorchMLIRTMTensorTransformsPassesIncGen", ":TorchMLIRTMTensorTransformsPassesIncGen",
@ -800,6 +799,7 @@ cc_library(
hdrs = [ hdrs = [
"include/torch-mlir/InitAll.h", "include/torch-mlir/InitAll.h",
], ],
copts = ["-DTORCH_MLIR_ENABLE_REFBACKEND"],
strip_include_prefix = "include", strip_include_prefix = "include",
deps = [ deps = [
":TorchMLIRConversionPasses", ":TorchMLIRConversionPasses",

2
utils/bazel/torch-mlir-overlay/test/BUILD.bazel
View File

@ -22,8 +22,8 @@ expand_template(
"\"@LLVM_TOOLS_DIR@\"": "os.path.join(os.environ['TEST_SRCDIR'], 'llvm-project', 'llvm')", "\"@LLVM_TOOLS_DIR@\"": "os.path.join(os.environ['TEST_SRCDIR'], 'llvm-project', 'llvm')",
# All disabled, but required to substituted because they are not in quotes. # All disabled, but required to substituted because they are not in quotes.
"@MLIR_ENABLE_BINDINGS_PYTHON@": "0", "@MLIR_ENABLE_BINDINGS_PYTHON@": "0",
"@TORCH_MLIR_ENABLE_JIT_IR_IMPORTER@": "0",
"@TORCH_MLIR_ENABLE_STABLEHLO@": "0", "@TORCH_MLIR_ENABLE_STABLEHLO@": "0",
"@TORCH_MLIR_ENABLE_REFBACKEND@": "1",
}, },
template = "lit.site.cfg.py.in", template = "lit.site.cfg.py.in",
) )