Fix AddReluAdd for building without int4 support

1dd03dda · Rosty Geyyer · 451aef90 · 1dd03dda
Commit 1dd03dda authored Aug 22, 2022 by Rosty Geyyer
Hide whitespace changes
Inline Side-by-side

Showing with 2 additions and 0 deletions

include/ck/tensor_operation/gpu/element/element_wise_operation.hpp ...k/tensor_operation/gpu/element/element_wise_operation.hpp +2 -0

No files found.
--- a/include/ck/tensor_operation/gpu/element/element_wise_operation.hpp
+++ b/include/ck/tensor_operation/gpu/element/element_wise_operation.hpp
@@ -99,6 +99,7 @@ struct AddReluAdd
        y         = c;
    }

+#ifdef CK_EXPERIMENTAL_BIT_INT_EXTENSION_INT4
    template <>
    __host__ __device__ constexpr void operator()<int4_t, int8_t, int4_t, int4_t>(
        int4_t& y, const int8_t& x0, const int4_t& x1, const int4_t& x2) const
@@ -108,6 +109,7 @@ struct AddReluAdd
        int32_t c = b + x2;
        y         = c;
    }
+#endif // CK_EXPERIMENTAL_BIT_INT_EXTENSION_INT4
 };

 struct AddHardswishAdd