"...composable_kernel.git" did not exist on "334361cbde76a2566fb215a64a6652205b0d2336"
Commit adf2e4b1 authored by Adam Osewski's avatar Adam Osewski
Browse files

Add support for halft in reduction op Add.

parent 1c5b049d
......@@ -130,9 +130,7 @@ __global__ void
const auto StrideA = gemm_desc_ptr[group_id].StrideA;
const auto StrideB = gemm_desc_ptr[group_id].StrideB;
auto gridwise_gemm = GridwiseGemm();
auto& results_buffer = gridwise_gemm.GetCThreadBuffer();
auto gridwise_gemm = GridwiseGemm();
b2c_tile_map.CalculateBottomIndex(work_scheduler.tile_id_ - offset);
// Iterate over K dimension for this [M,N] tile
......
......@@ -52,7 +52,7 @@ struct Add
__host__ __device__ inline constexpr void operator()(T& a, T b) const
{
static_assert(is_same<T, float>::value || is_same<T, double>::value ||
is_same<T, int32_t>::value,
is_same<T, half_t>::value || is_same<T, int32_t>::value,
"The data type is not supported by the Add accumulator!");
a = a + b;
......
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