From 99df4a7996336bb5351e912a7aacf2f09d22bb7c Mon Sep 17 00:00:00 2001
From: Megvii Engine Team <megengine@megvii.com>
Date: Tue, 12 Apr 2022 19:29:32 +0800
Subject: [PATCH] fix(dtype): dtype scalar set_retain_dtype supports bool

GitOrigin-RevId: aafd378e1b958c37d21c64c01bf9476f8d590c8a
---
 src/core/impl/dtype.cpp | 1 +
 1 file changed, 1 insertion(+)

diff --git a/src/core/impl/dtype.cpp b/src/core/impl/dtype.cpp
index 5f2d44f26..494dbcd0b 100644
--- a/src/core/impl/dtype.cpp
+++ b/src/core/impl/dtype.cpp
@@ -209,6 +209,7 @@ typename ctype_enable_if<ctype>::type DTypeScalar::set_retain_dtype(ctype val) {
     }
         MEGDNN_FOREACH_COMPUTING_DTYPE(cb)
         MEGDNN_FOREACH_QUANTIZED_DTYPE(cb)
+        cb(dt_bool);
 #undef cb
         default:
             mgb_throw(ConversionError, "can not assign to dtype %s", m_dtype.name());
-- 
GitLab