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

Fix：While valid equals zero problem.

Browse Source
This commit is contained in:
sunwen
2025-10-17 09:53:03 +08:00
parent 0bbb02839a
commit 52da7bcd35
1 changed files with 5 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
src/Function2D.cu
View File

@@ -1713,6 +1713,11 @@ __global__ void validKernel(const float* aData, const float* aValid, float* aOut
Aurora::CudaMatrix Aurora::valid(const Aurora::CudaMatrix aData, const Aurora::CudaMatrix aValid) Aurora::CudaMatrix Aurora::valid(const Aurora::CudaMatrix aData, const Aurora::CudaMatrix aValid)
{ {
auto validEnable = Aurora::sum(aValid);
if(validEnable.toHostMatrix()[0] == 0)
{
return aData;
}
int validSize = aValid.getDataSize(); int validSize = aValid.getDataSize();
int rowCount = aData.getDimSize(0); int rowCount = aData.getDimSize(0);
float* hostValid = new float[validSize]; float* hostValid = new float[validSize];