diff --git a/python/python_direct/runtime.cpp b/python/python_direct/runtime.cpp index 5b581dc7a3d..f982b010132 100644 --- a/python/python_direct/runtime.cpp +++ b/python/python_direct/runtime.cpp @@ -186,6 +186,16 @@ Returns ------- str The fusion intermediate representation (IR) as a string. +)") + .def( + "print", + [](Fusion& f) { f.print(std::cout); }, + R"( +Print the fusion, including transforms, to stdout. + +Returns +------- +None )") .def( "print_kernel",