add c backend to CreateTarget (#2256)

94b309a1 · Liang ZOU · Tianqi Chen · 50cfd3cd · 94b309a1
Commit 94b309a1 authored Dec 07, 2018 by Liang ZOU Committed by Tianqi Chen Dec 07, 2018
Hide whitespace changes
Inline Side-by-side

Showing with 1 additions and 1 deletions

src/codegen/build_module.cc
+1 -1

No files found.
--- a/src/codegen/build_module.cc
+++ b/src/codegen/build_module.cc
@@ -58,7 +58,7 @@ Target CreateTarget(const std::string& target_name,
  }
  t->device_type = kDLCPU;
  t->thread_warp_size = 1;
-  if (target_name == "llvm") {
+  if (target_name == "c" || target_name == "llvm") {
    t->keys_array.push_back(ir::StringImm::make("cpu"));
  } else if (target_name == "cuda" || target_name == "nvptx") {
    t->device_type = kDLGPU;