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; + CUTLASS_HOST_DEVICE + T operator()(T const &scalar, Params const ¶ms_) const { + return this->operator()(scalar); + } + }; template @@ -603,6 +608,11 @@ struct GELU_taylor > { } using Params = LinearCombinationGenericParams; + + CUTLASS_HOST_DEVICE + Array operator()(Array const &rhs, Params const ¶ms_) const { + return this->operator()(rhs); + } }; template @@ -622,6 +632,11 @@ struct GELU_taylor > { } using Params = LinearCombinationGenericParams; + + CUTLASS_HOST_DEVICE + Array operator()(Array const &rhs, Params const ¶ms_) const { + return this->operator()(rhs); + } }; /// Computes backwards pass for GELU operator assuming d_t is the layer gradient and