summaryrefslogtreecommitdiffstats
diff options
context:
space:
mode:
authordkazanc <dkazanc@hotmail.com>2019-02-26 15:29:56 +0000
committerdkazanc <dkazanc@hotmail.com>2019-02-26 15:29:56 +0000
commitaa80b7b0895cd7b2701b8c6d4136eeaf12d29929 (patch)
tree70988a437ba7794809f761890d9846c54663a1b4
parent57e60c85cb56de6c77a56737af5e6ad57d801420 (diff)
downloadregularization-aa80b7b0895cd7b2701b8c6d4136eeaf12d29929.tar.gz
regularization-aa80b7b0895cd7b2701b8c6d4136eeaf12d29929.tar.bz2
regularization-aa80b7b0895cd7b2701b8c6d4136eeaf12d29929.tar.xz
regularization-aa80b7b0895cd7b2701b8c6d4136eeaf12d29929.zip
tgv_upd2
-rw-r--r--Core/regularisers_GPU/TGV_GPU_core.cu6
1 files changed, 3 insertions, 3 deletions
diff --git a/Core/regularisers_GPU/TGV_GPU_core.cu b/Core/regularisers_GPU/TGV_GPU_core.cu
index 3f3c67c..849219b 100644
--- a/Core/regularisers_GPU/TGV_GPU_core.cu
+++ b/Core/regularisers_GPU/TGV_GPU_core.cu
@@ -70,7 +70,7 @@ __global__ void DualP_2D_kernel(float *U, float *V1, float *V2, float *P1, float
return;
}
-__global__ void ProjP_2D_kernel(float *P1, float *P2, long dimX, long dimY, long num_total, float alpha1)
+__global__ void ProjP_2D_kernel(float *P1, float *P2, long dimX, long dimY, float alpha1)
{
float grad_magn;
@@ -288,10 +288,10 @@ __global__ void DualP_3D_kernel(float *U, float *V1, float *V2, float *V3, float
else if (i == dimX-1) P1[index] -= sigma*(V1[index]);
else P1[index] = 0.0f;
if ((j >= 0) && (j < dimY-1)) P2[index] += sigma*((U[(dimX*dimY)*k + i*dimX+(j+1)] - U[index]) - V2[index]);
- else if P2[index] -= sigma*(V2[index]);
+ else if (j == dimY-1) P2[index] -= sigma*(V2[index]);
else P2[index] = 0.0f;
if ((k >= 0) && (k < dimZ-1)) P3[index] += sigma*((U[(dimX*dimY)*(k+1) + i*dimX+(j)] - U[index]) - V3[index]);
- else if P3[index] -= sigma*(V3[index]);
+ else if (k == dimZ-1) P3[index] -= sigma*(V3[index]);
else P3[index] = 0.0f;
}
return;