Skip to content

GitLab

Commit 1dd03dda authored by Rosty Geyyer's avatar Rosty Geyyer
Browse files

Fix AddReluAdd for building without int4 support

parent 451aef90
Hide whitespace changes
Inline Side-by-side
Showing with 2 additions and 0 deletions
include/ck/tensor_operation/gpu/element/element_wise_operation.hpp
View file @ 1dd03dda
...@@ -99,6 +99,7 @@ struct AddReluAdd ...@@ -99,6 +99,7 @@ struct AddReluAdd
y = c; y = c;
} }
#ifdef CK_EXPERIMENTAL_BIT_INT_EXTENSION_INT4
template <> template <>
__host__ __device__ constexpr void operator()<int4_t, int8_t, int4_t, int4_t>( __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 int4_t& y, const int8_t& x0, const int4_t& x1, const int4_t& x2) const
...@@ -108,6 +109,7 @@ struct AddReluAdd ...@@ -108,6 +109,7 @@ struct AddReluAdd
int32_t c = b + x2; int32_t c = b + x2;
y = c; y = c;
} }
#endif // CK_EXPERIMENTAL_BIT_INT_EXTENSION_INT4
}; };
struct AddHardswishAdd struct AddHardswishAdd
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment