sunwen/Aurora
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Add CudaMatrix block, setBlock, setBlockValue.

Browse Source
This commit is contained in:
sunwen
2023-11-09 17:56:13 +08:00
parent 029b86013e
commit d520489823
3 changed files with 271 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
src/CudaMatrixPrivate.cuh
View File

@@ -2,6 +2,8 @@
#ifndef __CUDAMATRIX_CUH__
#define __CUDAMATRIX_CUH__
#include <complex>
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);
@@ -17,5 +19,8 @@ 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);
void thrustFill(float* aBegin, float* aEnd, float aValue);
void thrustFill(float* aBegin, float* aEnd, std::complex<float> aValue);
#endif // __CUDAMATRIX_H__