Correct "use_device_api" attribute name on Target

apache · Nov 12, 2021 · f63a0c8 · f63a0c8
1 parent 21c2832
commit f63a0c8
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/src/relay/backend/aot_executor_codegen.cc b/src/relay/backend/aot_executor_codegen.cc
@@ -397,7 +397,7 @@ class AOTExecutorCodegen : public MixedModeVisitor {
    */
   void CollectDeviceVariables(const Map<GlobalVar, String>& device_contexts) {
     Map<TargetKind, tir::Var> target_contexts;
-    TargetKindAttrMap<Bool> target_attr_map = tvm::TargetKind::GetAttrMap<Bool>("c_device_api");
+    TargetKindAttrMap<Bool> target_attr_map = tvm::TargetKind::GetAttrMap<Bool>("use_device_api");
 
     for (const auto& it : device_contexts) {
       const GlobalVar& global_var = it.first;

diff --git a/src/target/target_kind.cc b/src/target/target_kind.cc
@@ -373,7 +373,7 @@ TVM_REGISTER_TARGET_KIND("hybrid", kDLCPU)  // line break
 
 TVM_REGISTER_TARGET_KIND("composite", kDLCPU).add_attr_option<Array<Target>>("devices");
 
-TVM_REGISTER_TARGET_KIND("ethos-u", kDLCPU).set_attr<Bool>("c_device_api", Bool(true));
+TVM_REGISTER_TARGET_KIND("ethos-u", kDLCPU).set_attr<Bool>("use_device_api", Bool(true));
 
 /**********  Registry  **********/