SymbolicML / DynamicExpressions.jl

Ridiculously fast symbolic expressions
https://ai.damtp.cam.ac.uk/dynamicexpressions
Apache License 2.0
106 stars 15 forks source link

Native GPU support #62

Closed MilesCranmer closed 9 months ago

MilesCranmer commented 10 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 10 months ago