[Relay] fix target string (#3071)

dd63c755 · Bing Xu · Haichen Shen · 4335db0a · dd63c755
Commit dd63c755 authored Apr 22, 2019 by Bing Xu Committed by Haichen Shen Apr 22, 2019
Hide whitespace changes
Inline Side-by-side

Showing with 3 additions and 3 deletions

src/relay/backend/graph_runtime_codegen.cc
+3 -3

No files found.
--- a/src/relay/backend/graph_runtime_codegen.cc
+++ b/src/relay/backend/graph_runtime_codegen.cc
@@ -429,11 +429,11 @@ class GraphRuntimeCodegen
    }
    CCacheKey key = (*pf0)(func, target);
    CachedFunc lowerd_func = (*pf1)(compile_engine_, key);
-    if (!lowered_funcs_.count(target->target_name)) {
-      lowered_funcs_[target->target_name] = {};
+    if (!lowered_funcs_.count(target->str())) {
+      lowered_funcs_[target->str()] = {};
    }
    for (auto f : lowerd_func->funcs) {
-      lowered_funcs_[target->target_name].insert(f);
+      lowered_funcs_[target->str()].insert(f);
    }

    std::vector<GraphNodeRef> inputs;