phynexis-deps/armadillo-code
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

enable fp16 support for SpMat

Browse Source
This commit is contained in:
conrad
2025-07-11 16:06:01 +10:00
parent 9264db2876
commit 30ee79ee37
2 changed files with 6 additions and 9 deletions
Download Patch File Download Diff File Expand all files Collapse all files
include/armadillo_bits/SpMat_meat.hpp
-3
View File
@@ -54,9 +54,6 @@ SpMat<eT>::~SpMat()
if(values ) { memory::release(access::rw(values)); }
if(row_indices) { memory::release(access::rw(row_indices)); }
if(col_ptrs ) { memory::release(access::rw(col_ptrs)); }
// FP16 is not currently supported for sparse matrices
arma_type_check(( is_fp16<eT>::value == true ));
}
include/armadillo_bits/typedef_mat.hpp
+6 -6
View File
@@ -148,12 +148,12 @@ typedef SpCol <cx_double> sp_cx_vec;
typedef SpCol <cx_double> sp_cx_colvec;
typedef SpRow <cx_double> sp_cx_rowvec;
//#if defined(ARMA_HAVE_FP16)
//typedef SpMat <fp16> sp_fp16_mat;
//typedef SpCol <fp16> sp_fp16_vec;
//typedef SpCol <fp16> sp_fp16_colvec;
//typedef SpRow <fp16> sp_fp16_rowvec;
//#endif
#if defined(ARMA_HAVE_FP16)
typedef SpMat <fp16> sp_fp16_mat;
typedef SpCol <fp16> sp_fp16_vec;
typedef SpCol <fp16> sp_fp16_colvec;
typedef SpRow <fp16> sp_fp16_rowvec;
#endif