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 @@
"""
from os.path import dirname, isdir, join
from typing import Union
import numpy as np
......@@ -28,7 +29,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,
......@@ -48,7 +49,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