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];