"examples/vscode:/vscode.git/clone" did not exist on "ad5aff39316c716ee4034de10b65109ae5d04d24"
Commit 7489c75d authored by Paul's avatar Paul
Browse files

Formatting

parent cb9bfaf4
...@@ -13,7 +13,7 @@ void mul_add_relu(hipStream_t stream, ...@@ -13,7 +13,7 @@ void mul_add_relu(hipStream_t stream,
const argument& arg3) const argument& arg3)
{ {
nary(stream, result, arg1, arg2, arg3)( nary(stream, result, arg1, arg2, arg3)(
[](auto x, auto a, auto b) { return std::max<decltype(a*x + b)>(0, a*x + b); }); [](auto x, auto a, auto b) { return std::max<decltype(a * x + b)>(0, a * x + b); });
} }
void add_relu(hipStream_t stream, void add_relu(hipStream_t stream,
......
...@@ -230,7 +230,8 @@ struct hip_mul_add_relu ...@@ -230,7 +230,8 @@ struct hip_mul_add_relu
} }
argument compute(context& ctx, const shape&, const std::vector<argument>& args) const argument compute(context& ctx, const shape&, const std::vector<argument>& args) const
{ {
device::mul_add_relu(ctx.get_stream().get(), args.at(3), args.at(0), args.at(1), args.at(2)); device::mul_add_relu(
ctx.get_stream().get(), args.at(3), args.at(0), args.at(1), args.at(2));
return args.at(3); return args.at(3);
} }
std::ptrdiff_t output_alias(const std::vector<shape>& shapes) const std::ptrdiff_t output_alias(const std::vector<shape>& shapes) const
......
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