2023-10-31 14:35:29 +08:00
|
|
|
|
|
|
|
|
#ifndef __CUDAMATRIX_CUH__
|
|
|
|
|
#define __CUDAMATRIX_CUH__
|
|
|
|
|
|
2023-11-09 17:56:13 +08:00
|
|
|
#include <complex>
|
2023-11-21 13:13:28 +08:00
|
|
|
namespace{
|
|
|
|
|
enum CompareType
|
|
|
|
|
{
|
|
|
|
|
G,GE,E,NE,LE,L
|
|
|
|
|
};
|
|
|
|
|
}
|
2023-10-31 14:35:29 +08:00
|
|
|
void unaryAdd(float* in1, float* in2, float* out, unsigned long length);
|
|
|
|
|
void unaryAdd(float* in1, const float& in2, float* out, unsigned long length);
|
|
|
|
|
void unaryMul(float* in1, float* in2, float* out, unsigned long length);
|
2023-12-08 16:18:05 +08:00
|
|
|
void unaryMulc(float* in1, float* in2, float* out, unsigned long length);
|
|
|
|
|
|
2023-11-01 14:31:29 +08:00
|
|
|
void unaryMul(float* in1, const float& in2, float* out, unsigned long length);
|
|
|
|
|
|
2023-10-31 14:35:29 +08:00
|
|
|
void unaryNeg(float* in1, float* out, unsigned long length);
|
|
|
|
|
void unaryPow(float* in1, float N,float* out, unsigned long length);
|
|
|
|
|
|
|
|
|
|
void unarySub(float* in1, float* in2, float* out, unsigned long length);
|
|
|
|
|
void unaryDiv(float* in1, float* in2, float* out, unsigned long length);
|
2023-12-08 16:18:05 +08:00
|
|
|
void unaryDivc(float* in1, float* in2, float* out, unsigned long length);
|
|
|
|
|
|
2023-11-01 14:31:29 +08:00
|
|
|
void unarySub(const float& in1, float* in2, float* out, unsigned long length);
|
|
|
|
|
void unaryDiv(const float& in1, float* in2, float* out, unsigned long length);
|
|
|
|
|
void unarySub(float* in1, const float& in2, float* out, unsigned long length);
|
|
|
|
|
void unaryDiv(float* in1, const float& in2, float* out, unsigned long length);
|
2023-10-31 14:35:29 +08:00
|
|
|
|
2023-11-21 13:13:28 +08:00
|
|
|
|
|
|
|
|
void unaryCompare(float* in1, const float& in2, float* out, unsigned long length,int type);
|
|
|
|
|
void unaryCompare(const float& in1, float* in2, float* out, unsigned long length, int type);
|
|
|
|
|
void unaryCompare(float* in1, float* in2, float* out, unsigned long length, int type);
|
|
|
|
|
|
|
|
|
|
|
2023-11-09 17:56:13 +08:00
|
|
|
void thrustFill(float* aBegin, float* aEnd, float aValue);
|
|
|
|
|
void thrustFill(float* aBegin, float* aEnd, std::complex<float> aValue);
|
|
|
|
|
|
2023-10-31 14:35:29 +08:00
|
|
|
|
|
|
|
|
#endif // __CUDAMATRIX_H__
|