From 52da7bcd35c975c663155b57329f567b2087689b Mon Sep 17 00:00:00 2001
From: sunwen <sunwen@eq9hj.com>
Date: Fri, 17 Oct 2025 09:53:03 +0800
Subject: [PATCH] =?UTF-8?q?Fix=EF=BC=9AWhile=20valid=20equals=20zero=20pro?=
 =?UTF-8?q?blem.?=
MIME-Version: 1.0
Content-Type: text/plain; charset=UTF-8
Content-Transfer-Encoding: 8bit

---
 src/Function2D.cu | 5 +++++
 1 file changed, 5 insertions(+)

diff --git a/src/Function2D.cu b/src/Function2D.cu
index 6838292..57f4f46 100644
--- 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];