[Relay] use transform instead of ir_pass for CPS (#3485)

diff --git a/python/tvm/relay/transform.py b/python/tvm/relay/transform.py
index f77a532..2805e0b 100644 (file)
--- a/python/tvm/relay/transform.py
+++ b/python/tvm/relay/transform.py
@@ -457,7 +457,7 @@ def ToCPS(expr, mod=None):
     result: tvm.relay.Pass
         The registered pass that transforms an expression into CPS.
     """
-    return _ir_pass.to_cps(expr, mod)
+    return _transform.to_cps(expr, mod)
 
 
 def EtaExpand():