Commit 59c4cdb0 authored by Paul's avatar Paul
Browse files

Formatting

parent 7f05ac8a
...@@ -39,12 +39,12 @@ auto nary_nonstandard(argument result, Arguments... args) ...@@ -39,12 +39,12 @@ auto nary_nonstandard(argument result, Arguments... args)
inline auto binary_broadcast(argument result, argument arg1, argument arg2) inline auto binary_broadcast(argument result, argument arg1, argument arg2)
{ {
return [=](auto f) { return [=](auto f) {
const auto& b_shape = arg2.get_shape(); const auto& b_shape = arg2.get_shape();
auto bdim = std::distance(b_shape.strides().begin(), auto bdim = std::distance(b_shape.strides().begin(),
std::find_if(b_shape.strides().begin(), std::find_if(b_shape.strides().begin(),
b_shape.strides().end(), b_shape.strides().end(),
[](auto x) { return x != 0; })); [](auto x) { return x != 0; }));
auto bdim_len = b_shape.lens()[bdim]; auto bdim_len = b_shape.lens()[bdim];
visit_all(result, arg1, arg2)([&](auto output, auto input1, auto input2) { visit_all(result, arg1, arg2)([&](auto output, auto input1, auto input2) {
using type = std::remove_cv_t<typename decltype(output)::value_type>; using type = std::remove_cv_t<typename decltype(output)::value_type>;
...@@ -52,9 +52,9 @@ inline auto binary_broadcast(argument result, argument arg1, argument arg2) ...@@ -52,9 +52,9 @@ inline auto binary_broadcast(argument result, argument arg1, argument arg2)
auto* yp = input2.data(); auto* yp = input2.data();
auto* outp = output.data(); auto* outp = output.data();
const std::size_t nlocal = 256; const std::size_t nlocal = 256;
const std::size_t nglobal = 256 * nlocal; const std::size_t nglobal = 256 * nlocal;
const std::size_t n = output.size(); const std::size_t n = output.size();
launch(nglobal, nlocal)([=](auto idx) __device__ { launch(nglobal, nlocal)([=](auto idx) __device__ {
__shared__ type buffer[2048]; __shared__ type buffer[2048];
......
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