Ejemplos de get_unary_func_kernel en Python

Ejemplo n.º 1

Archivo: array.py Proyecto: sys-git/pyopencl

 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)

Ejemplo n.º 2

Archivo: clmath.py Proyecto: stefanv/PyOpenCL

    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

Ejemplo n.º 3

Archivo: clmath.py Proyecto: Gormse/MacMiner

    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)

Ejemplo n.º 4

Archivo: clmath.py Proyecto: sys-git/pyopencl

    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)

Ejemplo n.º 5

    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

Ejemplo n.º 6

Archivo: array.py Proyecto: initcrash/pyopencl

    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)

Ejemplo n.º 7

    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)

Ejemplo n.º 8

Archivo: array.py Proyecto: EyNuel/pyopencl

    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)

Ejemplo n.º 9

    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)

Ejemplo n.º 10

Archivo: array.py Proyecto: stefanv/PyOpenCL

 def _abs(result, arg):
     return elementwise.get_unary_func_kernel(
             arg.context, fname, arg.dtype)

Ejemplo n.º 11

 def knl_runner(result, arg):
     return elementwise.get_unary_func_kernel(result.context, name,
                                              arg.dtype)

Ejemplo n.º 12

Archivo: array.py Proyecto: EyNuel/pyopencl

    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)

Ejemplo n.º 13

 def _abs(result, arg):
     return elementwise.get_unary_func_kernel(arg.context, fname, arg.dtype)

Ejemplo n.º 14

Archivo: clmath.py Proyecto: Almclean/pyopencl

 def knl_runner(result, arg):
     return elementwise.get_unary_func_kernel(
             result.context, name, arg.dtype)