Fix：While valid equals zero problem.

2025-10-17 09:53:03 +08:00
parent 0bbb02839a
commit 52da7bcd35
1 changed files with 5 additions and 0 deletions
--- a/src/Function2D.cu
+++ b/src/Function2D.cu
@@ -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)
 {
+    auto validEnable = Aurora::sum(aValid);
+    if(validEnable.toHostMatrix()[0] == 0)
+    {
+        return aData;
+    }
    int validSize = aValid.getDataSize();
    int rowCount = aData.getDimSize(0);
    float* hostValid = new float[validSize];