Skip to content

Commit 81511bb

Browse files
Uplift third_party/tt-mlir to origin/main (9a75141d932abc4a068efdb800a3e2c9b52b3926) 2025-10-26 (#3002)
This PR uplifts the third_party/tt-mlir submodule to the origin/main --------- Co-authored-by: Aleksandar Zecevic <[email protected]>
1 parent 7233dc4 commit 81511bb

File tree

3 files changed

+4
-2
lines changed

3 files changed

+4
-2
lines changed

forge/csrc/CMakeLists.txt

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -23,6 +23,8 @@ set(TTFORGE_CSRC_INCLUDES
2323
${CMAKE_SOURCE_DIR}/third_party/json/single_include
2424
${CMAKE_SOURCE_DIR}/third_party/pybind11_json/include
2525
${CMAKE_SOURCE_DIR}/third_party/tt-mlir/build/include
26+
${CMAKE_SOURCE_DIR}/third_party/tt-mlir/build/runtime/include
27+
${CMAKE_SOURCE_DIR}/third_party/tt-mlir/build/install/include
2628
${CMAKE_SOURCE_DIR}/third_party/tt-mlir/runtime/include
2729
${CMAKE_SOURCE_DIR}/third_party/tt-mlir/include
2830
${TTMLIR_TOOLCHAIN_DIR}/include

forge/csrc/passes/lower_to_mlir.cpp

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -809,7 +809,7 @@ class MLIRGenerator
809809
lowering_handler_map["atan"] = &MLIRGenerator::emit_mlir_ttforge_op<mlir::tt::ttir::AtanOp>;
810810
lowering_handler_map["argmax"] = &MLIRGenerator::emit_mlir_ttforge_op<mlir::tt::ttir::ArgMaxOp>;
811811
lowering_handler_map["avg_pool2d"] = &MLIRGenerator::emit_mlir_ttforge_op<mlir::tt::ttir::AvgPool2dOp>;
812-
lowering_handler_map["batchnorm"] = &MLIRGenerator::emit_mlir_ttforge_op<mlir::tt::ttir::BatchNormOp>;
812+
lowering_handler_map["batchnorm"] = &MLIRGenerator::emit_mlir_ttforge_op<mlir::tt::ttir::BatchNormInferenceOp>;
813813
lowering_handler_map["bitwise_and"] = &MLIRGenerator::emit_mlir_ttforge_op<mlir::tt::ttir::BitwiseAndOp>;
814814
lowering_handler_map["cast"] = &MLIRGenerator::emit_mlir_ttforge_op<mlir::tt::ttir::TypecastOp>;
815815
lowering_handler_map["clip"] = &MLIRGenerator::emit_mlir_ttforge_op<mlir::tt::ttir::ClampScalarOp>;

third_party/tt-mlir

Submodule tt-mlir updated 380 files

0 commit comments

Comments
 (0)