Skip to content

Commit

Permalink
MRA: Fix transform in compress kernel
Browse files Browse the repository at this point in the history
Signed-off-by: Joseph Schuchart <joseph.schuchart@stonybrook.edu>
  • Loading branch information
devreal committed Sep 5, 2024
1 parent d71b9a1 commit a65614e
Showing 1 changed file with 3 additions and 3 deletions.
6 changes: 3 additions & 3 deletions examples/madness/mra-device/kernels.cu
Original file line number Diff line number Diff line change
Expand Up @@ -391,9 +391,9 @@ __global__ void compress_kernel(
const TensorView<T, NDIM> in(in_ptrs[bid], K);
sumabssq(in, &sumsqs[bid]);
s(child_slice) = in;
//filter<T,K,NDIM>(s,d); // Apply twoscale transformation=
transform<NDIM>(s, hgT, r, workspace);
}
//filter<T,K,NDIM>(s,d); // Apply twoscale transformation=
transform<NDIM>(s, hgT, r, workspace);
if (key.level() > 0 && blockid == 0) {
auto child_slice = get_child_slice<NDIM>(key, K, 0);
p = r(child_slice);
Expand Down Expand Up @@ -424,7 +424,7 @@ void submit_compress_kernel(
thread_dims = dim3(K, 1, 1);
}

CALL_KERNEL(compress_kernel, key.num_children, thread_dims, 0, stream)(
CALL_KERNEL(compress_kernel, 1, thread_dims, 0, stream)(
key, p_view.data(), result_view.data(), hgT_view.data(), tmp, sumsqs, in_ptrs, K);
}

Expand Down

0 comments on commit a65614e

Please sign in to comment.