diff --git a/tools/op.cmake b/tools/op.cmake index c4346c21f9dade37df76841abb3b4446e8f79e91..c7d1840bc552d6b45fe2fcc9f8d19e6784598ee6 100644 --- a/tools/op.cmake +++ b/tools/op.cmake @@ -148,4 +148,7 @@ if (TRANSPOSE_OP) endif() if (FUSION_CONVADD_RELU_OP) add_definitions(-DFUSION_CONVADD_RELU_OP) -endif() \ No newline at end of file +endif() +if (FUSION_CONVADDBNRELU_OP) + add_definitions(-DFUSION_CONVADDBNRELU_OP) +endif()