replacing tid

NCAR · Sep 20, 2024 · 431c9f1 · 431c9f1
1 parent 569624d
commit 431c9f1
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 6 deletions.
diff --git a/src/solver/lu_decomposition.cu b/src/solver/lu_decomposition.cu
@@ -62,12 +62,12 @@ namespace micm
  if (d_do_aik[do_aik_offset++])
  {
  size_t U_idx = d_uik_nkj[uik_nkj_offset].first + tid;
- size_t A_idx = d_aik[aik_offset++];
+ size_t A_idx = d_aik[aik_offset++] + tid;
  if (A_idx == sentinel) {
  d_U[U_idx] = 0;
  }
  else {
- d_U[U_idx] = d_A[A_idx + tid];
+ d_U[U_idx] = d_A[A_idx];
  }
  }
 
@@ -90,12 +90,12 @@ namespace micm
  if (d_do_aki[do_aki_offset++])
  {
  size_t L_idx = d_lki_nkj[lki_nkj_offset].first + tid;
- size_t A_idx = d_aki[aki_offset++];
+ size_t A_idx = d_aki[aki_offset++] + tid;
  if (A_idx == sentinel) {
  d_L[L_idx] = 0;
  }
  else {
- d_L[L_idx] = d_A[A_idx + tid];
+ d_L[L_idx] = d_A[A_idx];
  }
  }
  for (size_t ikj = 0; ikj < d_lki_nkj[lki_nkj_offset].second; ++ikj)

diff --git a/test/unit/solver/test_linear_solver_policy.hpp b/test/unit/solver/test_linear_solver_policy.hpp
@@ -201,7 +201,7 @@ void testRandomMatrix(std::size_t number_of_blocks)
  CopyToHostDense<MatrixPolicy>(x);
 
  check_results<FloatingPointType, MatrixPolicy, SparseMatrixPolicy>(
- A, b, x, [&](const FloatingPointType a, const FloatingPointType b) -> void { EXPECT_NEAR(a, b, 1.0e-06); });
+ A, b, x, [&](const FloatingPointType a, const FloatingPointType b) -> void { EXPECT_NEAR(a, b, 1.0e-09); });
 }
 
 template<class MatrixPolicy, class SparseMatrixPolicy, class LinearSolverPolicy>
@@ -308,7 +308,7 @@ void testDiagonalMatrix(std::size_t number_of_blocks)
  CopyToHostDense<MatrixPolicy>(x);
 
  check_results<FloatingPointType, MatrixPolicy, SparseMatrixPolicy>(
- A, b, x, [&](const FloatingPointType a, const FloatingPointType b) -> void { EXPECT_NEAR(a, b, 1.0e-8); });
+ A, b, x, [&](const FloatingPointType a, const FloatingPointType b) -> void { EXPECT_NEAR(a, b, 1.0e-13); });
 }
 
 template<class MatrixPolicy, class SparseMatrixPolicy>