Skip to content

GitLab

"vscode:/vscode.git/clone" did not exist on "9969f3b5b5ccad5524a9dab237b551c0ea6359f4"
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
Show 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