Skip to content

GitLab

"qa/git@developer.sourcefind.cn:OpenDAS/TransformerEngine.git" did not exist on "1269b2e209c392d41d81f12391cdabc0d5a132fd"
Unverified Commit 59c0f096 authored by Kirthi Shankar Sivamani's avatar Kirthi Shankar Sivamani Committed by GitHub
Browse files

[PyTorch] Minor bug fix (#343) 



Minor bug fix
Signed-off-by: default avatarKirthi Shankar Sivamani <ksivamani@nvidia.com>
parent fdc09f42
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
transformer_engine/pytorch/module/layernorm_mlp.py
View file @ 59c0f096
...@@ -61,7 +61,7 @@ def _act_func(activation: str): ...@@ -61,7 +61,7 @@ def _act_func(activation: str):
'swiglu': (tex.swiglu, tex.dswiglu), 'swiglu': (tex.swiglu, tex.dswiglu),
} }
if activation not in funcs: if activation not in funcs:
raise "Activation type " + activation + " is not supported!" raise NotImplementedError("Activation type " + activation + " is not supported!")
return funcs[activation] return funcs[activation]
......
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