"vscode:/vscode.git/clone" did not exist on "5a5ca8e7ff042842013cf7ad1462a5d6c705d39b"
mul.cpp 348 Bytes
Newer Older
Khalique's avatar
Khalique committed
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
#include <migraph/gpu/device/mul.hpp>
#include <migraph/gpu/device/nary.hpp>

namespace migraph {
namespace gpu {
namespace device {

void mul(const argument& result, const argument& arg1, const argument& arg2)
{
    nary(result, arg1, arg2)([](auto x, auto y) { return x * y; });
}

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