Skip to content
Snippets Groups Projects
Commit d3f31844 authored by Andreas Klöckner's avatar Andreas Klöckner
Browse files

Delete IPython support in original location

parent ad235236
No related branches found
No related tags found
No related merge requests found
from __future__ import division
from IPython.core.magic import (magics_class, Magics, cell_magic)
import pyopencl as cl
from warnings import warn
warn("pyopencl.ipython is deprecated. Use pyopencl.ipython_ext instead.")
@magics_class
class PyOpenCLMagics(Magics):
@cell_magic
def cl_kernel(self, line, cell):
try:
ctx = self.shell.user_ns["cl_ctx"]
except KeyError:
ctx = None
if not isinstance(ctx, cl.Context):
ctx = None
if ctx is None:
try:
ctx = self.shell.user_ns["ctx"]
except KeyError:
ctx = None
if ctx is None:
raise RuntimeError("unable to locate cl context, which must be "
"present in namespace as 'cl_ctx' or 'ctx'")
prg = cl.Program(ctx, cell.encode("utf8")).build()
for knl in prg.all_kernels():
self.shell.user_ns[knl.function_name] = knl
def load_ipython_extension(ip):
ip.register_magics(PyOpenCLMagics)
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment