catalyst icon indicating copy to clipboard operation
catalyst copied to clipboard

Improve how the SciPy BLAS library is found

Open dime10 opened this issue 1 year ago • 1 comments

Catalyst support for certain JAX functions relies on a BLAS runtime library being available, specifically the OpenBLAS library shipped with SciPy. The relevant source code is located in the Python compiler driver: https://github.com/PennyLaneAI/catalyst/blob/72698d8120bd691e7c197bb7c852e1228a261db4/frontend/catalyst/compiler.py#L289-L304

This code snippet is brittle and susceptible to different packaging configurations (for instance the assumption of the folder locations relative to the scipy package location.

At minimum we should raise a user friendly error message that there is a problem with the scipy installation.

Ideally, we would also find ways to make this more robust on the various systems, and also consider conda installations.

dime10 avatar Apr 01 '24 19:04 dime10

At minimum we should raise a user friendly error message that there is a problem with the scipy installation.

The first part has been completed, but I'll leave this issue open to track the improvement to the library detection system.

dime10 avatar Apr 05 '24 19:04 dime10