From 30ee79ee373f0779deee456592a73db2fe5ef125 Mon Sep 17 00:00:00 2001 From: conrad Date: Fri, 11 Jul 2025 16:06:01 +1000 Subject: [PATCH] enable fp16 support for SpMat --- include/armadillo_bits/SpMat_meat.hpp | 3 --- include/armadillo_bits/typedef_mat.hpp | 12 ++++++------ 2 files changed, 6 insertions(+), 9 deletions(-) diff --git a/include/armadillo_bits/SpMat_meat.hpp b/include/armadillo_bits/SpMat_meat.hpp index 16134348..9f381b27 100644 --- a/include/armadillo_bits/SpMat_meat.hpp +++ b/include/armadillo_bits/SpMat_meat.hpp @@ -54,9 +54,6 @@ SpMat::~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::value == true )); } diff --git a/include/armadillo_bits/typedef_mat.hpp b/include/armadillo_bits/typedef_mat.hpp index e5ba5f51..de8567d0 100644 --- a/include/armadillo_bits/typedef_mat.hpp +++ b/include/armadillo_bits/typedef_mat.hpp @@ -148,12 +148,12 @@ typedef SpCol sp_cx_vec; typedef SpCol sp_cx_colvec; typedef SpRow sp_cx_rowvec; -//#if defined(ARMA_HAVE_FP16) -//typedef SpMat sp_fp16_mat; -//typedef SpCol sp_fp16_vec; -//typedef SpCol sp_fp16_colvec; -//typedef SpRow sp_fp16_rowvec; -//#endif +#if defined(ARMA_HAVE_FP16) +typedef SpMat sp_fp16_mat; +typedef SpCol sp_fp16_vec; +typedef SpCol sp_fp16_colvec; +typedef SpRow sp_fp16_rowvec; +#endif