From 101b58d1cadfed5fb5d9ff9eeec9e6d58899e044 Mon Sep 17 00:00:00 2001
From: Megvii Engine Team <megengine@megvii.com>
Date: Fri, 21 Aug 2020 14:12:14 +0800
Subject: [PATCH] fix(dnn): enable bool input to cond_take

GitOrigin-RevId: 532fa7c073849f47c22ee1f7cf4a24913d14a829
---
 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 f9761a70..39fba5ab 100644
--- a/dnn/src/naive/cond_take/opr_impl.cpp
+++ b/dnn/src/naive/cond_take/opr_impl.cpp
@@ -91,6 +91,7 @@ CondTakeImpl::Output CondTakeImpl::exec(
             break; \
         }
         MEGDNN_FOREACH_COMPUTING_DTYPE(cb)
+        cb(::megdnn::dtype::Bool)
 #undef cb
         default:
             megdnn_throw("bad data dtype");