diff --git a/vllm_fl/dispatch/builtin_ops.py b/vllm_fl/dispatch/builtin_ops.py index c8a8c990a..6df530058 100644 --- a/vllm_fl/dispatch/builtin_ops.py +++ b/vllm_fl/dispatch/builtin_ops.py @@ -135,22 +135,17 @@ def register_builtins(registry: OpRegistry) -> None: registry: Registry to register into """ # Register FlagGems (DEFAULT) implementations - try: - from .backends.flaggems.register_ops import register_builtins as register_flaggems + from .backends.flaggems.register_ops import register_builtins as register_flaggems + + register_flaggems(registry) + logger.info("Registered FlagGems operators") - register_flaggems(registry) - logger.debug("Registered FlagGems operators") - except Exception as e: - logger.warning(f"Failed to register FlagGems operators: {e}") # Register PyTorch (REFERENCE) implementations - try: - from .backends.reference.register_ops import register_builtins as register_reference + from .backends.reference.register_ops import register_builtins as register_reference - register_reference(registry) - logger.debug("Registered Reference operators") - except Exception as e: - logger.warning(f"Failed to register Reference operators: {e}") + register_reference(registry) + logger.info("Registered Reference operators") # Auto-discover and register VENDOR implementations _register_vendor_backends(registry)