Skip to content
Snippets Groups Projects

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

Merged Stephan Seitz requested to merge seitz/pystencils:cuda.TextureReference.set_array into master
@@ -9,6 +9,7 @@
@@ -9,6 +9,7 @@
"""
"""
from os.path import dirname, isdir, join
from os.path import dirname, isdir, join
 
from typing import Union
import numpy as np
import numpy as np
@@ -28,7 +29,7 @@ def pow_two_divider(n):
@@ -28,7 +29,7 @@ def pow_two_divider(n):
return divider
return divider
def ndarray_to_tex(tex_ref,
def ndarray_to_tex(tex_ref, # type: Union[cuda.TextureReference, cuda.SurfaceReference],
ndarray,
ndarray,
address_mode=None,
address_mode=None,
filter_mode=None,
filter_mode=None,
@@ -48,7 +49,7 @@ def ndarray_to_tex(tex_ref,
@@ -48,7 +49,7 @@ def ndarray_to_tex(tex_ref,
raise TypeError(
raise TypeError(
'ndarray must be numpy.ndarray or pycuda.gpuarray.GPUArray')
'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)
tex_ref.set_address_mode(0, address_mode)
if ndarray.ndim >= 2:
if ndarray.ndim >= 2:
Loading