diff --git a/lib/Conversion/StableHLOToTTIR/StableHLOToTTIRPatterns.cpp b/lib/Conversion/StableHLOToTTIR/StableHLOToTTIRPatterns.cpp index 468813b522..ddf6eca0cd 100644 --- a/lib/Conversion/StableHLOToTTIR/StableHLOToTTIRPatterns.cpp +++ b/lib/Conversion/StableHLOToTTIR/StableHLOToTTIRPatterns.cpp @@ -859,8 +859,7 @@ void addElementwiseUnaryOpsConversionPatterns(MLIRContext *ctx, patterns.add>(typeConverter, ctx); patterns.add>(typeConverter, - ctx); + mlir::stablehlo::CbrtOp, mlir::tt::ttir::CbrtOp>>(typeConverter, ctx); patterns.add>(typeConverter, ctx); diff --git a/runtime/lib/ttnn/operations/eltwise/unary.cpp b/runtime/lib/ttnn/operations/eltwise/unary.cpp index 5b7accb5ab..e13447064c 100644 --- a/runtime/lib/ttnn/operations/eltwise/unary.cpp +++ b/runtime/lib/ttnn/operations/eltwise/unary.cpp @@ -5,8 +5,8 @@ #include "tt/runtime/detail/logger.h" #include "tt/runtime/detail/ttnn.h" #include "tt/runtime/ttnn/operations/utils.h" -#include "ttnn/operations/eltwise/unary/unary_composite.hpp" #include "ttnn/operations/copy.hpp" +#include "ttnn/operations/eltwise/unary/unary_composite.hpp" namespace tt::runtime::ttnn::operations::unary { @@ -40,8 +40,8 @@ static void runEltwiseUnaryOP( static void runEltwiseUnaryCompositeOP( const ::tt::target::ttnn::EltwiseOp *op, ProgramTensorPool &tensorPool, - std::function< - ::ttnn::Tensor(const ::ttnn::Tensor &, const ::tt::tt_metal::MemoryConfig&)> + std::function<::ttnn::Tensor(const ::ttnn::Tensor &, + const ::tt::tt_metal::MemoryConfig &)> ttnnOp) { ::ttnn::Tensor *in = nullptr;