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
......@@ -61,7 +61,7 @@ def _act_func(activation: str):
'swiglu': (tex.swiglu, tex.dswiglu),
}
if activation not in funcs:
raise "Activation type " + activation + " is not supported!"
raise NotImplementedError("Activation type " + activation + " is not supported!")
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