Fix

Author: vinx13

include/tvm/relay/transform.h

@@ -533,11 +533,11 @@ TVM_DLL Pass CanonicalizeOps();
 TVM_DLL Pass AlterOpLayout();
 
 /*!
- * \brief Canonicalize an expression to make operator fusion more efficient.
+ * \brief Canonicalize cast expressions to make operator fusion more efficient.
  *
  * \return The pass.
  */
-TVM_DLL Pass CanonicalizeExpr();
+TVM_DLL Pass CanonicalizeCasr();
 
 }  // namespace transform
 }  // namespace relay

src/relay/pass/canonicalize_cast.cc

@@ -129,7 +129,7 @@ namespace transform {
 Pass CanonicalizeCast() {
   runtime::TypedPackedFunc<Function(Function, Module, PassContext)> pass_func =
     [=](Function f, Module m, PassContext pc) {
-    return Downcast<Function>(CanonicalizeExpr(f));
+    return Downcast<Function>(CanonicalizeCast(f));
   };
   return CreateFunctionPass(pass_func, 3, "CanonicalizeCast",
                             {ir::StringImm::make("InferType")});