From f79aac314c7b46115c850babc6fd31457cdc52d5 Mon Sep 17 00:00:00 2001
From: "yuenan.li" <yuenan.li@verisilicon.com>
Date: Tue, 16 Mar 2021 14:20:18 +0800
Subject: [PATCH] Fix bug in tensor attribute setting

Signed-off-by: yuenan.li <yuenan.li@verisilicon.com>
Signed-off-by: Kainan Cha <kainan.cha@verisilicon.com>
---
 src/tim/vx/tensor.cc | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/src/tim/vx/tensor.cc b/src/tim/vx/tensor.cc
index 7c72604..d4c9ee5 100644
--- a/src/tim/vx/tensor.cc
+++ b/src/tim/vx/tensor.cc
@@ -151,8 +151,8 @@ bool TensorImpl::Init() {
 
   memset(&attr, 0x00, sizeof(attr));
   attr.dim_num = spec_.shape_.size();
-  attr.is_const = spec_.attr_ & TensorAttribute::CONSTANT;
-  attr.vtl = spec_.attr_ & TensorAttribute::TRANSIENT;
+  attr.is_const = static_cast<bool>(spec_.attr_ & TensorAttribute::CONSTANT);
+  attr.vtl = static_cast<bool>(spec_.attr_ & TensorAttribute::TRANSIENT);
 
   for (ShapeType::size_type i = 0; i < spec_.shape_.size(); i++) {
     attr.size[i] = spec_.shape_[i];