| # RUN: %PYTHON %s 2>&1 |
| |
| import os |
| |
| from mlir._mlir_libs import get_include_dirs, get_lib_dirs |
| |
| |
| header_file = os.path.join(get_include_dirs()[0], "mlir-c", "IR.h") |
| assert os.path.isfile(header_file), f"Header does not exist: {header_file}" |
| |
| # Since actual library names are platform specific, just scan the directory |
| # for a filename that contains the library name. |
| expected_lib_name = "MLIRPythonCAPI" |
| all_libs = os.listdir(get_lib_dirs()[0]) |
| found_lib = False |
| for file_name in all_libs: |
| if expected_lib_name in file_name: |
| found_lib = True |
| assert found_lib, f"Did not find '{expected_lib_name}' lib in {all_libs}" |
