mirror of
https://github.com/amd/blis.git
synced 2026-05-04 06:21:12 +00:00
f164c7fe70
- Functions to convert to cblas enums from char. - Functions to print matrix and vector elements. - Functions to set matrix and vector elements with the given value. AMD-Internal: [CPUPL-2732] Change-Id: I1046b9578c8456e89eddba4a4e8577016b9361ca
82 lines
3.1 KiB
C++
82 lines
3.1 KiB
C++
/*
|
|
|
|
BLIS
|
|
An object-based framework for developing high-performance BLAS-like
|
|
libraries.
|
|
|
|
Copyright (C) 2023, Advanced Micro Devices, Inc. All rights reserved.
|
|
|
|
Redistribution and use in source and binary forms, with or without
|
|
modification, are permitted provided that the following conditions are
|
|
met:
|
|
- Redistributions of source code must retain the above copyright
|
|
notice, this list of conditions and the following disclaimer.
|
|
- Redistributions in binary form must reproduce the above copyright
|
|
notice, this list of conditions and the following disclaimer in the
|
|
documentation and/or other materials provided with the distribution.
|
|
- Neither the name(s) of the copyright holder(s) nor the names of its
|
|
contributors may be used to endorse or promote products derived
|
|
from this software without specific prior written permission.
|
|
|
|
THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
|
|
"AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
|
|
LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR
|
|
A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT
|
|
HOLDER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL,
|
|
SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT
|
|
LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
|
|
DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
|
|
THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
|
|
(INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE
|
|
OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
|
|
|
|
*/
|
|
|
|
#pragma once
|
|
|
|
#include <iostream>
|
|
#include "common/type_info.h"
|
|
|
|
namespace std {
|
|
// Overload std::abs to work with scomplex and dcomplex.
|
|
float abs(const scomplex x);
|
|
double abs(const dcomplex x);
|
|
// Overload the stream operator to be able to print scomplex in error messages.
|
|
ostream& operator<<(ostream& os, const scomplex& x);
|
|
ostream& operator<<(ostream& os, const dcomplex& x);
|
|
}
|
|
|
|
// Operator overloading for scomplex and dcomplex types.
|
|
scomplex operator+(const scomplex x, const scomplex y);
|
|
dcomplex operator+(const dcomplex x, const dcomplex y);
|
|
|
|
scomplex operator-(const scomplex x, const scomplex y);
|
|
dcomplex operator-(const dcomplex x, const dcomplex y);
|
|
|
|
scomplex operator*(const scomplex x, const scomplex y);
|
|
dcomplex operator*(const dcomplex x, const dcomplex y);
|
|
|
|
bool operator== (const scomplex x, const scomplex y);
|
|
bool operator== (const dcomplex x, const dcomplex y);
|
|
|
|
bool operator!= (const scomplex x, const scomplex y);
|
|
bool operator!= (const dcomplex x, const dcomplex y);
|
|
|
|
// Since we may only specialise template functions and classes
|
|
// in the std namespace for custom types, and std::to_string is not
|
|
// a template function, we put to_string into our namespace.
|
|
namespace testinghelpers {
|
|
template<typename T>
|
|
std::string to_string(const T& x) {
|
|
if constexpr (testinghelpers::type_info<T>::is_complex)
|
|
{
|
|
std::string ss = "(" + std::to_string(x.real) + ", " + std::to_string(x.imag) + ")";
|
|
return ss;
|
|
}
|
|
else
|
|
{
|
|
return std::to_string(x);
|
|
}
|
|
}
|
|
|
|
} |