Commit 84279e64 authored by Martin Bauer's avatar Martin Bauer
Browse files

Merge branch 'cuda.TextureReference.set_array' into 'master'

cuda.TextureReference.set_array(tex_ref, cu_array) -> tex_ref.set_array(cu_array)

See merge request pycodegen/pystencils!134
parents d48ca2af 8568124d
......@@ -9,6 +9,7 @@
"""
from os.path import dirname, isdir, join
from typing import Union
import numpy as np
......@@ -29,7 +30,7 @@ def pow_two_divider(n):
return divider
def ndarray_to_tex(tex_ref,
def ndarray_to_tex(tex_ref, # type: Union[cuda.TextureReference, cuda.SurfaceReference]
ndarray,
address_mode=None,
filter_mode=None,
......@@ -51,7 +52,7 @@ def ndarray_to_tex(tex_ref,
raise TypeError(
'ndarray must be numpy.ndarray or pycuda.gpuarray.GPUArray')
cuda.TextureReference.set_array(tex_ref, cu_array)
tex_ref.set_array(cu_array)
tex_ref.set_address_mode(0, address_mode)
if ndarray.ndim >= 2:
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment