def _conj(result, arg): from pyopencl.elementwise import complex_dtype_to_name fname = "%s_conj" % complex_dtype_to_name(arg.dtype) return elementwise.get_unary_func_kernel(arg.context, fname, arg.dtype, out_dtype=result.dtype)
def f(array, queue=None): result = array._new_like_me() knl = elementwise.get_unary_func_kernel(array.context, name, array.dtype) knl(queue or array.queue, array._global_size, array._local_size, array.data, result.data, array.mem_size) return result
def knl_runner(result, arg): if arg.dtype.kind == "c": from pyopencl.elementwise import complex_dtype_to_name fname = "%s_%s" % (complex_dtype_to_name(arg.dtype), name) else: fname = name return elementwise.get_unary_func_kernel( result.context, fname, arg.dtype)
def knl_runner(result, arg): if arg.dtype.kind == "c": from pyopencl.elementwise import complex_dtype_to_name fname = "%s_%s" % (complex_dtype_to_name(arg.dtype), name) else: fname = name return elementwise.get_unary_func_kernel(result.context, fname, arg.dtype)
def _abs(result, arg): if arg.dtype.kind == "f": fname = "fabs" elif arg.dtype.kind in ["u", "i"]: fname = "abs" else: raise TypeError("unsupported dtype in _abs()") return elementwise.get_unary_func_kernel( arg.context, fname, arg.dtype)
def _abs(result, arg): if arg.dtype.kind == "c": from pyopencl.elementwise import complex_dtype_to_name fname = "%s_abs" % complex_dtype_to_name(arg.dtype) elif arg.dtype.kind == "f": fname = "fabs" elif arg.dtype.kind in ["u", "i"]: fname = "abs" else: raise TypeError("unsupported dtype in _abs()") return elementwise.get_unary_func_kernel(arg.context, fname, arg.dtype, out_dtype=result.dtype)
def _abs(result, arg): if arg.dtype.kind == "c": from pyopencl.elementwise import complex_dtype_to_name fname = "%s_abs" % complex_dtype_to_name(arg.dtype) elif arg.dtype.kind == "f": fname = "fabs" elif arg.dtype.kind in ["u", "i"]: fname = "abs" else: raise TypeError("unsupported dtype in _abs()") return elementwise.get_unary_func_kernel( arg.context, fname, arg.dtype, out_dtype=result.dtype)
def _abs(result, arg): return elementwise.get_unary_func_kernel( arg.context, fname, arg.dtype)
def knl_runner(result, arg): return elementwise.get_unary_func_kernel(result.context, name, arg.dtype)
def _abs(result, arg): return elementwise.get_unary_func_kernel(arg.context, fname, arg.dtype)
def knl_runner(result, arg): return elementwise.get_unary_func_kernel( result.context, name, arg.dtype)