예제 #1
0
 def make_func_for_chunk_size(chunk_size):
     return elementwise.get_put_kernel(
             a_dtype, dest_indices.dtype, vec_count=chunk_size)
예제 #2
0
파일: gpuarray.py 프로젝트: Benli11/pycuda 
 def make_func_for_chunk_size(chunk_size):
     return elementwise.get_put_kernel(
             a_dtype, dest_indices.dtype, vec_count=chunk_size)
예제 #3
0
 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