Skip to content

Commit f1a94e3

Browse files
committed
follow comments
1 parent bf32411 commit f1a94e3

File tree

4 files changed

+21
-22
lines changed

4 files changed

+21
-22
lines changed

paddle/function/cross_map_normal_op.cpp

Lines changed: 11 additions & 11 deletions
Original file line numberDiff line numberDiff line change
@@ -20,7 +20,7 @@ namespace paddle {
2020
template <>
2121
void CrossMapNormal<DEVICE_TYPE_CPU>(real* outputs,
2222
real* denoms,
23-
real* inputs,
23+
const real* inputs,
2424
size_t numSamples,
2525
size_t channels,
2626
size_t height,
@@ -32,7 +32,7 @@ void CrossMapNormal<DEVICE_TYPE_CPU>(real* outputs,
3232
size_t oneSample = channels * oneImage;
3333

3434
CpuVector outputsV(numSamples * oneSample, outputs);
35-
CpuVector inputsV(numSamples * oneSample, inputs);
35+
CpuVector inputsV(numSamples * oneSample, const_cast<real*>(inputs));
3636
CpuVector denomsV(numSamples * oneSample, denoms);
3737

3838
// f(x) = x * ( 1 + scale * SUM((x)^2) )^(-pow)
@@ -44,7 +44,7 @@ void CrossMapNormal<DEVICE_TYPE_CPU>(real* outputs,
4444
const int end = (int)size + start;
4545
for (size_t i = 0; i < numSamples; i++) {
4646
real* oneDenom = denoms + i * oneSample;
47-
real* oneInput = inputs + i * oneSample;
47+
real* oneInput = const_cast<real*>(inputs) + i * oneSample;
4848
for (int c = 0; c < (int)channels; c++) {
4949
CpuVector denom(oneImage, oneDenom + c * oneImage);
5050
for (int s = start; s < end; s++) {
@@ -61,10 +61,10 @@ void CrossMapNormal<DEVICE_TYPE_CPU>(real* outputs,
6161

6262
template <>
6363
void CrossMapNormalGrad<DEVICE_TYPE_CPU>(real* inputsGrad,
64-
real* inputsValue,
65-
real* outputsValue,
66-
real* outputsGrad,
67-
real* denoms,
64+
const real* inputsValue,
65+
const real* outputsValue,
66+
const real* outputsGrad,
67+
const real* denoms,
6868
size_t numSamples,
6969
size_t channels,
7070
size_t height,
@@ -84,10 +84,10 @@ void CrossMapNormalGrad<DEVICE_TYPE_CPU>(real* inputsGrad,
8484
for (size_t i = 0; i < numSamples; i++) {
8585
size_t sOffset = i * oneSample;
8686
real* oneInputGrad = inputsGrad + sOffset;
87-
real* oneInputValue = inputsValue + sOffset;
88-
real* oneDenom = denoms + sOffset;
89-
real* oneOutputGrad = outputsGrad + sOffset;
90-
real* oneOutputValue = outputsValue + sOffset;
87+
real* oneInputValue = const_cast<real*>(inputsValue) + sOffset;
88+
real* oneDenom = const_cast<real*>(denoms) + sOffset;
89+
real* oneOutputGrad = const_cast<real*>(outputsGrad) + sOffset;
90+
real* oneOutputValue = const_cast<real*>(outputsValue) + sOffset;
9191

9292
for (int c = 0; c < (int)channels; c++) {
9393
size_t cOffset = c * height * width;

paddle/function/cross_map_normal_op.h

Lines changed: 5 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -37,7 +37,7 @@ namespace paddle {
3737
template <DeviceType Device>
3838
void CrossMapNormal(real* outputs,
3939
real* denoms,
40-
real* inputs,
40+
const real* inputs,
4141
size_t numSamples,
4242
size_t channels,
4343
size_t height,
@@ -66,10 +66,10 @@ void CrossMapNormal(real* outputs,
6666
*/
6767
template <DeviceType Device>
6868
void CrossMapNormalGrad(real* inputsGrad,
69-
real* inputsValue,
70-
real* outputsValue,
71-
real* outputsGrad,
72-
real* denoms,
69+
const real* inputsValue,
70+
const real* outputsValue,
71+
const real* outputsGrad,
72+
const real* denoms,
7373
size_t numSamples,
7474
size_t channels,
7575
size_t height,

paddle/function/cross_map_normal_op_gpu.cu

Lines changed: 5 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -63,7 +63,7 @@ __global__ void KeCMRNormOutput(size_t inputSize, const real* in,
6363
template <>
6464
void CrossMapNormal<DEVICE_TYPE_GPU>(real* outputs,
6565
real* denoms,
66-
real* inputs,
66+
const real* inputs,
6767
size_t numSamples,
6868
size_t channels,
6969
size_t height,
@@ -132,10 +132,10 @@ __global__ void KeCMRNormDiff(size_t imageSize, const real* bottom_data,
132132

133133
template <>
134134
void CrossMapNormalGrad<DEVICE_TYPE_GPU>(real* inputsGrad,
135-
real* inputsValue,
136-
real* outputsValue,
137-
real* outputsGrad,
138-
real* denoms,
135+
const real* inputsValue,
136+
const real* outputsValue,
137+
const real* outputsGrad,
138+
const real* denoms,
139139
size_t numSamples,
140140
size_t channels,
141141
size_t height,

paddle/math/tests/test_matrixCompare.cpp

Lines changed: 0 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -18,7 +18,6 @@ limitations under the License. */
1818

1919
#include <gtest/gtest.h>
2020
#include "TensorCheck.h"
21-
#include "paddle/function/Function.h"
2221
#include "paddle/gserver/tests/TestUtil.h"
2322
#include "paddle/math/Matrix.h"
2423
#include "paddle/math/SparseMatrix.h"

0 commit comments

Comments
 (0)