rebase

apache · Aug 11, 2020 · 782c542 · 782c542
1 parent 522c466
commit 782c542
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());