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
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)