mlir/test/python/develoment_files.py

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