From 9f2e3faa69920d9c8ea81465f5a22bd2ffc01b51 Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Tianqi=20Zhang=20=28=E5=BC=A0=E5=A4=A9=E5=90=AF=29?=
 <lygztq@gmail.com>
Date: Wed, 21 Sep 2022 09:00:55 +0800
Subject: [PATCH] fix call of GELU_Taylor in LinearCombinationGeneric (#634)

---
 include/cutlass/epilogue/thread/activation.h | 15 +++++++++++++++
 1 file changed, 15 insertions(+)

diff --git a/include/cutlass/epilogue/thread/activation.h b/include/cutlass/epilogue/thread/activation.h
index 4854f3ad..5f8ba63e 100644
--- a/include/cutlass/epilogue/thread/activation.h
+++ b/include/cutlass/epilogue/thread/activation.h
@@ -575,6 +575,11 @@ struct GELU_taylor {
 
   using Params = LinearCombinationGenericParams<T>;
 
+  CUTLASS_HOST_DEVICE
+  T operator()(T const &scalar, Params const &params_) const {
+    return this->operator()(scalar);
+  }
+
 };
 
 template <int N>
@@ -603,6 +608,11 @@ struct GELU_taylor<Array<half_t, N> > {
   }
 
   using Params = LinearCombinationGenericParams<half_t>;
+
+  CUTLASS_HOST_DEVICE
+  Array<half_t, N> operator()(Array<half_t, N> const &rhs, Params const &params_) const {
+    return this->operator()(rhs);
+  }
 };
 
 template <typename T, int N>
@@ -622,6 +632,11 @@ struct GELU_taylor<Array<T, N> > {
   }
 
   using Params = LinearCombinationGenericParams<T>;
+
+  CUTLASS_HOST_DEVICE
+  Array<T, N> operator()(Array<T, N> const &rhs, Params const &params_) const {
+    return this->operator()(rhs);
+  }
 };
 
 /// Computes backwards pass for GELU operator assuming d_t is the layer gradient and