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

20 lines
439 B
CMake
Raw Normal View History

[MLIR][TORCH] Add TorchConversionToMLProgram and MLProgramBufferize pass This commit changes the `InsertRngGlobalsPass` to `TorchConversionToMLProgram` pass. This commit also adds the `MLProgramBufferize` pass for the bufferization of ml_program dialect ops to run on refbackend. Signed-Off By: Vivek Khandelwal<vivek@nod-labs.com>
2022-11-24 12:33:47 +08:00
add_mlir_conversion_library(TorchMLIRTorchConversionToMLProgram
TorchConversionToMLProgram.cpp
ADDITIONAL_HEADER_DIRS
${PROJECT_SOURCE_DIR}/include/torch-mlir/Conversion/TorchConversionToMLProgram
DEPENDS
TorchMLIRConversionPassIncGen
LINK_LIBS PUBLIC
MLIRIR
MLIRLinalgDialect
Fix a few missing dependencies. (#2014) `TorchToTMTensor` depends on `TorchMLIRTorchUtils` for `mlir::torch::torch_upstream::get_reduction_enum`. `TorchMLIRTorchConversionPasses` depends on multiple libs for both tblgen'd headers and definitions. Test with `ninja TorchMLIRTorchConversionPasses` from a clean build.
2023-04-12 02:18:49 +08:00
MLIRMLProgramDialect
[MLIR][TORCH] Add TorchConversionToMLProgram and MLProgramBufferize pass This commit changes the `InsertRngGlobalsPass` to `TorchConversionToMLProgram` pass. This commit also adds the `MLProgramBufferize` pass for the bufferization of ml_program dialect ops to run on refbackend. Signed-Off By: Vivek Khandelwal<vivek@nod-labs.com>
2022-11-24 12:33:47 +08:00
MLIRMathDialect
Fix a few missing dependencies. (#2014) `TorchToTMTensor` depends on `TorchMLIRTorchUtils` for `mlir::torch::torch_upstream::get_reduction_enum`. `TorchMLIRTorchConversionPasses` depends on multiple libs for both tblgen'd headers and definitions. Test with `ninja TorchMLIRTorchConversionPasses` from a clean build.
2023-04-12 02:18:49 +08:00
MLIRPass
[MLIR][TORCH] Add TorchConversionToMLProgram and MLProgramBufferize pass This commit changes the `InsertRngGlobalsPass` to `TorchConversionToMLProgram` pass. This commit also adds the `MLProgramBufferize` pass for the bufferization of ml_program dialect ops to run on refbackend. Signed-Off By: Vivek Khandelwal<vivek@nod-labs.com>
2022-11-24 12:33:47 +08:00
TorchMLIRTorchDialect
)
torch_mlir_target_includes(TorchMLIRTorchConversionToMLProgram)