Skip to content

Commit

Permalink
clean up
Browse files Browse the repository at this point in the history
  • Loading branch information
qinatan committed Aug 29, 2023
1 parent 5378e1c commit 57a8ffa
Show file tree
Hide file tree
Showing 2 changed files with 13 additions and 13 deletions.
20 changes: 10 additions & 10 deletions src/process/process_set.cu
Original file line number Diff line number Diff line change
Expand Up @@ -52,20 +52,20 @@ namespace micm
for (std::size_t i_rxn = 0; i_rxn < n_reactions; ++i_rxn)
{
double rate = device->rate_constants[i_rxn * n_grids + tid];
for (std::size_t i_react = 0; i_react < device->number_of_reactants_[i_rxn]; ++i_react)
rate *= device->state_variables[device->reactant_ids_[react_id_offset + i_react] * n_grids + tid];
for (std::size_t i_react = 0; i_react < device->number_of_reactants_[i_rxn]; ++i_react)
for (std::size_t i_react = 0; i_react < device->number_of_reactants[i_rxn]; ++i_react)
rate *= device->state_variables[device->reactant_ids[react_id_offset + i_react] * n_grids + tid];
for (std::size_t i_react = 0; i_react < device->number_of_reactants[i_rxn]; ++i_react)
{
device->forcing[device->reactant_ids_[react_id_offset + i_react] * n_grids + tid] -= rate;
device->forcing[device->reactant_ids[react_id_offset + i_react] * n_grids + tid] -= rate;
}
for (std::size_t i_prod = 0; i_prod < device->number_of_products_[i_rxn]; ++i_prod)
for (std::size_t i_prod = 0; i_prod < device->number_of_products[i_rxn]; ++i_prod)
{
size_t index = product_ids_[prod_id_offset + i_prod] * n_grids + tid;
device->forcing[index] += yields_[yield_offset + i_prod] * rate;
size_t index = device->product_ids[prod_id_offset + i_prod] * n_grids + tid;
device->forcing[index] += device->yields[yield_offset + i_prod] * rate;
}
react_id_offset += device->number_of_reactants_[i_rxn];
prod_id_offset += device->number_of_products_[i_rxn];
yield_offset += device->number_of_products_[i_rxn];
react_id_offset += device->number_of_reactants[i_rxn];
prod_id_offset += device->number_of_products[i_rxn];
yield_offset += device->number_of_products[i_rxn];
} // for loop over number of reactions
} // if check for valid CUDA threads
} // end of AddForcingTerms_kernel
Expand Down
6 changes: 3 additions & 3 deletions test/unit/process/test_cuda_process_set.cpp
Original file line number Diff line number Diff line change
Expand Up @@ -168,14 +168,14 @@ void testRandomSystemAddJacobianTerms(std::size_t n_cells, std::size_t n_reactio
}

template<class T>
using Group10000VectorMatrix = micm::VectorMatrix<T, 5>;
using Group10000VectorMatrix = micm::VectorMatrix<T, 10000>;

template<class T>
using Group10000SparseVectorMatrix = micm::SparseMatrix<T, micm::SparseMatrixVectorOrdering<5>>;
using Group10000SparseVectorMatrix = micm::SparseMatrix<T, micm::SparseMatrixVectorOrdering<10000>>;

TEST(RandomCudaProcessSet, Forcing)
{
testRandomSystemAddForcingTerms<Group10000VectorMatrix>(5, 8, 6);
testRandomSystemAddForcingTerms<Group10000VectorMatrix>(10000, 500, 400);
}
TEST(RandomCudaProcessSet, Jacobian)
{
Expand Down

0 comments on commit 57a8ffa

Please sign in to comment.