Skip to content

Parallelize the Gradient and Hessian Finite Difference Scheme #22

@lisa-gm

Description

@lisa-gm

Add JIT compiled and/or raw kernel of the finite difference stencils in: https://github.com/lisa-gm/pyINLA/blob/dev/src/pyinla/utils/finite_difference_stencils.py

Support for CPU and GPU backend needs to be added and explored:

  • JIT CPU using Numba
  • JIT GPU using Numba
  • CuPy RawKernel

Then the support inside pyinla/core needs to be added with possible user configuration option.

Metadata

Metadata

Assignees

No one assigned

    Labels

    New FeatureThe issue relate to a new issue wanted in the framework.Performance improvementThe issue relate to runtime and/or memory performance improvement.

    Type

    No type
    No fields configured for issues without a type.

    Projects

    No projects

    Milestone

    No milestone

    Relationships

    None yet

    Development

    No branches or pull requests

    Issue actions