Browse Source

clean_code_for_master

tags/v1.1.0
yefeng 5 years ago
parent
commit
86dff13835
5 changed files with 16 additions and 16 deletions
  1. +7
    -7
      mindspore/lite/src/common/utils.cc
  2. +1
    -1
      mindspore/lite/src/common/utils.h
  3. +2
    -2
      mindspore/lite/src/lite_kernel.cc
  4. +2
    -2
      mindspore/lite/src/lite_kernel.h
  5. +4
    -4
      mindspore/lite/src/runtime/kernel/opencl/subgraph_opencl_kernel.cc

+ 7
- 7
mindspore/lite/src/common/utils.cc View File

@@ -52,20 +52,20 @@ uint64_t GetTimeUs() {
return ret_val;
}

std::string RemoveSubStr(const std::string &from, const std::string &subStr, RemoveSubStrMode mode) {
std::string RemoveSubStr(const std::string &from, const std::string &sub_str, RemoveSubStrMode mode) {
std::string result = from;
if (mode == PREFIX) {
if (from.substr(0, subStr.length()) == subStr) {
result = from.substr(subStr.size());
if (from.substr(0, sub_str.length()) == sub_str) {
result = from.substr(sub_str.size());
}
} else if (mode == SUFFIX) {
if (from.rfind(subStr) == from.size() - subStr.size()) {
result = from.substr(0, from.size() - subStr.size());
if (from.rfind(sub_str) == from.size() - sub_str.size()) {
result = from.substr(0, from.size() - sub_str.size());
}
} else {
size_t index;
while ((index = result.find(subStr)) != std::string::npos) {
result = result.erase(index, subStr.size());
while ((index = result.find(sub_str)) != std::string::npos) {
result = result.erase(index, sub_str.size());
}
}



+ 1
- 1
mindspore/lite/src/common/utils.h View File

@@ -151,7 +151,7 @@ std::vector<std::string> Tokenize(const std::string &src, const std::string &del
enum RemoveSubStrMode { PREFIX, SUFFIX, ANY };

// remove redundant charactor
std::string RemoveSubStr(const std::string &from, const std::string &subStr, RemoveSubStrMode mode = ANY);
std::string RemoveSubStr(const std::string &from, const std::string &sub_str, RemoveSubStrMode mode = ANY);

template <typename T>
inline Option<T> GenericParseValue(const std::string &value) {


+ 2
- 2
mindspore/lite/src/lite_kernel.cc View File

@@ -281,8 +281,8 @@ int LiteKernelUtil::TopologicalSortKernels(std::vector<kernel::LiteKernel *> *ke
void LiteKernelUtil::InitIOKernels(std::vector<kernel::LiteKernel *> &kernels) {
for (auto *kernel : kernels) {
// clean io kernels
kernel->set_in_kernel({});
kernel->set_out_kernel({});
kernel->set_in_kernels({});
kernel->set_out_kernels({});
// find io kernels
for (auto *search_kernel : kernels) {
if (search_kernel == kernel) {


+ 2
- 2
mindspore/lite/src/lite_kernel.h View File

@@ -146,9 +146,9 @@ class LiteKernel {
}
}

void set_in_kernel(const std::vector<LiteKernel *> &kernel) { this->in_kernels_ = kernel; }
void set_in_kernels(const std::vector<LiteKernel *> &kernel) { this->in_kernels_ = kernel; }

void set_out_kernel(const std::vector<LiteKernel *> &kernel) { this->out_kernels_ = kernel; }
void set_out_kernels(const std::vector<LiteKernel *> &kernel) { this->out_kernels_ = kernel; }

const std::vector<LiteKernel *> &in_kernels() const { return this->in_kernels_; }



+ 4
- 4
mindspore/lite/src/runtime/kernel/opencl/subgraph_opencl_kernel.cc View File

@@ -53,10 +53,10 @@ void SubGraphOpenCLKernel::ReplaceOutTensorAndKernelToNull(
nullptr);
if (mem_type == MemType::IMG) {
jv->set_in_tensors(tensors);
jv->set_in_kernel(kernels);
jv->set_in_kernels(kernels);
} else {
jv->set_out_tensors(tensors);
jv->set_out_kernel(kernels);
jv->set_out_kernels(kernels);
}
}
}
@@ -85,11 +85,11 @@ void SubGraphOpenCLKernel::ReplaceOutTensorAndKernelToConvert(const lite::Tensor
tensors.emplace_back(new_tensor);
}
if (mem_type == MemType::IMG) {
iv->set_in_kernel(kernels);
iv->set_in_kernels(kernels);
iv->set_in_tensors(tensors);
in_opencl_op->AddOutKernel(iv);
} else {
iv->set_out_kernel(kernels);
iv->set_out_kernels(kernels);
iv->set_out_tensors(tensors);
in_convert_op->AddInKernel(iv);
}


Loading…
Cancel
Save