From 9b47403b2d918db4758a6780348abd1a8c622e05 Mon Sep 17 00:00:00 2001
From: hlu1 <14827759+hlu1@users.noreply.github.com>
Date: Fri, 21 Oct 2022 19:20:38 -0700
Subject: [PATCH] Add missing CUTLASS_HOST_DEVICE (#671)

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

diff --git a/include/cutlass/epilogue/thread/activation.h b/include/cutlass/epilogue/thread/activation.h
index 5f8ba63e..0162b6e4 100644
--- a/include/cutlass/epilogue/thread/activation.h
+++ b/include/cutlass/epilogue/thread/activation.h
@@ -347,6 +347,7 @@ struct Sigmoid<Array<half_t, N>> {
 
   using Params = LinearCombinationGenericParams<T>;
 
+  CUTLASS_HOST_DEVICE
   Array<T, N> operator()(Array<T, N> const &z, Params const &params_) const {
     return this->operator()(z);
   }