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

Add functions to 1D(nan, finite, padding, not)

Browse Source
This commit is contained in:
kradchen
2023-11-28 14:26:46 +08:00
parent 1716b139c4
commit 3438121b0c
4 changed files with 178 additions and 5 deletions
Download Patch File Download Diff File Expand all files Collapse all files

72
src/Function1D.cu
View File

@@ -326,6 +326,78 @@ CudaMatrix Aurora::sign(const CudaMatrix&& aMatrix)
return Aurora::CudaMatrix::fromRawData(data, aMatrix.getDimSize(0), aMatrix.getDimSize(1), aMatrix.getDimSize(2), aMatrix.getValueType());
}
void Aurora::nantoval(CudaMatrix& aMatrix,float val){
auto lambda = [=] __host__ __device__ (const float& x){
return ::isnan(x)?val:x;
};
thrust::transform(thrust::device,aMatrix.getData(),
aMatrix.getData()+aMatrix.getDataSize(),aMatrix.getData(),lambda);
}
CudaMatrix Aurora::isnan(const CudaMatrix& aMatrix){
size_t size = aMatrix.getDataSize() * aMatrix.getValueType();
float* data = nullptr;
cudaMalloc((void**)&data, sizeof(float) * size);
auto lambda = [=] __host__ __device__ (const float& x){
return ::isnan(x)?1.0:0;
};
thrust::transform(thrust::device,aMatrix.getData(),aMatrix.getData()+aMatrix.getDataSize(),
data,lambda);
return Aurora::CudaMatrix::fromRawData(data, aMatrix.getDimSize(0),
aMatrix.getDimSize(1), aMatrix.getDimSize(2), aMatrix.getValueType());
}
CudaMatrix Aurora::isfinite(const CudaMatrix& aMatrix){
size_t size = aMatrix.getDataSize() * aMatrix.getValueType();
float* data = nullptr;
cudaMalloc((void**)&data, sizeof(float) * size);
auto lambda = [=] __host__ __device__ (const float& x){
return ::isfinite(x)?1.0:0;
};
thrust::transform(thrust::device,aMatrix.getData(),aMatrix.getData()+aMatrix.getDataSize(),
data,lambda);
return Aurora::CudaMatrix::fromRawData(data, aMatrix.getDimSize(0),
aMatrix.getDimSize(1), aMatrix.getDimSize(2), aMatrix.getValueType());
}
void Aurora::padding(CudaMatrix& aMatrix, int aIndex, float aValue){
if(aMatrix.isNull() || !aMatrix.isVector() || aMatrix.isComplex())
{
std::cerr<<"padding only support real vector"<<std::endl;
return;
}
if (aMatrix.getDataSize()>aIndex){
aMatrix.setValue(aIndex, aValue);
return;
}
//长度不足需补齐
size_t size = (aIndex+1) ;
float* data = nullptr;
cudaMalloc((void**)&data, sizeof(float) * size);
cudaMemcpy(data, aMatrix.getData(), aMatrix.getDataSize(), cudaMemcpyDeviceToDevice);
thrust::fill_n(thrust::device, data+aMatrix.getDataSize(),size-aMatrix.getDataSize(),aValue);
aMatrix=CudaMatrix::fromRawData(data,size,1,1,aMatrix.getValueType());
}
CudaMatrix Aurora::auroraNot(const CudaMatrix& aMatrix){
return auroraNot(std::forward<CudaMatrix&&>(aMatrix.deepCopy()));
}
CudaMatrix Aurora::auroraNot(CudaMatrix&& aMatrix){
size_t size = aMatrix.getDataSize() * aMatrix.getValueType();
float* data = nullptr;
cudaMalloc((void**)&data, sizeof(float) * size);
auto lambda = [=] __host__ __device__ (const float& x){
return x<=0?1.0:0;
};
thrust::transform(thrust::device,aMatrix.getData(),aMatrix.getData()+aMatrix.getDataSize(),
data,lambda);
return Aurora::CudaMatrix::fromRawData(data, aMatrix.getDimSize(0),
aMatrix.getDimSize(1), aMatrix.getDimSize(2), aMatrix.getValueType());
}
void Aurora::compareSet(CudaMatrix& aValueMatrix,float compareValue, float newValue,CompareOp op)
{

27
src/Function1D.cuh
View File

@@ -77,11 +77,32 @@ namespace Aurora
CudaMatrix intersect(const CudaMatrix& aMatrix1, const CudaMatrix& aMatrix2);
CudaMatrix intersect(const CudaMatrix& aMatrix1, const CudaMatrix& aMatrix2, CudaMatrix& aIa);
CudaMatrix intersect(const CudaMatrix& aMatrix1, const CudaMatrix& aMatrix2, CudaMatrix& aIa); /**
* 将所有nan值设置为特定值
* @attention 直接在原数据上进行修改！
* @param aMatrix 向量
* @param val 指定值
*/
void nantoval(CudaMatrix& aMatrix,float val);
CudaMatrix isnan(const CudaMatrix& aMatrix);
CudaMatrix isfinite(const CudaMatrix& aMatrix);
/**
* 使用特定值补齐矩阵，默认为设置原数据矩阵数据到制定长度索引的所有值为制定值
* @attention 直接在原数据上进行修改！不支持复数！
* @param aMatrix 向量
* @param aIndex 长度索引
* @param aValue 指定值
*/
void padding(CudaMatrix& aMatrix, int aIndex, float aValue);
CudaMatrix auroraNot(const CudaMatrix& aMatrix);
CudaMatrix auroraNot(CudaMatrix&& aMatrix);
// ------compareSet----------------------------------------------------
void compareSet(CudaMatrix& aValueMatrix,float compareValue, float newValue,CompareOp op);
void compareSet(CudaMatrix& aValueMatrix,CudaMatrix& aCompareMatrix,float compareValue, float newValue,CompareOp op);
void compareSet(CudaMatrix& aDesAndCompareMatrix,CudaMatrix& aOtherCompareMatrix, float newValue,CompareOp op);