Skip to content

Commit

Permalink
use newline instead of commas
Browse files Browse the repository at this point in the history
  • Loading branch information
gabriele merlo committed Sep 2, 2024
1 parent 14388cc commit 66a58a7
Showing 1 changed file with 6 additions and 3 deletions.
9 changes: 6 additions & 3 deletions include/gtensor/assign.h
Original file line number Diff line number Diff line change
Expand Up @@ -253,10 +253,13 @@ __global__ void kernel_assign_7(Elhs lhs, Erhs _rhs)
int tidz = blockIdx.z;
if (tidx < lhs.shape(0) * lhs.shape(1) &&
tidy < lhs.shape(2) * lhs.shape(3)) {
int j = tidx / lhs.shape(0), i = tidx % lhs.shape(0);
int l = tidy / lhs.shape(2), k = tidy % lhs.shape(2);
int j = tidx / lhs.shape(0);
int i = tidx % lhs.shape(0);
int l = tidy / lhs.shape(2);
int k = tidy % lhs.shape(2);
int n = tidz % lhs.shape(5) / lhs.shape(4);
int m = tidz % lhs.shape(5) % lhs.shape(4);
int n = tidz % lhs.shape(5) / lhs.shape(4), int o = tidz / lhs.shape(5);
int o = tidz / lhs.shape(5);

lhs(i, j, k, l, m, n, o) = rhs(i, j, k, l, m, o);
}
Expand Down

0 comments on commit 66a58a7

Please sign in to comment.