Skip to content

Commit 9c655d0

Browse files
[Bazel] Add conversion targets for TorchToTensor (#2666)
Adapts bazel build per #2648. https://github.com/sjain-stanford/torch-mlir/actions/runs/7233207462/job/19708228888
1 parent ae1a6e4 commit 9c655d0

File tree

1 file changed

+24
-0
lines changed

1 file changed

+24
-0
lines changed

utils/bazel/torch-mlir-overlay/BUILD.bazel

+24
Original file line numberDiff line numberDiff line change
@@ -420,6 +420,29 @@ cc_library(
420420
],
421421
)
422422

423+
cc_library(
424+
name = "TorchMLIRTorchToTensor",
425+
srcs = glob([
426+
"lib/Conversion/*.h",
427+
"lib/Conversion/TorchToTensor/*.cpp",
428+
]),
429+
hdrs = glob([
430+
"include/torch-mlir/Conversion/TorchToTensor/*.h",
431+
]),
432+
strip_include_prefix = "include",
433+
deps = [
434+
":TorchMLIRConversionPassesIncGen",
435+
":TorchMLIRConversionUtils",
436+
":TorchMLIRTorchBackendTypeConversion",
437+
":TorchMLIRTorchConversionDialect",
438+
":TorchMLIRTorchDialect",
439+
"@llvm-project//mlir:Dialect",
440+
"@llvm-project//mlir:IR",
441+
"@llvm-project//mlir:Pass",
442+
"@llvm-project//mlir:TensorDialect",
443+
],
444+
)
445+
423446
cc_library(
424447
name = "TorchMLIRTorchConversionToMLProgram",
425448
srcs = glob([
@@ -515,6 +538,7 @@ cc_library(
515538
":TorchMLIRTorchToSCF",
516539
":TorchMLIRTorchToStablehlo",
517540
":TorchMLIRTorchToTMTensor",
541+
":TorchMLIRTorchToTensor",
518542
":TorchMLIRTorchToTosa",
519543
],
520544
)

0 commit comments

Comments
 (0)