Ejemplo n.º 1
0
 def f(subset, a, stream=None):
     from pycuda.reduction import get_subset_minmax_kernel
     krnl = get_subset_minmax_kernel(what, a.dtype, subset.dtype)
     return krnl(subset, a,  stream=stream)
Ejemplo n.º 2
0
 def f(subset, a, stream=None):
     from pycuda.reduction import get_subset_minmax_kernel
     krnl = get_subset_minmax_kernel(what, a.dtype, subset.dtype)
     return krnl(subset, a,  stream=stream)