Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
[MLIR][test] Check for ml_dtypes before running tests (llvm#123061)
We noticed that `mlir/python/requirements.txt` lists `ml_dtypes` as a requirement but when looking at the code in `mlir/python`, the only `import` is guarded: ```python try: import ml_dtypes except ModuleNotFoundError: # The third-party ml_dtypes provides some optional low precision data-types for NumPy. ml_dtypes = None ``` This makes `ml_dtypes` an optional dependency. Some python tests however partially depend on `ml_dtypes` and should not run if that module is unavailable. That is what this change does. This is a replacement for llvm#123051 which was excluding tests too broadly.
- Loading branch information