From dbaf84b0ef4074dbadc754f6f4a0c7ded42506af Mon Sep 17 00:00:00 2001 From: Megvii Engine Team Date: Tue, 11 Aug 2020 12:22:14 +0800 Subject: [PATCH] feat(imperative): add cond_take opr GitOrigin-RevId: 5272e6fa7181c929b2ce6cace25244154e1949a6 --- dnn/src/naive/cond_take/opr_impl.cpp | 1 + 1 file changed, 1 insertion(+) diff --git a/dnn/src/naive/cond_take/opr_impl.cpp b/dnn/src/naive/cond_take/opr_impl.cpp index 89354973..f9761a70 100644 --- a/dnn/src/naive/cond_take/opr_impl.cpp +++ b/dnn/src/naive/cond_take/opr_impl.cpp @@ -66,6 +66,7 @@ CondTakeImpl::Output CondTakeImpl::exec( break; \ } MEGDNN_FOREACH_COMPUTING_DTYPE(cb) + cb(::megdnn::dtype::Bool) #undef cb default: megdnn_throw("bad mask dtype");