fimpyΒΆ

Modules

fimpy.cupy_kernels

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

fimpy.fim_base

This file contains the base implementation of the Fast Iterative Method, common to all solvers.

fimpy.fim_cupy

This file contains the GPU implementation of the Fast Iterative Method, based on cupy.

fimpy.fim_cutils

This subpackage contains the Cython implementations for multiple functions: * Generating the point to element maps * Generating the point to neighborhood maps * Computation of the permutation/active indices mask

fimpy.fim_np

This file contains the CPU implementation of the Fast Iterative Method, based on numpy and cython.

fimpy.solver

This file contains the interface to create the Fast Iterative Method solvers.

fimpy.utils

This subpackage contains small custom functions to efficiently compute \(\left<A \mathbf{x}_1, \mathbf{x}_2 \right>\) on the CPU and GPU for different dimensions \(d\).

previous

Benchmark

next

fimpy.cupy_kernels