State of GPU support

Abierto
#431 3 comentarios 2 reacciones 0 asignados Ver en GitHub

Nadie ha tomado este issue todavía.

Evaluación

Dificultad
5/5
Tiempo estimado
Más de una semana
Aptitud para principiantes
25/100
Tipo de issue
Nueva funcionalidad
Claridad
Bastante claro
Estado de actividad
Estancado
Stack tecnológico
julia

Línea de trabajo

Empieza revisando los Issues relacionados con GPU arrays y diferenciación automática mencionados en la discusión; después, ejecuta el ejemplo de Julia mostrado con CUDA, KernelFunctions, SEKernel, kernelmatrix y CUDA.allowscalar(false). Se considera terminado cuando el ejemplo funciona con una entrada de CUDA array y queda documentado un camino claro para avanzar con la compatibilidad con GPU.

Escrito por el modelo de indexación a partir del texto del issue.

Descripción

I wanted to ask for an overview of the current state of GPU support of this package. It appears as though there are several issues related to whether the package works nicely on the GPU (i.e. with GPU arrays as inputs) and several proposed solutions, but getting KernelFunctions.jl to work on the GPU seems to be delayed by other things breaking, like AD.

I was wondering whether a clear path forward is already emerging. Since I've done some GPU work before, I'd be happy to help getting this package work on the GPU.

For what I mean by GPU support, at least the following should be possible:

using CUDA, KernelFunctions
CUDA.allowscalar(false)
x = CUDA.rand(16)
k = SEKernel()
kernelmatrix(k, x)
Lenguaje dominante
Julia
Estrellas
276
Forks
42
Merge medio
3 d 12 h
PR fusionados (30 d)
2

Preparar el entorno

Primeros pasos

  1. Lee el issue completo y luego la guía de contribución del proyecto.
  2. Comenta en el issue que vas a ocuparte — evita que dos personas hagan lo mismo.
  3. Haz un fork del repositorio y trabaja en una rama.
  4. Abre un pull request que haga referencia al número del issue.

Más de JuliaGaussianProcesses/KernelFunctions.jl

Todos los issues de JuliaGaussianProcesses/KernelFunctions.jl

Issues similares

Más issues de Julia

Recibe los nuevos issues en tu correo

Un resumen breve de issues de GitHub para principiantes.