"include/composable_kernel/utility/amd_inline_asm.hpp" did not exist on "7a251a0922e1ec7e27b2612e4136b5f5849c89ca"
Commit 28a68428 authored by Adam Osewski's avatar Adam Osewski
Browse files

Add support for halft in reduction op Add.

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