Beispiel #1
0
    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
0
 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
0
 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)