SymbolicML / DynamicExpressions.jl

Ridiculously fast symbolic expressions
https://symbolicml.org/DynamicExpressions.jl/dev
Apache License 2.0
90 stars 11 forks source link

Native GPU support #62

Closed MilesCranmer closed 4 months ago

MilesCranmer commented 5 months ago

This PR adds native GPU support. This is a single CUDA kernel which evaluates an expression directly on the GPU!

TODO:

It works by launching a CUDA kernel to evaluate many independent nodes of a tree at once, from the leafs, upwards:

graphviz

It also lets you evaluate multiple trees at once – which are dispatched to the GPU during the same kernel launches!

MilesCranmer commented 5 months ago