Skip to content

GitLab

Unverified Commit 274c1a3b authored by Frank Lee's avatar Frank Lee Committed by GitHub
Browse files

[fx] fixed apex normalization patch exception (#1352)

parent ce470ba3
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
colossalai/fx/tracer/meta_patch/patched_module/normalization.py
View file @ 274c1a3b
...@@ -26,5 +26,5 @@ try: ...@@ -26,5 +26,5 @@ try:
meta_patched_module.register(apex.normalization.FusedRMSNorm)(torch_nn_normalize) meta_patched_module.register(apex.normalization.FusedRMSNorm)(torch_nn_normalize)
meta_patched_module.register(apex.normalization.MixedFusedLayerNorm)(torch_nn_normalize) meta_patched_module.register(apex.normalization.MixedFusedLayerNorm)(torch_nn_normalize)
meta_patched_module.register(apex.normalization.MixedFusedRMSNorm)(torch_nn_normalize) meta_patched_module.register(apex.normalization.MixedFusedRMSNorm)(torch_nn_normalize)
except ImportError: except (ImportError, AttributeError):
pass pass
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