[Bugfix][Target] Correct passing of target-queried bool/int parameters (

#8660) Co-authored-by: Eric Lunderberg <elunderberg@octoml.ai>
apache · Aug 5, 2021 · a495f95 · a495f95
1 parent fe2cdf3
commit a495f95
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/src/target/target.cc b/src/target/target.cc
@@ -768,9 +768,9 @@ ObjectPtr<Object> TargetInternal::FromConfig(std::unordered_map<String, ObjectRe
 
         case kTVMArgInt:
           if (type_info.type_index == Integer::ContainerType::_GetOrAllocRuntimeTypeIndex()) {
-            attrs[key] = Bool(static_cast<bool>(ret));
-          } else if (type_info.type_index == Bool::ContainerType::_GetOrAllocRuntimeTypeIndex()) {
             attrs[key] = Integer(static_cast<int64_t>(ret));
+          } else if (type_info.type_index == Bool::ContainerType::_GetOrAllocRuntimeTypeIndex()) {
+            attrs[key] = Bool(static_cast<bool>(ret));
           } else {
             LOG(FATAL) << "Expected " << type_info.type_key << " parameter for attribute '" << key
                        << "', but received integer from device api";