derek.holloway
7502bf64a4
CodeQL / Analyze (actions) (push) Canceled after 0s
CodeQL / Analyze (python) (push) Canceled after 0s
pre-commit / pre-commit (push) Canceled after 0s
Multi-Arch CI / setup (push) Canceled after 0s
Publish build_manylinux_x86_64 images / publish_build_manylinux_x86_64 (push) Canceled after 0s
Setup (Multi-Arch) / setup (push) Canceled after 0s
Multi-Arch CI / Linux::${{ fromJSON(needs.setup.outputs.linux_build_config || '{}').build_variant_label || 'skip' }} (push) Canceled after 0s
Publish TheRock Docker image / build-and-push-image (push) Canceled after 0s
Publish no_rocm_image_ubi10 image / publish_no_rocm_image_ubi10 (push) Canceled after 0s
Multi-Arch CI / Windows::${{ fromJSON(needs.setup.outputs.windows_build_config || '{}').build_variant_label || 'skip' }} (push) Canceled after 0s
Publish TheRock Docker image / build-and-push-image (push) Canceled after 0s
Multi-Arch CI / Manifest Diff (push) Canceled after 0s
TheRock Manifest Diff Report / Generate Manifest Diff Report (push) Canceled after 0s
Multi-Arch CI / CI Summary (push) Canceled after 0s
Publish no_rocm_image_ubuntu24_04 images / publish_no_rocm_image_ubuntu24_04 (push) Canceled after 0s
Publish no_rocm_image_ubuntu24_04_ocl_rt images / publish_no_rocm_image_ubuntu24_04_ocl_rt (push) Canceled after 0s
Publish TheRock Docker image / build-and-push-image (push) Canceled after 0s
Publish TheRock Docker image / build-and-push-image (push) Canceled after 0s
Publish no_rocm_image_ubuntu24_04 images / publish_no_rocm_image_ubuntu24_04_rocgdb (push) Canceled after 0s
Publish no_rocm_image_ubuntu24_04_rocgdb images / publish_no_rocm_image_ubuntu24_04_rocgdb (push) Canceled after 0s
Publish no_rocm_image_ubuntu24_04 images / publish_no_rocm_image_ubuntu24_04_ocl_rt (push) Canceled after 0s
Publish TheRock Docker image / build-and-push-image (push) Canceled after 0s
Unit Tests / Unit Tests :: ubuntu-24.04 (push) Canceled after 0s
Unit Tests / Unit Tests :: windows-2022 (push) Canceled after 0s
Publish no_rocm_image_ubuntu24_04 images / publish_no_rocm_image_ubuntu24_04_media (push) Canceled after 0s
Unit Tests / Unit Tests Summary (push) Canceled after 0s
Publish no_rocm_image_ubuntu24_04 images / publish_no_rocm_image_ubuntu24_04_openmpi (push) Canceled after 0s
189 lines
10 KiB
C++
189 lines
10 KiB
C++
/*! \file */
|
|
/* ************************************************************************
|
|
* Copyright (C) 2025 Advanced Micro Devices, Inc. All rights Reserved.
|
|
*
|
|
* Permission is hereby granted, free of charge, to any person obtaining a copy
|
|
* of this software and associated documentation files (the "Software"), to deal
|
|
* in the Software without restriction, including without limitation the rights
|
|
* to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
|
|
* copies of the Software, and to permit persons to whom the Software is
|
|
* furnished to do so, subject to the following conditions:
|
|
*
|
|
* The above copyright notice and this permission notice shall be included in
|
|
* all copies or substantial portions of the Software.
|
|
*
|
|
* THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
|
|
* IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
|
|
* FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
|
|
* AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
|
|
* LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
|
|
* OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN
|
|
* THE SOFTWARE.
|
|
*
|
|
* ************************************************************************ */
|
|
|
|
#include "rocsparse_csrsm.hpp"
|
|
|
|
#include "rocsparse_enum_utils.hpp"
|
|
#include "rocsparse_utility.hpp"
|
|
|
|
#include <map>
|
|
namespace rocsparse
|
|
{
|
|
typedef rocsparse_status (*csrsm_analysis_t)(rocsparse_handle handle,
|
|
rocsparse_operation trans_A,
|
|
rocsparse_operation trans_B,
|
|
int64_t m,
|
|
int64_t nrhs,
|
|
int64_t nnz,
|
|
const void* alpha,
|
|
const rocsparse_mat_descr descr,
|
|
const void* csr_val,
|
|
const void* csr_row_ptr,
|
|
const void* csr_col_ind,
|
|
const void* B,
|
|
int64_t ldb,
|
|
rocsparse_mat_info info,
|
|
rocsparse_analysis_policy analysis,
|
|
rocsparse_solve_policy solve,
|
|
rocsparse_csrsm_info* p_csrsm_info,
|
|
void* temp_buffer);
|
|
|
|
using csrsm_analysis_tuple
|
|
= std::tuple<rocsparse_indextype, rocsparse_indextype, rocsparse_datatype>;
|
|
|
|
#define CSRSM_ANALYSIS_CONFIG(I_, J_, T_) \
|
|
{ \
|
|
csrsm_analysis_tuple(I_, J_, T_), \
|
|
csrsm_analysis_template<typename rocsparse::indextype_traits<I_>::type_t, \
|
|
typename rocsparse::indextype_traits<J_>::type_t, \
|
|
typename rocsparse::datatype_traits<T_>::type_t> \
|
|
}
|
|
|
|
static const std::map<csrsm_analysis_tuple, csrsm_analysis_t> s_csrsm_analysis_dispatch{
|
|
{CSRSM_ANALYSIS_CONFIG(
|
|
rocsparse_indextype_i32, rocsparse_indextype_i32, rocsparse_datatype_f32_r),
|
|
CSRSM_ANALYSIS_CONFIG(
|
|
rocsparse_indextype_i32, rocsparse_indextype_i32, rocsparse_datatype_f64_r),
|
|
CSRSM_ANALYSIS_CONFIG(
|
|
rocsparse_indextype_i32, rocsparse_indextype_i32, rocsparse_datatype_f32_c),
|
|
CSRSM_ANALYSIS_CONFIG(
|
|
rocsparse_indextype_i32, rocsparse_indextype_i32, rocsparse_datatype_f64_c),
|
|
|
|
CSRSM_ANALYSIS_CONFIG(
|
|
rocsparse_indextype_i64, rocsparse_indextype_i32, rocsparse_datatype_f32_r),
|
|
CSRSM_ANALYSIS_CONFIG(
|
|
rocsparse_indextype_i64, rocsparse_indextype_i32, rocsparse_datatype_f64_r),
|
|
CSRSM_ANALYSIS_CONFIG(
|
|
rocsparse_indextype_i64, rocsparse_indextype_i32, rocsparse_datatype_f32_c),
|
|
CSRSM_ANALYSIS_CONFIG(
|
|
rocsparse_indextype_i64, rocsparse_indextype_i32, rocsparse_datatype_f64_c),
|
|
|
|
CSRSM_ANALYSIS_CONFIG(
|
|
rocsparse_indextype_i64, rocsparse_indextype_i64, rocsparse_datatype_f32_r),
|
|
CSRSM_ANALYSIS_CONFIG(
|
|
rocsparse_indextype_i64, rocsparse_indextype_i64, rocsparse_datatype_f64_r),
|
|
CSRSM_ANALYSIS_CONFIG(
|
|
rocsparse_indextype_i64, rocsparse_indextype_i64, rocsparse_datatype_f32_c),
|
|
CSRSM_ANALYSIS_CONFIG(
|
|
rocsparse_indextype_i64, rocsparse_indextype_i64, rocsparse_datatype_f64_c)}};
|
|
|
|
static rocsparse_status csrsm_analysis_find(csrsm_analysis_t* function_,
|
|
rocsparse_indextype i_type_,
|
|
rocsparse_indextype j_type_,
|
|
rocsparse_datatype t_type_)
|
|
{
|
|
const auto& it = rocsparse::s_csrsm_analysis_dispatch.find(
|
|
rocsparse::csrsm_analysis_tuple(i_type_, j_type_, t_type_));
|
|
|
|
if(it != rocsparse::s_csrsm_analysis_dispatch.end())
|
|
{
|
|
function_[0] = it->second;
|
|
}
|
|
// LCOV_EXCL_START
|
|
else
|
|
{
|
|
#ifndef NDEBUG
|
|
std::cout << "invalid precision configuration: "
|
|
<< "i_type: " << rocsparse::enum_utils::to_string(i_type_) << std::endl
|
|
<< ", j_type: " << rocsparse::enum_utils::to_string(j_type_) << std::endl
|
|
<< ", t_type: " << rocsparse::enum_utils::to_string(t_type_) << std::endl;
|
|
|
|
std::cout << "available configuration are: " << std::endl;
|
|
for(const auto& p : rocsparse::s_csrsm_analysis_dispatch)
|
|
{
|
|
const auto& t = p.first;
|
|
const auto i_type = std::get<0>(t);
|
|
const auto j_type = std::get<1>(t);
|
|
const auto t_type = std::get<2>(t);
|
|
std::cout << std::endl
|
|
<< std::endl
|
|
<< "i_type: " << rocsparse::enum_utils::to_string(i_type) << std::endl
|
|
<< ", j_type: " << rocsparse::enum_utils::to_string(j_type) << std::endl
|
|
<< ", t_type: " << rocsparse::enum_utils::to_string(t_type) << std::endl;
|
|
}
|
|
#endif
|
|
|
|
std::stringstream sstr;
|
|
sstr << "invalid precision configuration: "
|
|
<< "i_type: " << rocsparse::enum_utils::to_string(i_type_)
|
|
<< ", j_type: " << rocsparse::enum_utils::to_string(j_type_)
|
|
<< ", t_type: " << rocsparse::enum_utils::to_string(t_type_);
|
|
|
|
RETURN_WITH_MESSAGE_IF_ROCSPARSE_ERROR(rocsparse_status_invalid_value,
|
|
sstr.str().c_str());
|
|
}
|
|
// LCOV_EXCL_STOP
|
|
|
|
return rocsparse_status_success;
|
|
}
|
|
}
|
|
|
|
rocsparse_status rocsparse::csrsm_analysis(rocsparse_handle handle,
|
|
rocsparse_operation trans_A,
|
|
rocsparse_operation trans_B,
|
|
int64_t m,
|
|
int64_t nrhs,
|
|
int64_t nnz,
|
|
rocsparse_datatype alpha_datatype,
|
|
const void* alpha,
|
|
const rocsparse_mat_descr descr,
|
|
rocsparse_datatype csr_val_datatype,
|
|
const void* csr_val,
|
|
rocsparse_indextype csr_row_ptr_indextype,
|
|
const void* csr_row_ptr,
|
|
rocsparse_indextype csr_col_ind_indextype,
|
|
const void* csr_col_ind,
|
|
rocsparse_datatype B_datatype,
|
|
const void* B,
|
|
int64_t ldb,
|
|
rocsparse_mat_info info,
|
|
rocsparse_analysis_policy analysis,
|
|
rocsparse_solve_policy solve,
|
|
rocsparse_csrsm_info* p_csrsm_info,
|
|
void* temp_buffer)
|
|
{
|
|
rocsparse::csrsm_analysis_t f;
|
|
RETURN_IF_ROCSPARSE_ERROR(rocsparse::csrsm_analysis_find(
|
|
&f, csr_row_ptr_indextype, csr_col_ind_indextype, csr_val_datatype));
|
|
RETURN_IF_ROCSPARSE_ERROR(f(handle,
|
|
trans_A,
|
|
trans_B,
|
|
m,
|
|
nrhs,
|
|
nnz,
|
|
alpha,
|
|
descr,
|
|
csr_val,
|
|
csr_row_ptr,
|
|
csr_col_ind,
|
|
B,
|
|
ldb,
|
|
info,
|
|
analysis,
|
|
solve,
|
|
p_csrsm_info,
|
|
temp_buffer));
|
|
return rocsparse_status_success;
|
|
}
|