Skip to content

Commit

Permalink
Add __host__ annotation to lambda whose return type needs to be queri…
Browse files Browse the repository at this point in the history
…ed on the host (#816)
  • Loading branch information
msimberg authored Mar 23, 2023
1 parent b9ec3d2 commit 0f0ae83
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion src/eigensolver/tridiag_solver/kernels.cu
Original file line number Diff line number Diff line change
Expand Up @@ -293,7 +293,7 @@ void divideEvecsByDiagonal(const SizeType& k, const SizeType& i_subm_el, const S
// Multiply along rows
//
// Note: the output of the reduction is saved in the first column.
auto mult_op = [] __device__(const T& a, const T& b) { return a * b; };
auto mult_op = [] __host__ __device__(const T& a, const T& b) { return a * b; };
size_t temp_storage_bytes;

using OffsetIterator =
Expand Down

0 comments on commit 0f0ae83

Please sign in to comment.