jiri/LA_library
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

NRVec::iamax()

Browse Source
This commit is contained in:
Jiri Pittner
2026-03-18 14:53:52 +01:00
parent fa59833e1e
commit 4bacfc02f9
1 changed files with 37 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

37
vec.h
View File

@@ -445,7 +445,9 @@ public:
const int findthr(const T &val, const typename LA_traits<T>::normtype &thr=0) const; const int findthr(const T &val, const typename LA_traits<T>::normtype &thr=0) const;
//! determine the maximal element (in the absolute value) of this vector //! determine the maximal element (in the absolute value) of this vector
inline int iamax() const;
inline const T amax() const; inline const T amax() const;
//! determine the minimal element (in the absolute value) of this vector //! determine the minimal element (in the absolute value) of this vector
inline const T amin() const; inline const T amin() const;
@@ -539,6 +541,7 @@ public:
explicit NRVec_from1(const std::list<T> l) : NRVec<T>(l) {}; explicit NRVec_from1(const std::list<T> l) : NRVec<T>(l) {};
inline const T& operator[] (const int i) const; inline const T& operator[] (const int i) const;
inline T& operator[] (const int i); inline T& operator[] (const int i);
inline int iamax() const {return NRVec<T>::iamax() + 1;};
}; };
@@ -1944,6 +1947,23 @@ inline const double NRVec<double>::amax() const {
return ret; return ret;
} }
template<>
inline int NRVec<double>::iamax() const {
int ret;
#ifdef CUDALA
if(location == cpu){
#endif
ret = cblas_idamax(nn, v, 1);
#ifdef CUDALA
}else{
ret = cublasIdamax(nn, v, 1) ;
TEST_CUBLAS("cublasIdamax");
}
#endif
return ret;
}
/***************************************************************************//** /***************************************************************************//**
* for this real vector \f$\vec{x}\f$ determine the element with smallest absolute value * for this real vector \f$\vec{x}\f$ determine the element with smallest absolute value
* @return \f$\vec{x}_i\f$ where \f$\left|\vec{x]_i\right|=\mathrm{min}_{j}\left|\vec{x}_{j}\right|\f$ * @return \f$\vec{x}_i\f$ where \f$\left|\vec{x]_i\right|=\mathrm{min}_{j}\left|\vec{x}_{j}\right|\f$
@@ -1994,6 +2014,23 @@ inline const std::complex<double> NRVec<std::complex<double> >::amax() const {
return ret; return ret;
} }
template<>
inline int NRVec<std::complex<double> >::iamax() const {
int ret;
#ifdef CUDALA
if(location == cpu){
#endif
ret = cblas_izamax(nn, v, 1);
#ifdef CUDALA
}else{
ret = cublasIzamax(nn, (cuDoubleComplex*)v, 1);
TEST_CUBLAS("cublasIzamax");
}
#endif
return ret;
}
/***************************************************************************//** /***************************************************************************//**
* for a given complex vector \f$\vec{v}\f$, determine the smallest index of the minimum * for a given complex vector \f$\vec{v}\f$, determine the smallest index of the minimum
* magnitude element, i.e. minimal element in the 1-norm * magnitude element, i.e. minimal element in the 1-norm