Skip to content

Commit

Permalink
allow debug/experimental compiler backends (microsoft#5191)
Browse files Browse the repository at this point in the history
As mentioned at
microsoft#4878 (comment),
we are currently unable to enable debug or experimental backends for the
compiler. This PR enables users to utilize these backends.
  • Loading branch information
tohtana authored and SNahir committed Mar 11, 2024
1 parent 507af5c commit d69de8f
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion deepspeed/runtime/compiler.py
Original file line number Diff line number Diff line change
Expand Up @@ -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
Expand Down

0 comments on commit d69de8f

Please sign in to comment.