Fix gpu extension

[dionhaefner]

No description provided.

[codecov]

Codecov Report

Merging #541 (8a7879d) into main (00b4cc0) will decrease coverage by 0.06%.
The diff coverage is 42.85%.

@@            Coverage Diff             @@
##             main     #541      +/-   ##
==========================================
- Coverage   84.52%   84.47%   -0.06%     
==========================================
  Files          86       86              
  Lines        7450     7451       +1     
  Branches     1298     1298              
==========================================
- Hits         6297     6294       -3     
- Misses        881      884       +3     
- Partials      272      273       +1     

Files	Coverage Δ
veros/core/operators.py	80.28% <100.00%> (-1.28%)	⬇️
veros/runtime.py	75.45% <ø> (ø)
veros/core/special/tdma_.py	51.32% <0.00%> (-1.77%)	⬇️