Skip to content

Commit 16b52fe

Browse files
committed
fix cuda mismatch template
1 parent e57f107 commit 16b52fe

File tree

2 files changed

+4
-3
lines changed

2 files changed

+4
-3
lines changed

include/micm/GPU.hpp

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -20,6 +20,7 @@
2020
#include <micm/cuda/util/cuda_param.hpp>
2121
#include <micm/cuda/util/cuda_sparse_matrix.hpp>
2222
#include <micm/cuda/util/cuda_util.cuh>
23+
#include <micm/constraint/constraint_set.hpp>
2324
#include <micm/solver/solver_builder.hpp>
2425

2526
namespace micm
@@ -30,7 +31,7 @@ namespace micm
3031
using GpuState = CudaState<CudaDenseMatrixVector, CudaSparseMatrixVector, CudaLuDecompositionMozartInPlace>;
3132

3233
using CudaRosenbrockVectorType = typename CudaRosenbrockSolverParameters::
33-
template SolverType<CudaProcessSet<CudaDenseMatrixVector, CudaSparseMatrixVector>, CudaLinearSolverInPlace<CudaSparseMatrixVector>>;
34+
template SolverType<CudaProcessSet<CudaDenseMatrixVector, CudaSparseMatrixVector>, CudaLinearSolverInPlace<CudaSparseMatrixVector>, ConstraintSet<CudaDenseMatrixVector, CudaSparseMatrixVector>>;
3435
using CudaRosenbrock = Solver<CudaRosenbrockVectorType, GpuState>;
3536

3637
using GpuRosenbrockThreeStageBuilder = CudaSolverBuilderInPlace<CudaRosenbrockSolverParameters>;

include/micm/cuda/solver/cuda_solver_parameters.hpp

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -12,8 +12,8 @@ namespace micm
1212
/// @brief Parameters for the CUDA Rosenbrock solver
1313
struct CudaRosenbrockSolverParameters : public RosenbrockSolverParameters
1414
{
15-
template<class RatesPolicy, class LinearSolverPolicy>
16-
using SolverType = CudaRosenbrockSolver<RatesPolicy, LinearSolverPolicy>;
15+
template<class RatesPolicy, class LinearSolverPolicy, class ConstraintSetPolicy>
16+
using SolverType = CudaRosenbrockSolver<RatesPolicy, LinearSolverPolicy, ConstraintSetPolicy>;
1717

1818
/// @brief Constructor from base class
1919
/// @param base

0 commit comments

Comments
 (0)