rebase

apache · Aug 4, 2020 · 9105a96 · 9105a96
1 parent d6ab607
commit 9105a96
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/auto_scheduler/feature.cc b/src/auto_scheduler/feature.cc
@@ -1222,7 +1222,7 @@ void GetPerStoreFeaturesWorkerFunc(const SearchTask& task, const State& state, i
     }
     auto mod = IRModule(Map<GlobalVar, BaseFunc>({{global_var, f}}));
 
-    if (task->target->id->device_type == kDLGPU) {
+    if (task->target->kind->device_type == kDLGPU) {
       auto pass_list = Array<tvm::transform::Pass>();
       // Phase 0
       pass_list.push_back(tir::transform::InjectPrefetch());