Python get_take_put_kernel Examples

Programming Language: Python

Namespace/Package Name: pycuda.elementwise

Method/Function: get_take_put_kernel

Examples at hotexamples.com: 3

Python get_take_put_kernel - 3 examples found. These are the top rated real world Python examples of pycuda.elementwise.get_take_put_kernel extracted from open source projects. You can rate examples to help us improve the quality of examples.

Example #1

Show file

    def make_func_for_chunk_size(chunk_size):
        func, tex_src = elementwise.get_take_put_kernel(
                a_dtype, src_indices.dtype, 
                with_offsets=src_offsets is not None,
                vec_count=chunk_size)

        func.set_block_shape(*src_indices._block)
        return func, tex_src

Example #2

Show file

File: gpuarray.py Project: Benli11/pycuda

 def make_func_for_chunk_size(chunk_size):
     return elementwise.get_take_put_kernel(
             a_dtype, src_indices.dtype,
             with_offsets=src_offsets is not None,
             vec_count=chunk_size)

Example #3

Show file

 def make_func_for_chunk_size(chunk_size):
     return elementwise.get_take_put_kernel(
             a_dtype, src_indices.dtype,
             with_offsets=src_offsets is not None,
             vec_count=chunk_size)