@@ -420,6 +420,29 @@ cc_library(
420
420
],
421
421
)
422
422
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
+
423
446
cc_library (
424
447
name = "TorchMLIRTorchConversionToMLProgram" ,
425
448
srcs = glob ([
@@ -515,6 +538,7 @@ cc_library(
515
538
":TorchMLIRTorchToSCF" ,
516
539
":TorchMLIRTorchToStablehlo" ,
517
540
":TorchMLIRTorchToTMTensor" ,
541
+ ":TorchMLIRTorchToTensor" ,
518
542
":TorchMLIRTorchToTosa" ,
519
543
],
520
544
)
0 commit comments