Diff Coverage

    module_path = Path(__file__).resolve()
    repository_root = module_path.parents[4]
    search_paths = []
    if (repository_root / "setup.py").is_file():
        search_paths.append(str(
            repository_root / "build" / "native" / "payload" / "hyper_parallel"
            / "platform" / "mindspore" / "custom_ops" / "lib"
        ))
    search_paths.append(_BUILD_LIB)

            module = None
            try:
                spec = util.spec_from_file_location(_MS_EXTENSION_NAME, library_path)
                if spec is None or spec.loader is None:
                    raise ImportError(f"Cannot create an extension loader for {library_path}")
                module = util.module_from_spec(spec)
                sys.modules[_MS_EXTENSION_NAME] = module
                spec.loader.exec_module(module)
                return module
            except (ImportError, OSError, RuntimeError) as error:
                if module is not None and sys.modules.get(_MS_EXTENSION_NAME) is module:
                    sys.modules.pop(_MS_EXTENSION_NAME, None)
                load_errors.append(f"{library_path}: {error}")

        "No compatible prebuilt extension could be loaded. For a source/PYTHONPATH checkout, run "
        "`./build.sh --multicore off --shmem off --custom-ops on`."
    ) from error
if not hasattr(_custom_ops, "npu_mhc_pre_clamp_sinkhorn"):
    raise ImportError(
        "[HP-NATIVE-INCOMPATIBLE] component=custom_ops framework=mindspore. "
        "The prebuilt extension is stale or was built for a different HyperParallel source revision."
    )