Skip to content

GitLab

Commit 35a85789 authored by ptrblck's avatar ptrblck Committed by mcarilli
Browse files

remove deprecated backaned.FunctionBackend calls (#466)

parent 53eae198
Hide whitespace changes
Inline Side-by-side
Showing with 3 additions and 9 deletions
apex/amp/utils.py
View file @ 35a85789
...@@ -129,25 +129,19 @@ def as_inplace(fns): ...@@ -129,25 +129,19 @@ def as_inplace(fns):
yield x + '_' yield x + '_'
def has_func(mod, fn): def has_func(mod, fn):
if isinstance(mod, torch.nn.backends.backend.FunctionBackend): if isinstance(mod, dict):
return fn in mod.function_classes
elif isinstance(mod, dict):
return fn in mod return fn in mod
else: else:
return hasattr(mod, fn) return hasattr(mod, fn)
def get_func(mod, fn): def get_func(mod, fn):
if isinstance(mod, torch.nn.backends.backend.FunctionBackend): if isinstance(mod, dict):
return mod.function_classes[fn]
elif isinstance(mod, dict):
return mod[fn] return mod[fn]
else: else:
return getattr(mod, fn) return getattr(mod, fn)
def set_func(mod, fn, new_fn): def set_func(mod, fn, new_fn):
if isinstance(mod, torch.nn.backends.backend.FunctionBackend): if isinstance(mod, dict):
mod.function_classes[fn] = new_fn
elif isinstance(mod, dict):
mod[fn] = new_fn mod[fn] = new_fn
else: else:
setattr(mod, fn, new_fn) setattr(mod, fn, new_fn)
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment