search

omlins / ParallelStencil.jl

Package for writing high-level code for parallel high-performance stencil computations that can be deployed on both GPUs and CPUs
BSD 3-Clause "New" or "Revised" License
323 stars 38 forks source link

Enzyme `>= 0.12` compatibility #169

Closed aelligp closed 1 month ago

aelligp commented 1 month ago

With Enzyme v0.12 a few changes have been made to autodiff which are partly exposed to ParallelStencil. In this PR we wrap the necessary args to Const, add the necessary arg that checks if there is a returned value (in the test case there is Nothing -> Const).

Co-authored by Valentin Churavy during the Hackathon

omlins commented 1 month ago

Thanks for the PR! Indeed, the CI for another reason: https://github.com/JuliaGPU/AMDGPU.jl/issues/685 I will merge and therefore anyways.