Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Make spiluk_handle::reset backwards compatible #2087

Merged
merged 4 commits into from
Jan 17, 2024
Merged

Make spiluk_handle::reset backwards compatible #2087

Changes from all commits
Commits
Show all changes
4 commits
Select commit Hold shift + click to select a range
0895a71
Make spiluk_handle::reset backwards compatible
jgfouca Jan 12, 2024
da8826f
Switch default val for block_size for reset_handle
jgfouca Jan 15, 2024
8c65783
formatting
jgfouca Jan 15, 2024
981e688
Fix comment
jgfouca Jan 15, 2024
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
24 changes: 15 additions & 9 deletions sparse/src/KokkosSparse_spiluk_handle.hpp
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -131,22 +131,28 @@ class SPILUKHandle {
team_size(-1),
vector_size(-1) {}

void reset_handle(const size_type nrows_, const size_type nnzL_,
const size_type nnzU_, const size_type block_size_) {
void reset_handle(
const size_type nrows_, const size_type nnzL_, const size_type nnzU_,
const size_type block_size_ = Kokkos::ArithTraits<size_type>::max()) {
set_nrows(nrows_);
set_num_levels(0);
set_nnzL(nnzL_);
set_nnzU(nnzU_);
set_block_size(block_size_);
// user likely does not want to reset block size to 0, so set default
// to size_type::max
if (block_size_ != Kokkos::ArithTraits<size_type>::max()) {
set_block_size(block_size_);
}
set_level_maxrows(0);
set_level_maxrowsperchunk(0);
level_list = nnz_row_view_t("level_list", nrows_),
level_idx = nnz_lno_view_t("level_idx", nrows_),
level_ptr = nnz_lno_view_t("level_ptr", nrows_ + 1),
hlevel_ptr = nnz_lno_view_host_t("hlevel_ptr", nrows_ + 1),
level_nchunks = nnz_lno_view_host_t(),
level_nrowsperchunk = nnz_lno_view_host_t(), reset_symbolic_complete(),
level_list = nnz_row_view_t("level_list", nrows_);
level_idx = nnz_lno_view_t("level_idx", nrows_);
level_ptr = nnz_lno_view_t("level_ptr", nrows_ + 1);
hlevel_ptr = nnz_lno_view_host_t("hlevel_ptr", nrows_ + 1);
level_nchunks = nnz_lno_view_host_t();
level_nrowsperchunk = nnz_lno_view_host_t();
iw = work_view_t();
reset_symbolic_complete();
}

virtual ~SPILUKHandle(){};
Expand Down