Files
Aurora/src/CudaMatrixPrivate.cuh

15 lines
586 B
Plaintext
Raw Normal View History

2023-10-31 14:35:29 +08:00
#ifndef __CUDAMATRIX_CUH__
#define __CUDAMATRIX_CUH__
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);
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);
#endif // __CUDAMATRIX_H__