More clean up

2021-04-18 04:29:20 +00:00
parent b7e43f5eb9
commit 83036ed646
2 changed files with 3 additions and 5 deletions
--- a/include/cutlass/epilogue/thread/activation.h
+++ b/include/cutlass/epilogue/thread/activation.h
@@ -155,7 +155,7 @@ struct GELU<float> {
 template <>
 struct GELU<double> {
  CUTLASS_HOST_DEVICE
-  float operator()(double const &scalar) const {
+  double operator()(double const &scalar) const {
    return cutlass::constants::half<double>() * scalar *
      (cutlass::constants::one<double>() + erf( scalar / cutlass::constants::root_two<double>() ));
  }