add.cpp 369 Bytes
Newer Older
Paul's avatar
Paul committed
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
#include <migraph/gpu/device/add.hpp>
#include <migraph/gpu/device/nary.hpp>

namespace migraph {
namespace gpu {
namespace device {

void add(argument result, argument arg1, argument arg2)
{
    nary(std::move(result), std::move(arg1), std::move(arg2))(
        [](auto x, auto y) { return x + y; });
}

} // namespace device
} // namespace gpu
} // namespace migraph