fimpy.cupy_kernelsΒΆ

This file contains some custom CUDA kernels, used in the CUDA implementation of FIMPY

Module Attributes

compute_perm_kernel_str

CUDA kernel to compute a mask of all element permutations containing at least one active index.

compute_perm_kernel_shared

CUDA kernel to compute a mask of all element permutations containing at least one active index.