Fixed typo and type mismatch (apache#5259)

Co-authored-by: Adrian Muresan <[email protected]>
jinqiua · Apr 7, 2020 · 7902f76 · 7902f76
1 parent 8df97ff
commit 7902f76
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/tutorials/dev/relay_pass_infra.py b/tutorials/dev/relay_pass_infra.py
@@ -216,13 +216,13 @@ def transform_function(self, func, mod, ctx):
         obj = self
 
         class ReplaceConstant(tvm.relay.ExprMutator):
-            def visit_const(self, c):
+            def visit_constant(self, c):
                 return relay.multiply(obj.multiplier, c)
         return ReplaceConstant().visit(func)
 
 f = example()
 mod = tvm.IRModule.from_expr(f)
-custom_pass = CustomPipeline(multiplier=relay.const(3, "float"))
+custom_pass = CustomPipeline(multiplier=relay.const(3, "float32"))
 assert custom_pass.info.name == "CustomPipeline"
 mod3 = custom_pass(mod)
 print(mod3)