inducer / loopy

A code generator for array-based code on CPUs and GPUs
http://mathema.tician.de/software/loopy
MIT License
565 stars 71 forks source link

Support `TypeCast` in `CombineMapper` #848

Closed inducer closed 6 days ago

inducer commented 6 days ago

Should address failures like

pymbolic.mapper.UnsupportedExpressionError: <class 'loopy.kernel.tools._IndexCollector'> cannot handle expressions of type <class 'loopy.symbolic.TypeCast'>

observed in e.g. this run

NB: _IndexCollector is not in main (yet?), it was added in @kaushikcfd's loop fusion branch.

cc @mtcam @matthiasdiener