np.meshgrid not supported by Numba

libffcv / ffcv

FFCV: Fast Forward Computer Vision (and other ML workloads!)

https://ffcv.io

Apache License 2.0

2.81k stars 180 forks source link

np.meshgrid not supported by Numba #173

Closed AmmaraRazzaq closed 2 years ago

AmmaraRazzaq commented 2 years ago

Hi, I am using np.meshgrid function inside one of the custom transformation function that I have written using the ffcv.pipeline.operation class and it is giving the following error

numba.core.errors.TypingError: Failed in nopython mode pipeline (step: nopython frontend) Failed in nopython mode pipeline (step: nopython frontend) Use of unsupported NumPy function 'numpy.meshgrid' or unsupported use of the function.

andrewilyas commented 2 years ago

Hi! I think this is an issue better raised in Numba---it seems like there are workarounds for implementing compilable version of mesh grid, see e.g., https://stackoverflow.com/questions/70613681/numba-compatible-numpy-meshgrid.

GuillaumeLeclerc commented 2 years ago

I agree, we are not maintaining numba, we are just using it as a dependency.