Function import_custom_ops

fastdeploy/import_ops.py:24–48

Imports custom operations from a specified module within a package and adds them to a global namespace. Args: package (str): The name of the package containing the module. module_name (str): The name of the module within the package. global_ns (dict): The global nam

(package, module_name, global_ns)

def import_custom_ops(package, module_name, global_ns):
    """
    Imports custom operations from a specified module within a package and adds them to a global namespace.

    Args:
        package (str): The name of the package containing the module.
        module_name (str): The name of the module within the package.
        global_ns (dict): The global namespace to add the imported functions to.
    """
    try:
        module = importlib.import_module(module_name, package=package)
        functions = inspect.getmembers(module)
        for func_name, func in functions:
            if func_name.startswith("__") or func_name == "_C_ops":
                continue
            logger.debug(f"Import {func_name} from {package}")
            try:
                global_ns[func_name] = func
            except Exception as e:
                logger.warning(f"Failed to import op {func_name}: {e}")

    except Exception as e:
        logger.warning(f"Ops of {package} import failed, it may be not compiled. {e}")

    preprocess_static_op(global_ns)


def rename_imported_op(old_name, new_name, global_ns):