Python get_take_put_kernel Beispiele

Beispiel #1

    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

Beispiel #2

Datei: gpuarray.py Projekt: 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)

Beispiel #3

 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)