From d274ebf3474503f56c707e6379245356dfbd4699 Mon Sep 17 00:00:00 2001 From: Masahiro Tanaka <81312776+tohtana@users.noreply.github.com> Date: Mon, 26 Feb 2024 07:08:19 -0800 Subject: [PATCH] allow debug/experimental compiler backends (#5191) As mentioned at https://github.com/microsoft/DeepSpeed/pull/4878#discussion_r1493712812, we are currently unable to enable debug or experimental backends for the compiler. This PR enables users to utilize these backends. --- deepspeed/runtime/compiler.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/deepspeed/runtime/compiler.py b/deepspeed/runtime/compiler.py index b2b612c85180..b5e4e33425d0 100644 --- a/deepspeed/runtime/compiler.py +++ b/deepspeed/runtime/compiler.py @@ -35,7 +35,7 @@ def get_backend_fn(backend: Union[str, Callable]) -> Union[str, Callable]: return backend elif isinstance(backend, str): - if backend in torch._dynamo.list_backends(): + if backend in torch._dynamo.list_backends(exclude_tags=()): return backend # Get module name from backend name