Browse Source

remove debuge code

tags/v1.0.0
lz 5 years ago
parent
commit
ac00c3809b
3 changed files with 9 additions and 8 deletions
  1. +5
    -5
      mindspore/lite/nnacl/fp32_grad/batch_norm.c
  2. +3
    -3
      mindspore/lite/nnacl/fp32_grad/softmax_grad.c
  3. +1
    -0
      mindspore/lite/src/executor.cc

+ 5
- 5
mindspore/lite/nnacl/fp32_grad/batch_norm.c View File

@@ -33,13 +33,13 @@ static void meanVar(const float *in, int size, int ch, float eps, float *mean, f
for (int f = 0; f < ch; ++f) {
mean[f] /= N;
}
for (int f=0; f< ch; f++) {
for (int f = 0; f < ch; f++) {
float tvar = 0;
for (int i =0; i< N; i++) {
float x = in[i*ch +f];
tvar += (x-mean[f]) *(x-mean[f]);
for (int i = 0; i < N; i++) {
float x = in[i * ch + f];
tvar += (x - mean[f]) * (x - mean[f]);
}
invar[f] = 1.0f/(sqrt(tvar/N+eps));
invar[f] = 1.0f / (sqrt(tvar / N + eps));
}
}



+ 3
- 3
mindspore/lite/nnacl/fp32_grad/softmax_grad.c View File

@@ -39,9 +39,9 @@ void SoftmaxGrad(const float *input_ptr, const float *yt_ptr, float *output_ptr,
dim /= outter_size;
memcpy(output_ptr, yt_ptr, ele_size * sizeof(float));

int M = input_shape[axis];
int N = inner_size;
int K = 1;
const int M = input_shape[axis];
const int N = inner_size;
const int K = 1;
for (int i = 0; i < outter_size; i++) {
int outter_offset = i * dim;
memset(sum_data, 0.0f, inner_size * sizeof(float));


+ 1
- 0
mindspore/lite/src/executor.cc View File

@@ -51,6 +51,7 @@ int Executor::Run(std::vector<Tensor *> &in_tensors, std::vector<Tensor *> &out_
MS_LOG(ERROR) << "run kernel before_callback failed, name: " << kernel->name();
}
}

auto ret = kernel->Run();
if (0 != ret) {
MS_LOG(ERROR) << "run kernel failed, name: " << kernel->name();


Loading…
Cancel
Save