Example #1
0
 def make_func_for_chunk_size(chunk_size):
     knl = elementwise.get_put_kernel(
             a_dtype, dest_indices.dtype, vec_count=chunk_size)
     knl.set_block_shape(*dest_indices._block)
     return knl
Example #2
0
 def make_func_for_chunk_size(chunk_size):
     knl = elementwise.get_put_kernel(a_dtype,
                                      dest_indices.dtype,
                                      vec_count=chunk_size)
     knl.set_block_shape(*dest_indices._block)
     return knl
Example #3
0
 def make_func_for_chunk_size(chunk_size):
     knl = elementwise.get_put_kernel(context,
                                      a_dtype,
                                      dest_indices.dtype,
                                      vec_count=chunk_size)
     return knl
Example #4
0
 def make_func_for_chunk_size(chunk_size):
     knl = elementwise.get_put_kernel(
             context,
             a_dtype, dest_indices.dtype, vec_count=chunk_size)
     return knl