def make_func_for_chunk_size(chunk_size): return elementwise.get_put_kernel( a_dtype, dest_indices.dtype, vec_count=chunk_size)
def make_func_for_chunk_size(chunk_size): func = elementwise.get_put_kernel( a_dtype, dest_indices.dtype, vec_count=chunk_size) func.set_block_shape(*dest_indices._block) return func