File tree Expand file tree Collapse file tree 3 files changed +4
-2
lines changed Expand file tree Collapse file tree 3 files changed +4
-2
lines changed Original file line number Diff line number Diff 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
Original file line number Diff line number Diff 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>;
You can’t perform that action at this time.
0 commit comments