diff options
| author | algol <dkazanc@hotmail.com> | 2018-02-22 12:34:39 +0000 |
|---|---|---|
| committer | algol <dkazanc@hotmail.com> | 2018-02-22 12:34:39 +0000 |
| commit | d4d12945f5396b10259fdeeb2b09f99b0e2c6afd (patch) | |
| tree | 378fcd77b72b7e433db2e2b2e121d34d479e988d /Wrappers/Python/src | |
| parent | ce313c7dcf45edaf4c824690d8caa7df5df90120 (diff) | |
| download | regularization-d4d12945f5396b10259fdeeb2b09f99b0e2c6afd.tar.gz regularization-d4d12945f5396b10259fdeeb2b09f99b0e2c6afd.tar.bz2 regularization-d4d12945f5396b10259fdeeb2b09f99b0e2c6afd.tar.xz regularization-d4d12945f5396b10259fdeeb2b09f99b0e2c6afd.zip | |
GPU test fixed, cpu vs gpu test added for ROF-TV
Diffstat (limited to 'Wrappers/Python/src')
| -rw-r--r-- | Wrappers/Python/src/gpu_regularizers.pyx | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/Wrappers/Python/src/gpu_regularizers.pyx b/Wrappers/Python/src/gpu_regularizers.pyx index fcb91cc..c724471 100644 --- a/Wrappers/Python/src/gpu_regularizers.pyx +++ b/Wrappers/Python/src/gpu_regularizers.pyx @@ -345,7 +345,7 @@ def ROFTV2D(np.ndarray[np.float32_t, ndim=2, mode="c"] inputData, # Running CUDA code here TV_ROF_GPU_kernel( &inputData[0,0], &B[0,0], - dims[0], dims[1], 0, + dims[0], dims[1], 1, iterations , time_marching_parameter, regularization_parameter); |