def subset_dot(subset, a, b, dtype=None, stream=None, allocator=None): from pycuda.reduction import get_subset_dot_kernel krnl = get_subset_dot_kernel(dtype, subset.dtype, a.dtype, b.dtype) return krnl(subset, a, b, stream=stream, allocator=allocator)
def subset_dot(subset, a, b, dtype=None, stream=None, allocator=None): from pycuda.reduction import get_subset_dot_kernel krnl = get_subset_dot_kernel(dtype, subset.dtype, a.dtype, b.dtype) return krnl(subset, a, b, stream=stream, allocator=allocator)