/*! \file */ /* ************************************************************************ * Copyright (C) 2024-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_enum_name.hpp" template inline constexpr int token_count(F f) { return 1; } template inline constexpr int token_count(F f, P... p) { return token_count(p...) + 1; } #define DEF(TOKEN, ...) \ struct TOKEN##_st \ { \ using value_t = int; \ static constexpr int s = token_count(__VA_ARGS__); \ static constexpr TOKEN all[s] = {__VA_ARGS__}; \ }; \ constexpr TOKEN TOKEN##_st::all[TOKEN##_st::s] DEF(rocsparse_indextype, rocsparse_indextype_i32, rocsparse_indextype_i64); DEF(rocsparse_datatype, rocsparse_datatype_f16_r, rocsparse_datatype_f32_r, rocsparse_datatype_f64_r, rocsparse_datatype_f32_c, rocsparse_datatype_f64_c, rocsparse_datatype_i8_r, rocsparse_datatype_u8_r, rocsparse_datatype_i32_r, rocsparse_datatype_u32_r, rocsparse_datatype_bf16_r); DEF(rocsparse_index_base, rocsparse_index_base_zero, rocsparse_index_base_one); DEF(rocsparse_operation, rocsparse_operation_none, rocsparse_operation_transpose, rocsparse_operation_conjugate_transpose); DEF(rocsparse_matrix_type, rocsparse_matrix_type_general, rocsparse_matrix_type_symmetric, rocsparse_matrix_type_hermitian, rocsparse_matrix_type_triangular); DEF(rocsparse_diag_type, rocsparse_diag_type_non_unit, rocsparse_diag_type_unit); DEF(rocsparse_fill_mode, rocsparse_fill_mode_lower, rocsparse_fill_mode_upper); DEF(rocsparse_storage_mode, rocsparse_storage_mode_sorted, rocsparse_storage_mode_unsorted); DEF(rocsparse_action, rocsparse_action_symbolic, rocsparse_action_numeric); DEF(rocsparse_hyb_partition, rocsparse_hyb_partition_auto, rocsparse_hyb_partition_user, rocsparse_hyb_partition_max); DEF(rocsparse_analysis_policy, rocsparse_analysis_policy_reuse, rocsparse_analysis_policy_force); DEF(rocsparse_solve_policy, rocsparse_solve_policy_auto); DEF(rocsparse_direction, rocsparse_direction_row, rocsparse_direction_column); DEF(rocsparse_order, rocsparse_order_row, rocsparse_order_column); DEF(rocsparse_format, rocsparse_format_coo, rocsparse_format_coo_aos, rocsparse_format_csr, rocsparse_format_bsr, rocsparse_format_csc, rocsparse_format_ell, rocsparse_format_bell, rocsparse_format_sell); DEF(rocsparse_sddmm_alg, rocsparse_sddmm_alg_default, rocsparse_sddmm_alg_dense); DEF(rocsparse_itilu0_alg, rocsparse_itilu0_alg_default, rocsparse_itilu0_alg_async_inplace, rocsparse_itilu0_alg_async_split, rocsparse_itilu0_alg_sync_split, rocsparse_itilu0_alg_sync_split_fusion); DEF(rocsparse_spmv_alg, rocsparse_spmv_alg_default, rocsparse_spmv_alg_bsr, rocsparse_spmv_alg_coo, rocsparse_spmv_alg_csr_adaptive, rocsparse_spmv_alg_csr_rowsplit, rocsparse_spmv_alg_ell, rocsparse_spmv_alg_sell, rocsparse_spmv_alg_coo_atomic, rocsparse_spmv_alg_csr_lrb, rocsparse_spmv_alg_csr_nnzsplit); DEF(rocsparse_spsv_alg, rocsparse_spsv_alg_default); DEF(rocsparse_sptrsv_alg, rocsparse_sptrsv_alg_default); DEF(rocsparse_spitsv_alg, rocsparse_spitsv_alg_default); DEF(rocsparse_spsm_alg, rocsparse_spsm_alg_default); DEF(rocsparse_sptrsm_alg, rocsparse_sptrsm_alg_default); DEF(rocsparse_spmm_alg, rocsparse_spmm_alg_default, rocsparse_spmm_alg_bsr, rocsparse_spmm_alg_csr, rocsparse_spmm_alg_coo_segmented, rocsparse_spmm_alg_coo_atomic, rocsparse_spmm_alg_bell, rocsparse_spmm_alg_coo_segmented_atomic, rocsparse_spmm_alg_csr_row_split, rocsparse_spmm_alg_csr_merge); DEF(rocsparse_spgemm_alg, rocsparse_spgemm_alg_default); DEF(rocsparse_spgeam_alg, rocsparse_spgeam_alg_default); DEF(rocsparse_sparse_to_dense_alg, rocsparse_sparse_to_dense_alg_default); DEF(rocsparse_dense_to_sparse_alg, rocsparse_dense_to_sparse_alg_default); DEF(rocsparse_gtsv_interleaved_alg, rocsparse_gtsv_interleaved_alg_default, rocsparse_gtsv_interleaved_alg_thomas, rocsparse_gtsv_interleaved_alg_lu, rocsparse_gtsv_interleaved_alg_qr); DEF(rocsparse_gpsv_interleaved_alg, rocsparse_gpsv_interleaved_alg_default, rocsparse_gpsv_interleaved_alg_qr); #define CASE(VALUE) \ case VALUE: \ { \ if(!strcmp(name, #VALUE)) \ { \ value = VALUE; \ return true; \ } \ break; \ } #define RETURN_INVALID return true; bool rocsparse_indextype_from_name(rocsparse_indextype& value, const char* name) { for(auto v : rocsparse_indextype_st::all) { switch(v) { CASE(rocsparse_indextype_i32); CASE(rocsparse_indextype_i64); } } return false; } bool rocsparse_datatype_from_name(rocsparse_datatype value, const char* name) { for(auto v : rocsparse_datatype_st::all) { switch(v) { CASE(rocsparse_datatype_f16_r); CASE(rocsparse_datatype_f32_r); CASE(rocsparse_datatype_f64_r); CASE(rocsparse_datatype_f32_c); CASE(rocsparse_datatype_f64_c); CASE(rocsparse_datatype_i8_r); CASE(rocsparse_datatype_u8_r); CASE(rocsparse_datatype_i32_r); CASE(rocsparse_datatype_u32_r); CASE(rocsparse_datatype_bf16_r); } } return false; } bool rocsparse_index_base_from_name(rocsparse_index_base value, const char* name) { for(auto v : rocsparse_index_base_st::all) { switch(v) { CASE(rocsparse_index_base_zero); CASE(rocsparse_index_base_one); } } return false; } bool rocsparse_operation_from_name(rocsparse_operation value, const char* name) { for(auto v : rocsparse_operation_st::all) { switch(v) { CASE(rocsparse_operation_none); CASE(rocsparse_operation_transpose); CASE(rocsparse_operation_conjugate_transpose); } } return false; } bool rocsparse_matrix_type_from_name(rocsparse_matrix_type value, const char* name) { for(auto v : rocsparse_matrix_type_st::all) { switch(v) { CASE(rocsparse_matrix_type_general); CASE(rocsparse_matrix_type_symmetric); CASE(rocsparse_matrix_type_hermitian); CASE(rocsparse_matrix_type_triangular); } } return false; } bool rocsparse_diag_type_from_name(rocsparse_diag_type value, const char* name) { for(auto v : rocsparse_diag_type_st::all) { switch(v) { CASE(rocsparse_diag_type_non_unit); CASE(rocsparse_diag_type_unit); } } return false; } bool rocsparse_fill_mode_from_name(rocsparse_fill_mode value, const char* name) { for(auto v : rocsparse_fill_mode_st::all) { switch(v) { CASE(rocsparse_fill_mode_lower); CASE(rocsparse_fill_mode_upper); } } return false; } bool rocsparse_storage_mode_from_name(rocsparse_storage_mode value, const char* name) { for(auto v : rocsparse_storage_mode_st::all) { switch(v) { CASE(rocsparse_storage_mode_sorted); CASE(rocsparse_storage_mode_unsorted); } } return false; } bool rocsparse_action_from_name(rocsparse_action value, const char* name) { for(auto v : rocsparse_action_st::all) { switch(v) { CASE(rocsparse_action_symbolic); CASE(rocsparse_action_numeric); } } return false; } bool rocsparse_hyb_partition_from_name(rocsparse_hyb_partition value, const char* name) { for(auto v : rocsparse_hyb_partition_st::all) { switch(v) { CASE(rocsparse_hyb_partition_auto); CASE(rocsparse_hyb_partition_user); CASE(rocsparse_hyb_partition_max); } } return false; } bool rocsparse_analysis_policy_from_name(rocsparse_analysis_policy value, const char* name) { for(auto v : rocsparse_analysis_policy_st::all) { switch(v) { CASE(rocsparse_analysis_policy_reuse); CASE(rocsparse_analysis_policy_force); } } return false; } bool rocsparse_solve_policy_from_name(rocsparse_solve_policy value, const char* name) { for(auto v : rocsparse_solve_policy_st::all) { switch(v) { CASE(rocsparse_solve_policy_auto); } } return false; } bool rocsparse_direction_from_name(rocsparse_direction value, const char* name) { for(auto v : rocsparse_direction_st::all) { switch(v) { CASE(rocsparse_direction_row); CASE(rocsparse_direction_column); } } return false; } bool rocsparse_order_from_name(rocsparse_order value, const char* name) { for(auto v : rocsparse_order_st::all) { switch(v) { CASE(rocsparse_order_row); CASE(rocsparse_order_column); } } return false; } bool rocsparse_format_from_name(rocsparse_format value, const char* name) { for(auto v : rocsparse_format_st::all) { switch(v) { CASE(rocsparse_format_coo); CASE(rocsparse_format_coo_aos); CASE(rocsparse_format_csr); CASE(rocsparse_format_bsr); CASE(rocsparse_format_csc); CASE(rocsparse_format_ell); CASE(rocsparse_format_bell); CASE(rocsparse_format_sell); } } return false; } bool rocsparse_sddmm_alg_from_name(rocsparse_sddmm_alg value, const char* name) { for(auto v : rocsparse_sddmm_alg_st::all) { switch(v) { CASE(rocsparse_sddmm_alg_default); CASE(rocsparse_sddmm_alg_dense); } } return false; } bool rocsparse_itilu0_alg_from_name(rocsparse_itilu0_alg value, const char* name) { for(auto v : rocsparse_itilu0_alg_st::all) { switch(v) { CASE(rocsparse_itilu0_alg_default); CASE(rocsparse_itilu0_alg_async_inplace); CASE(rocsparse_itilu0_alg_async_split); CASE(rocsparse_itilu0_alg_sync_split); CASE(rocsparse_itilu0_alg_sync_split_fusion); } } return false; } bool rocsparse_spmv_alg_from_name(rocsparse_spmv_alg value, const char* name) { for(auto v : rocsparse_spmv_alg_st::all) { switch(v) { CASE(rocsparse_spmv_alg_default); CASE(rocsparse_spmv_alg_bsr); CASE(rocsparse_spmv_alg_coo); CASE(rocsparse_spmv_alg_csr_adaptive); CASE(rocsparse_spmv_alg_csr_rowsplit); CASE(rocsparse_spmv_alg_ell); CASE(rocsparse_spmv_alg_sell); CASE(rocsparse_spmv_alg_coo_atomic); CASE(rocsparse_spmv_alg_csr_lrb); CASE(rocsparse_spmv_alg_csr_nnzsplit); } } return false; } bool rocsparse_spsv_alg_from_name(rocsparse_spsv_alg value, const char* name) { for(auto v : rocsparse_spsv_alg_st::all) { switch(v) { CASE(rocsparse_spsv_alg_default); } } return false; } bool rocsparse_sptrsv_alg_from_name(rocsparse_sptrsv_alg value, const char* name) { for(auto v : rocsparse_sptrsv_alg_st::all) { switch(v) { CASE(rocsparse_sptrsv_alg_default); } } return false; } bool rocsparse_spitsv_alg_from_name(rocsparse_spitsv_alg value, const char* name) { for(auto v : rocsparse_spitsv_alg_st::all) { switch(v) { CASE(rocsparse_spitsv_alg_default); } } return false; } bool rocsparse_spsm_alg_from_name(rocsparse_spsm_alg value, const char* name) { for(auto v : rocsparse_spsm_alg_st::all) { switch(v) { CASE(rocsparse_spsm_alg_default); } } return false; } bool rocsparse_sptrsm_alg_from_name(rocsparse_sptrsm_alg value, const char* name) { for(auto v : rocsparse_sptrsm_alg_st::all) { switch(v) { CASE(rocsparse_sptrsm_alg_default); } } return false; } bool rocsparse_spmm_alg_from_name(rocsparse_spmm_alg value, const char* name) { for(auto v : rocsparse_spmm_alg_st::all) { switch(v) { CASE(rocsparse_spmm_alg_default); CASE(rocsparse_spmm_alg_bsr); CASE(rocsparse_spmm_alg_csr); CASE(rocsparse_spmm_alg_csr_merge_path); CASE(rocsparse_spmm_alg_coo_segmented); CASE(rocsparse_spmm_alg_coo_atomic); CASE(rocsparse_spmm_alg_bell); CASE(rocsparse_spmm_alg_coo_segmented_atomic); CASE(rocsparse_spmm_alg_csr_row_split); CASE(rocsparse_spmm_alg_csr_merge); } } return false; } bool rocsparse_spgemm_alg_from_name(rocsparse_spgemm_alg value, const char* name) { for(auto v : rocsparse_spgemm_alg_st::all) { switch(v) { CASE(rocsparse_spgemm_alg_default); } } return false; } bool rocsparse_spgeam_alg_from_name(rocsparse_spgeam_alg value, const char* name) { for(auto v : rocsparse_spgeam_alg_st::all) { switch(v) { CASE(rocsparse_spgeam_alg_default); } } return false; } bool rocsparse_sparse_to_dense_alg_from_name(rocsparse_sparse_to_dense_alg value, const char* name) { for(auto v : rocsparse_sparse_to_dense_alg_st::all) { switch(v) { CASE(rocsparse_sparse_to_dense_alg_default); } } return false; } bool rocsparse_dense_to_sparse_alg_from_name(rocsparse_dense_to_sparse_alg value, const char* name) { for(auto v : rocsparse_dense_to_sparse_alg_st::all) { switch(v) { CASE(rocsparse_dense_to_sparse_alg_default); } } return false; } bool rocsparse_gtsv_interleaved_alg_from_name(rocsparse_gtsv_interleaved_alg value, const char* name) { for(auto v : rocsparse_gtsv_interleaved_alg_st::all) { switch(v) { CASE(rocsparse_gtsv_interleaved_alg_default); CASE(rocsparse_gtsv_interleaved_alg_thomas); CASE(rocsparse_gtsv_interleaved_alg_lu); CASE(rocsparse_gtsv_interleaved_alg_qr); } } return false; } bool rocsparse_gpsv_interleaved_alg_from_name(rocsparse_gpsv_interleaved_alg value, const char* name) { for(auto v : rocsparse_gpsv_interleaved_alg_st::all) { switch(v) { CASE(rocsparse_gpsv_interleaved_alg_default); CASE(rocsparse_gpsv_interleaved_alg_qr); } } return false; } #undef CASE #undef RETURN_INVALID #undef DEF #define DEF(TOKEN, ...) \ do \ { \ if(!strcmp(enum_type, #TOKEN)) \ { \ TOKEN tok = (TOKEN)0; \ bool is = TOKEN##_from_name(tok, value); \ if(is == false) \ return false; \ v = (int)tok; \ return true; \ } \ } while(false) bool convert(int& v, const char* enum_type, const char* value) { DEF(rocsparse_indextype); DEF(rocsparse_datatype); DEF(rocsparse_index_base); DEF(rocsparse_operation); DEF(rocsparse_matrix_type); DEF(rocsparse_diag_type); DEF(rocsparse_fill_mode); DEF(rocsparse_storage_mode); DEF(rocsparse_action); DEF(rocsparse_hyb_partition); DEF(rocsparse_analysis_policy); DEF(rocsparse_solve_policy); DEF(rocsparse_direction); DEF(rocsparse_order); DEF(rocsparse_format); DEF(rocsparse_sddmm_alg); DEF(rocsparse_itilu0_alg); DEF(rocsparse_spmv_alg); DEF(rocsparse_spsv_alg); DEF(rocsparse_sptrsv_alg); DEF(rocsparse_spitsv_alg); DEF(rocsparse_spsm_alg); DEF(rocsparse_sptrsm_alg); DEF(rocsparse_spmm_alg); DEF(rocsparse_spgemm_alg); DEF(rocsparse_spgeam_alg); DEF(rocsparse_sparse_to_dense_alg); DEF(rocsparse_dense_to_sparse_alg); DEF(rocsparse_gtsv_interleaved_alg); DEF(rocsparse_gpsv_interleaved_alg); return false; } #undef DEF #define DEF(TOKEN, ...) \ do \ { \ if(!strcmp(enum_type, #TOKEN)) \ { \ v = TOKEN##_st::s; \ return true; \ } \ } while(false) bool get_size(uint64_t& v, const char* enum_type) { DEF(rocsparse_indextype); DEF(rocsparse_datatype); DEF(rocsparse_index_base); DEF(rocsparse_operation); DEF(rocsparse_matrix_type); DEF(rocsparse_diag_type); DEF(rocsparse_fill_mode); DEF(rocsparse_storage_mode); DEF(rocsparse_action); DEF(rocsparse_hyb_partition); DEF(rocsparse_analysis_policy); DEF(rocsparse_solve_policy); DEF(rocsparse_direction); DEF(rocsparse_order); DEF(rocsparse_format); DEF(rocsparse_sddmm_alg); DEF(rocsparse_itilu0_alg); DEF(rocsparse_spmv_alg); DEF(rocsparse_spsv_alg); DEF(rocsparse_sptrsv_alg); DEF(rocsparse_spitsv_alg); DEF(rocsparse_spsm_alg); DEF(rocsparse_sptrsm_alg); DEF(rocsparse_spmm_alg); DEF(rocsparse_spgemm_alg); DEF(rocsparse_spgeam_alg); DEF(rocsparse_sparse_to_dense_alg); DEF(rocsparse_dense_to_sparse_alg); DEF(rocsparse_gtsv_interleaved_alg); DEF(rocsparse_gpsv_interleaved_alg); return false; } #undef DEF #define DEF(TOKEN, ...) \ do \ { \ if(!strcmp(enum_type, #TOKEN)) \ { \ return rocsparse_enum_name(TOKEN##_st::all[index]); \ } \ } while(false) const char* get_name(const char* enum_type, uint64_t index) { DEF(rocsparse_indextype); DEF(rocsparse_datatype); DEF(rocsparse_index_base); DEF(rocsparse_operation); DEF(rocsparse_matrix_type); DEF(rocsparse_diag_type); DEF(rocsparse_fill_mode); DEF(rocsparse_storage_mode); DEF(rocsparse_action); DEF(rocsparse_hyb_partition); DEF(rocsparse_analysis_policy); DEF(rocsparse_solve_policy); DEF(rocsparse_direction); DEF(rocsparse_order); DEF(rocsparse_format); DEF(rocsparse_sddmm_alg); DEF(rocsparse_itilu0_alg); DEF(rocsparse_spmv_alg); DEF(rocsparse_spsv_alg); DEF(rocsparse_sptrsv_alg); DEF(rocsparse_spitsv_alg); DEF(rocsparse_spsm_alg); DEF(rocsparse_sptrsm_alg); DEF(rocsparse_spmm_alg); DEF(rocsparse_spgemm_alg); DEF(rocsparse_spgeam_alg); DEF(rocsparse_sparse_to_dense_alg); DEF(rocsparse_dense_to_sparse_alg); DEF(rocsparse_gtsv_interleaved_alg); DEF(rocsparse_gpsv_interleaved_alg); return nullptr; } #undef DEF