From 8aa34e4a5d5e4761676ea2e44d2f05f13b3f2898 Mon Sep 17 00:00:00 2001 From: Megvii Engine Team Date: Wed, 19 Aug 2020 15:09:16 +0800 Subject: [PATCH] feat(imperative): add advance indexing with bool also fix bug to deal with empty tensor in np2tensor GitOrigin-RevId: 2bd8152c9064be7381191a251bf133f7c9f292ac --- dnn/src/cuda/cond_take/opr_impl.cpp | 1 + 1 file changed, 1 insertion(+) diff --git a/dnn/src/cuda/cond_take/opr_impl.cpp b/dnn/src/cuda/cond_take/opr_impl.cpp index 5dde8afc..cb0bf502 100644 --- a/dnn/src/cuda/cond_take/opr_impl.cpp +++ b/dnn/src/cuda/cond_take/opr_impl.cpp @@ -82,6 +82,7 @@ CondTakeImpl::Output CondTakeImpl::exec( break; \ } MEGDNN_FOREACH_COMPUTING_DTYPE(cb) + cb(::megdnn::dtype::Bool) #undef cb default: megdnn_throw("bad data dtype");