fix

open-mmlab · Jul 17, 2023 · aa22144 · aa22144
1 parent cd11954
commit aa22144
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/csrc/mmdeploy/net/rknn/rknn_net.cpp b/csrc/mmdeploy/net/rknn/rknn_net.cpp
@@ -170,7 +170,7 @@ Result<void> RKNNNet::Init(const Value& args) {
       MMDEPLOY_ERROR("rknn query 'RKNN_QUERY_INPUT_ATTR' fail! ret: {}", ret);
       return Status(eFail);
     }
-    if (attr.type != RKNN_TENSOR_UINT8 || attr.type != RKNN_TENSOR_INT8) {
+    if (!(attr.type == RKNN_TENSOR_UINT8 || attr.type == RKNN_TENSOR_INT8)) {
       MMDEPLOY_ERROR("MMDeploy SDK only supports RKNN-INT8 model");
       return Status(eInvalidArgument);
     }