Commit 8568124d authored by Stephan Seitz's avatar Stephan Seitz
Browse files

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

I missed that this is actually a member function. Better call it
directly to make it also work with `cuda.SurfaceReference`.
parent f16225d5
Pipeline #21154 passed with stage
in 13 minutes and 38 seconds
...@@ -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:
......
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