loopy/frontend/fortran/diagnostic.py

-from __future__ import division, with_statement
+from __future__ import annotations
+
 
 __copyright__ = "Copyright (C) 2009 Andreas Kloeckner"
 

loopy/ipython_ext.py

+from __future__ import annotations
+
+from IPython.core.magic import Magics, cell_magic, magics_class
+
+import loopy as lp
+
+
+@magics_class
+class LoopyMagics(Magics):
+    @cell_magic
+    def fortran_kernel(self, line, cell):
+        result = lp.parse_fortran(cell)
+        self.shell.user_ns["prog"] = result
+
+    @cell_magic
+    def transformed_fortran_kernel(self, line, cell):
+        result = lp.parse_transformed_fortran(
+                cell,
+                transform_code_context=self.shell.user_ns)
+
+        self.shell.user_ns["prog"] = result
+
+
+def load_ipython_extension(ip):
+    ip.register_magics(LoopyMagics)