nuoya
/
mindspore2022
forked from huawei/mindspore2022
0
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

fix code review bugs

This commit is contained in:
jjfeing 2021-07-28 11:28:57 +08:00
parent 90358906d2
commit 461aae0029
6 changed files with 48 additions and 58 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

79
mindspore/ccsrc/backend/kernel_compiler/kash/kernel_pack.cc
View File

@ -41,103 +41,94 @@ bool CheckHash(const std::string &json_file, const std::string &bin_file, const
}
} // namespace
const std::string KernelPack::Serialize() const {
MS_EXCEPTION_IF_NULL(json_);
MS_EXCEPTION_IF_NULL(kernel_);
std::string buffer;
(void)buffer.append((const char *)json_, json_->len + sizeof(json_->len));
(void)buffer.append((const char *)kernel_, kernel_->len + sizeof(kernel_->len));
return buffer;
}
bool KernelPack::ReadFromJsonFileHelper(std::ifstream &kernelbin) {
size_t binsize = LongToSize(kernelbin.seekg(0, std::ios::end).tellg());
bool KernelPack::ReadFromJsonFileHelper(std::ifstream &kernel_bin) {
size_t bin_size = LongToSize(kernel_bin.seekg(0, std::ios::end).tellg());
// free old data
if (kernel_ != nullptr) {
delete[] kernel_;
kernel_ = nullptr;
}
void *ptr = static_cast<void *>(new (std::nothrow) uint8_t[sizeof(KernelPack) + binsize]);
void *ptr = static_cast<void *>(new (std::nothrow) uint8_t[sizeof(KernelPack) + bin_size]);
if (ptr != nullptr) {
kernel_ = static_cast<FlexArray *>(ptr);
}
if (kernel_ == nullptr) {
MS_LOG(ERROR) << "memory malloc failed.";
kernelbin.close();
MS_LOG(ERROR) << "Memory malloc failed.";
kernel_bin.close();
return false;
}
if (memset_s(kernel_, sizeof(KernelPack) + binsize, 0, sizeof(KernelPack) + binsize) != EOK) {
MS_LOG(ERROR) << "memset kernel_ failed.";
if (memset_s(kernel_, sizeof(KernelPack) + bin_size, 0, sizeof(KernelPack) + bin_size) != EOK) {
MS_LOG(ERROR) << "Memset kernel_ failed.";
delete[] kernel_;
kernel_ = nullptr;
kernelbin.close();
kernel_bin.close();
return false;
}
kernel_->len = binsize;
(void)kernelbin.seekg(0, std::ios::beg);
(void)kernelbin.read(kernel_->contents, SizeToLong(kernel_->len));
kernel_->len = bin_size;
(void)kernel_bin.seekg(0, std::ios::beg);
(void)kernel_bin.read(kernel_->contents, SizeToLong(kernel_->len));
return true;
}
bool KernelPack::ReadFromJsonFile(const std::string &json_f, const std::string &processor) {
if (json_f.length() <= strlen(kJsonSuffix)) {
MS_LOG(ERROR) << "please check json path.";
MS_LOG(ERROR) << "Please check json path, file name: " << json_f;
return false;
}
std::ifstream kerneljson(json_f);
if (!kerneljson.is_open()) {
MS_LOG(DEBUG) << "read json file error, please check kernelmeta.";
std::ifstream kernel_json(json_f);
if (!kernel_json.is_open()) {
MS_LOG(DEBUG) << "Read json file(" << json_f << ") error, please check kernel_meta.";
return false;
}
nlohmann::json js;
kerneljson >> js;
kernel_json >> js;
size_t binsize = LongToSize(kerneljson.seekg(0, std::ios::end).tellg());
void *ptr = static_cast<void *>(new (std::nothrow) uint8_t[sizeof(KernelPack) + binsize]);
size_t bin_size = LongToSize(kernel_json.seekg(0, std::ios::end).tellg());
void *ptr = static_cast<void *>(new (std::nothrow) uint8_t[sizeof(KernelPack) + bin_size]);
if (ptr != nullptr) {
json_ = static_cast<FlexArray *>(ptr);
}
if (json_ == nullptr) {
MS_LOG(ERROR) << "memory malloc failed.";
kerneljson.close();
kernel_json.close();
return false;
}
json_->len = binsize;
(void)kerneljson.seekg(0, std::ios::beg);
(void)kerneljson.read(json_->contents, SizeToLong(json_->len));
json_->len = bin_size;
(void)kernel_json.seekg(0, std::ios::beg);
(void)kernel_json.read(json_->contents, SizeToLong(json_->len));
if (processor == kProcessorCuda) {
std::string bin_f = json_f.substr(0, json_f.length() - 5) + ".ptx";
std::ifstream kernelbin(bin_f);
if (!kernelbin.is_open()) {
MS_LOG(ERROR) << "read kernel ptx file error, please check kernelmeta.";
kerneljson.close();
kernel_json.close();
return false;
}
if (ReadFromJsonFileHelper(kernelbin) == false) {
if (!ReadFromJsonFileHelper(kernelbin)) {
delete[] json_;
json_ = nullptr;
kerneljson.close();
kernel_json.close();
return false;
}
kerneljson.close();
kernel_json.close();
if (!CheckHash(json_f, bin_f, js)) {
return false;
}
return true;
}
std::string binfilesuffix = js["binFileSuffix"];
std::string bin_f = json_f.substr(0, json_f.length() - 5) + binfilesuffix;
if (binfilesuffix.compare(".so") == 0) {
std::string binfile_suffix = js["binFileSuffix"];
std::string bin_f = json_f.substr(0, json_f.length() - 5) + binfile_suffix;
if (binfile_suffix == ".so") {
// change "xx/xx.so" -> "xx/libxx.so"
auto sp = bin_f.rfind('/');
if (sp == std::string::npos) {
MS_LOG(ERROR) << "illegal bin file path " << bin_f;
kerneljson.close();
kernel_json.close();
return false;
}
bin_f = bin_f.substr(0, sp + 1) + "lib" + bin_f.substr(sp + 1, bin_f.length() - sp - 1);
@ -146,20 +137,20 @@ bool KernelPack::ReadFromJsonFile(const std::string &json_f, const std::string &
std::ifstream kernelbin(bin_f, std::ios::binary);
if (!kernelbin.is_open()) {
MS_LOG(ERROR) << "read kernel binary file error, please check kernelmeta.";
kerneljson.close();
kernel_json.close();
delete[] json_;
json_ = nullptr;
return false;
}
MS_LOG(INFO) << "kernelbin_name:" << bin_f;
if (ReadFromJsonFileHelper(kernelbin) == false) {
if (!ReadFromJsonFileHelper(kernelbin)) {
delete[] json_;
json_ = nullptr;
kerneljson.close();
kernel_json.close();
return false;
}
kerneljson.close();
kernel_json.close();
if (!CheckHash(json_f, bin_f, js)) {
return false;
@ -211,7 +202,7 @@ bool KernelPack::LoadKernelMeta(const std::string &json_f, const std::string &pr
kernel_json >> js;
kernel_json.close();
} catch (std::exception &e) {
MS_LOG(WARNING) << "Parse json file error: " << json_f << ", sleep 500ms and retry again.";
MS_LOG(WARNING) << "Parse json file error: " << json_f << ", sleep 500ms and retry again. error ms: " << e.what();
kernel_json.close();
std::this_thread::sleep_for(std::chrono::microseconds(500000));
std::ifstream retry_tmp(json_f);

3
mindspore/ccsrc/backend/kernel_compiler/kernel.h
View File

@ -130,7 +130,6 @@ class KernelPack {
KernelJsonInfo kernel_json_info() const;
bool LoadKernelMeta(const std::string &json_f, const std::string &processor);
bool ReadFromJsonFile(const std::string &json_f, const std::string &processor);
const std::string Serialize() const;
const FlexArray *GetJson() const { return json_; }
const FlexArray *GetKernel() const { return kernel_; }
~KernelPack() {
@ -145,7 +144,7 @@ class KernelPack {
}
private:
bool ReadFromJsonFileHelper(std::ifstream &kernelbin);
bool ReadFromJsonFileHelper(std::ifstream &kernel_bin);
void ParseKernelJson(const nlohmann::json &js);
KernelJsonInfo kernel_json_info_;
FlexArray *json_;

5
mindspore/ccsrc/backend/kernel_compiler/oplib/oplib.cc
View File

@ -355,9 +355,8 @@ std::shared_ptr<OpInfo> OpLib::FindOp(const std::string &op_name, OpImplyType im
}
std::string target_processor = is_gpu ? kCUDA : kAiCore;
for (auto [iter, end] = op_info_.equal_range(op_name); iter != end; ++iter) {
auto &op_info = iter->second;
auto &op_info = (*iter).second;
MS_EXCEPTION_IF_NULL(op_info);
if (op_info->imply_type() != imply_type) {
continue;
}
@ -399,7 +398,7 @@ bool OpLib::GetRefInfo(const std::shared_ptr<OpInfo> &op_info) {
bool OpLib::CheckRepetition(const std::shared_ptr<OpInfo> &op_info) {
MS_EXCEPTION_IF_NULL(op_info);
for (auto [iter, end] = op_info_.equal_range(op_info->op_name()); iter != end; ++iter) {
auto &exist_op_info = iter->second;
auto &exist_op_info = (*iter).second;
MS_EXCEPTION_IF_NULL(exist_op_info);
if (exist_op_info->equals_to(op_info)) {
return true;

1
mindspore/ccsrc/backend/kernel_compiler/oplib/oplib.h
View File

@ -31,7 +31,6 @@ class OpLib {
OpLib() = default;
virtual ~OpLib() = default;
static bool RegOp(const std::string &json_string, const std::string &impl_path);
static void RegOpInfo(const std::shared_ptr<OpInfo> &opinfo) { op_info_.emplace(opinfo->op_name(), opinfo); }
static std::shared_ptr<OpInfo> FindOp(const std::string &op_name, OpImplyType imply_type,
bool is_dynamic_shape = false);

7
mindspore/ccsrc/runtime/device/executor/dynamic_kernel.cc
View File

@ -17,6 +17,7 @@
#include "runtime/device/executor/dynamic_kernel.h"
#include <vector>
#include <algorithm>
#include <stack>
#include "backend/session/anf_runtime_algorithm.h"
#include "backend/optimizer/common/helper.h"
#include "common/trans.h"
@ -57,13 +58,16 @@ void DynamicKernel::RebuildDependTensor() {
depend_tensor_map_.clear();
auto cnode = cnode_ptr_.lock();
MS_EXCEPTION_IF_NULL(cnode);
auto context = MsContext::GetInstance();
MS_EXCEPTION_IF_NULL(context);
for (auto depend : depend_list_) {
auto pre_node_with_index = AnfAlgo::GetPrevNodeOutput(cnode, depend);
bool visit_nop_node = MsContext::GetInstance()->get_param<bool>(MS_CTX_ENABLE_MINDRT) ? false : true;
bool visit_nop_node = !context->get_param<bool>(MS_CTX_ENABLE_MINDRT);
auto output_addr = AnfAlgo::GetPrevNodeMutableOutputAddr(cnode, depend, visit_nop_node);
std::vector<int64_t> shapes = trans::GetRuntimePaddingShape(pre_node_with_index.first, pre_node_with_index.second);
auto host_type = AnfAlgo::GetOutputInferDataType(pre_node_with_index.first, pre_node_with_index.second);
auto out_tensor = std::make_shared<tensor::Tensor>(host_type, shapes);
MS_EXCEPTION_IF_NULL(out_tensor);
out_tensor->set_device_address(output_addr);
auto ret = depend_tensor_map_.try_emplace(depend, out_tensor);
if (!ret.second) {
@ -124,6 +128,7 @@ void DynamicKernel::InferShapeForNopNode(AnfNodePtr *input_node) {
while (!nop_road.empty()) {
auto nop_node = nop_road.top();
MS_EXCEPTION_IF_NULL(nop_node);
AnfAlgo::InferShape(nop_node->cast<CNodePtr>());
nop_road.pop();
}

11
mindspore/ccsrc/runtime/device/executor/dynamic_kernel.h
View File

@ -40,18 +40,15 @@ class DynamicKernel {
virtual void UpdateArgs() = 0;
virtual void Execute() = 0;
virtual void PostExecute() = 0;
bool is_dynamic_shape() const { return is_dynamic_shape_; }
bool is_input_dynamic_shape() const { return is_input_dynamic_shape_; }
bool is_output_dynamic_shape() const { return is_output_dynamic_shape_; }
bool have_depends() const { return !depend_list_.empty(); }
[[nodiscard]] bool is_dynamic_shape() const { return is_dynamic_shape_; }
[[nodiscard]] bool have_depends() const { return !depend_list_.empty(); }
virtual void Initialize();
std::string GetKernelName() const { return cnode_ptr_.lock()->fullname_with_scope(); }
int GetKernelType() const;
[[nodiscard]] int GetKernelType() const;
protected:
void RebuildDependTensor();
void InferShapeRecursive();
void InferShapeForNopNode(AnfNodePtr *input_node);
static void InferShapeForNopNode(AnfNodePtr *input_node);
void *stream_;
const CNodeWeakPtr cnode_ptr_;