reduce_sum.cpp 425 Bytes
Newer Older
Paul's avatar
Paul committed
1
#include <migraphx/gpu/device/reduce_sum.hpp>
Paul's avatar
Paul committed
2
#include <migraphx/gpu/device/reduce.hpp>
Paul's avatar
Paul committed
3
4
5
6
7
8

namespace migraphx {
inline namespace MIGRAPHX_INLINE_NS {
namespace gpu {
namespace device {

Paul's avatar
Paul committed
9

Paul's avatar
Paul committed
10
11
void reduce_sum(hipStream_t stream, const argument& result, const argument& arg)
{
Paul's avatar
Paul committed
12
    reduce(stream, result, arg, sum{}, 0, id{}, id{});
Paul's avatar
Paul committed
13
14
15
16
17
18
}

} // namespace device
} // namespace gpu
} // namespace MIGRAPHX_INLINE_NS
} // namespace migraphx