Cleanup and comment

This commit is contained in:
jan 2016-08-04 20:14:17 -07:00
commit 9198779974
7 changed files with 69 additions and 15 deletions

View file

@ -1,3 +1,12 @@
"""
Basic PyOpenCL operations
The functions are mostly concerned with creating and compiling OpenCL
kernels for use by the other solvers.
See kernels/ for any of the .cl files loaded in this file.
"""
from typing import List, Callable
import numpy
@ -8,6 +17,7 @@ import pyopencl.array
from pyopencl.elementwise import ElementwiseKernel
from pyopencl.reduction import ReductionKernel
# Create jinja2 env on module load
jinja_env = jinja2.Environment(loader=jinja2.PackageLoader(__name__, 'kernels'))