Cleanup and comment
This commit is contained in:
parent
89caff471f
commit
9198779974
7 changed files with 69 additions and 15 deletions
|
|
@ -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'))
|
||||
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue