fix string downcast

apache · Apr 10, 2020 · 8801a51 · 8801a51
1 parent 38f8564
commit 8801a51
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 4 deletions.
diff --git a/src/relay/backend/contrib/dnnl/codegen.cc b/src/relay/backend/contrib/dnnl/codegen.cc
@@ -248,15 +248,15 @@ class CodegenDNNL : public ExprVisitor, public CodegenCBase {
 
   GenerateBodyOutput GenerateCompositeFunctionCall(const FunctionNode* callee,
                                                    const CallNode* caller) {
-    const auto pattern_name = callee->GetAttr<tir::StringImm>(attr::kComposite);
+    const auto pattern_name = callee->GetAttr<tvm::runtime::String>(attr::kComposite);
     CHECK(pattern_name.defined()) << "Only functions with composite attribute supported";
 
-    if (pattern_name->value == "dnnl.conv2d_bias_relu") {
+    if (pattern_name == "dnnl.conv2d_bias_relu") {
       const auto* conv_call =
           GetRootCall(callee->body.as<CallNode>(), 2, {"nn.conv2d", "add", "nn.relu"});
       return GenerateBody(conv_call, "dnnl_fused_conv2d_bias_relu", GetArgumentNames(caller),
                           Conv2d(conv_call));
-    } else if (pattern_name->value == "dnnl.conv2d_relu") {
+    } else if (pattern_name == "dnnl.conv2d_relu") {
       const auto* conv_call = GetRootCall(callee->body.as<CallNode>(), 1, {"nn.conv2d", "nn.relu"});
       return GenerateBody(conv_call, "dnnl_fused_conv2d_relu", GetArgumentNames(caller),
                           Conv2d(conv_call));

diff --git a/tests/python/relay/test_pass_partition_graph.py b/tests/python/relay/test_pass_partition_graph.py
@@ -26,7 +26,6 @@
 from tvm import relay, runtime
 from tvm.relay import transform
 from tvm.contrib import util
-from tvm.relay import transform
 from tvm.relay.backend import compile_engine
 from tvm.relay.expr_functor import ExprMutator
 from tvm.relay.op.annotation import compiler_begin, compiler_end