m53297601
/
mindspore
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

!7474 support int64 shape

Browse Source 
Merge pull request !7474 from yihuaijie/int64
pull/7474/MERGE
mindspore-ci-bot 4 years ago committed by Gitee
parent 615a705823 d7faa77b5e
commit 5792d50ca7
534 changed files with 3832 additions and 3530 deletions
Show Stats Download Patch File Download Diff File

10
mindspore/ccsrc/backend/kernel_compiler/aicpu/aicpu_kernel_build.cc
Unescape View File

@ -116,7 +116,7 @@ void ParseAttrValue(const std::string &type, const std::string &attr_name, const
MS_EXCEPTION_IF_NULL(node_attr); MS_EXCEPTION_IF_NULL(node_attr);
MS_EXCEPTION_IF_NULL(value); MS_EXCEPTION_IF_NULL(value);
if (type == "int") { if (type == "int") {
auto attr_value = GetValue<int>(value); auto attr_value = static_cast<int>(GetValue<int64_t>(value));
(*node_attr)[attr_name].set_i(attr_value); (*node_attr)[attr_name].set_i(attr_value);
} else if (type == "str") { } else if (type == "str") {
auto attr_value = GetValue<std::string>(value); auto attr_value = GetValue<std::string>(value);
@ -128,15 +128,15 @@ void ParseAttrValue(const std::string &type, const std::string &attr_name, const
auto attr_value = GetValue<float>(value); auto attr_value = GetValue<float>(value);
(*node_attr)[attr_name].set_f(attr_value); (*node_attr)[attr_name].set_f(attr_value);
} else if (type == "listInt") { } else if (type == "listInt") {
std::vector<int> attr_value; std::vector<int64_t> attr_value;
auto value_type = value->type(); auto value_type = value->type();
MS_EXCEPTION_IF_NULL(value_type); MS_EXCEPTION_IF_NULL(value_type);
auto value_type_str = value_type->ToString(); auto value_type_str = value_type->ToString();
if (value_type_str == "Int32") { if (value_type_str == "Int64") {
int data = GetValue<int>(value); int64_t data = GetValue<int64_t>(value);
attr_value.push_back(data); attr_value.push_back(data);
} else { } else {
attr_value = GetValue<std::vector<int>>(value); attr_value = GetValue<std::vector<int64_t>>(value);
} }
mindspore::AttrValue input_shape_attr; mindspore::AttrValue input_shape_attr;
mindspore::AttrValue_ArrayValue *input_shape_attr_list = input_shape_attr.mutable_array(); mindspore::AttrValue_ArrayValue *input_shape_attr_list = input_shape_attr.mutable_array();

2
mindspore/ccsrc/backend/kernel_compiler/akg/akg_kernel_json_decoder.cc
Unescape View File

@ -168,7 +168,7 @@ class CNodeDecoder {
output_formats_.push_back(output_desc[kJsonKeyFormat]); output_formats_.push_back(output_desc[kJsonKeyFormat]);
output_types_.push_back(DtypeToTypeId(output_desc[kJsonKeyDataType])); output_types_.push_back(DtypeToTypeId(output_desc[kJsonKeyDataType]));
auto get_item = auto get_item =
func_graph->NewCNode({NewValueNode(prim::kPrimTupleGetItem), cnode_, NewValueNode(SizeToInt(j))}); func_graph->NewCNode({NewValueNode(prim::kPrimTupleGetItem), cnode_, NewValueNode(SizeToLong(j))});
func_graph->AddNode(get_item); func_graph->AddNode(get_item);
nodes_map_[output_desc[kJsonKeyTensorName]] = get_item; nodes_map_[output_desc[kJsonKeyTensorName]] = get_item;
} }

13
mindspore/ccsrc/backend/kernel_compiler/akg/akg_kernel_json_generator.cc
Unescape View File

@ -35,7 +35,10 @@ std::vector<int> GetDynInputSize(const AnfNodePtr &anf_node) {
auto primitive = AnfAlgo::GetCNodePrimitive(anf_node); auto primitive = AnfAlgo::GetCNodePrimitive(anf_node);
MS_EXCEPTION_IF_NULL(primitive); MS_EXCEPTION_IF_NULL(primitive);
if (primitive->HasAttr(kAttrDynInputSizes)) { if (primitive->HasAttr(kAttrDynInputSizes)) {
dyn_input_sizes = GetValue<const std::vector<int>>(primitive->GetAttr(kAttrDynInputSizes)); std::vector<int64_t> dyn_input_sizes_me =
GetValue<const std::vector<int64_t>>(primitive->GetAttr(kAttrDynInputSizes));
(void)std::transform(dyn_input_sizes_me.begin(), dyn_input_sizes_me.end(), std::back_inserter(dyn_input_sizes),
[](const int64_t &value) { return static_cast<int>(value); });
} }
return dyn_input_sizes; return dyn_input_sizes;
} }
@ -256,7 +259,7 @@ void AkgKernelJsonGenerator::GetAttrJson(const AnfNodePtr &anf_node, const std::
std::string type = op_attr->type(); std::string type = op_attr->type();
(*attr_json)[kJsonKeyDataType] = type; (*attr_json)[kJsonKeyDataType] = type;
if (type == "int") { if (type == "int") {
(*attr_json)[kJsonKeyValue] = GetValue<int>(attr_value); (*attr_json)[kJsonKeyValue] = static_cast<int>(GetValue<int64_t>(attr_value));
} else if (type == "str") { } else if (type == "str") {
(*attr_json)[kJsonKeyValue] = GetValue<std::string>(attr_value); (*attr_json)[kJsonKeyValue] = GetValue<std::string>(attr_value);
} else if (type == "bool") { } else if (type == "bool") {
@ -264,7 +267,11 @@ void AkgKernelJsonGenerator::GetAttrJson(const AnfNodePtr &anf_node, const std::
} else if (type == "float") { } else if (type == "float") {
(*attr_json)[kJsonKeyValue] = GetValue<float>(attr_value); (*attr_json)[kJsonKeyValue] = GetValue<float>(attr_value);
} else if (type == "listInt") { } else if (type == "listInt") {
(*attr_json)[kJsonKeyValue] = GetValue<std::vector<int>>(attr_value); std::vector<int> list_int;
std::vector<int64_t> list_int_me = GetValue<std::vector<int64_t>>(attr_value);
(void)std::transform(list_int_me.begin(), list_int_me.end(), std::back_inserter(list_int),
[](const int64_t &value) { return static_cast<int>(value); });
(*attr_json)[kJsonKeyValue] = list_int;
} else if (type == "listStr") { } else if (type == "listStr") {
std::vector<std::string> data_format; std::vector<std::string> data_format;
if (op_attr->name() == kArgDataformat) { if (op_attr->name() == kArgDataformat) {

28
mindspore/ccsrc/backend/kernel_compiler/common_utils.cc
Unescape View File

@ -235,7 +235,7 @@ size_t GetDtypeNbyte(const std::string &dtypes) {
} }
bool SetInputKernelBuilderInfo(const std::vector<std::shared_ptr<OpIOInfo>> &inputs, size_t real_input_num, bool SetInputKernelBuilderInfo(const std::vector<std::shared_ptr<OpIOInfo>> &inputs, size_t real_input_num,
size_t builder_idex, const std::vector<int> &dyn_input_sizes, size_t builder_idex, const std::vector<int64_t> &dyn_input_sizes,
const std::shared_ptr<KernelBuildInfo::KernelBuildInfoBuilder> &builder) { const std::shared_ptr<KernelBuildInfo::KernelBuildInfoBuilder> &builder) {
MS_EXCEPTION_IF_NULL(builder); MS_EXCEPTION_IF_NULL(builder);
@ -262,7 +262,7 @@ bool SetInputKernelBuilderInfo(const std::vector<std::shared_ptr<OpIOInfo>> &inp
return false; return false;
} }
for (int t = 0; t < dyn_input_sizes[dyn_input_idx]; t++) { for (int64_t t = 0; t < dyn_input_sizes[dyn_input_idx]; t++) {
kernel_info_index++; kernel_info_index++;
auto type_id = DtypeToTypeId(dtypes[builder_idex]); auto type_id = DtypeToTypeId(dtypes[builder_idex]);
inputs_device_type.push_back(type_id); inputs_device_type.push_back(type_id);
@ -376,11 +376,11 @@ bool ParseMetadata(const CNodePtr &kernel_node, const std::shared_ptr<const OpIn
size_t real_output_num = AnfAlgo::GetOutputTensorNum(kernel_node); size_t real_output_num = AnfAlgo::GetOutputTensorNum(kernel_node);
std::vector<std::shared_ptr<OpIOInfo>> inputs = op_info_ptr->inputs_ptr(); std::vector<std::shared_ptr<OpIOInfo>> inputs = op_info_ptr->inputs_ptr();
std::vector<std::shared_ptr<OpIOInfo>> outputs = op_info_ptr->outputs_ptr(); std::vector<std::shared_ptr<OpIOInfo>> outputs = op_info_ptr->outputs_ptr();
std::vector<int> dyn_input_sizes; std::vector<int64_t> dyn_input_sizes;
auto primitive = AnfAlgo::GetCNodePrimitive(kernel_node); auto primitive = AnfAlgo::GetCNodePrimitive(kernel_node);
MS_EXCEPTION_IF_NULL(primitive); MS_EXCEPTION_IF_NULL(primitive);
if (primitive->GetAttr("dyn_input_sizes") != nullptr) { if (primitive->GetAttr("dyn_input_sizes") != nullptr) {
dyn_input_sizes = GetValue<std::vector<int>>(primitive->GetAttr("dyn_input_sizes")); dyn_input_sizes = GetValue<std::vector<int64_t>>(primitive->GetAttr("dyn_input_sizes"));
} }
if (inputs.size() > 0) { if (inputs.size() > 0) {
MS_EXCEPTION_IF_NULL(inputs[0]); MS_EXCEPTION_IF_NULL(inputs[0]);
@ -552,11 +552,11 @@ std::vector<std::pair<AnfNodePtr, std::pair<size_t, size_t>>> GetInputIndex(cons
continue; continue;
} }
std::vector<int> dyn_input_sizes; std::vector<int64_t> dyn_input_sizes;
auto prim = AnfAlgo::GetCNodePrimitive(anf_node); auto prim = AnfAlgo::GetCNodePrimitive(anf_node);
MS_EXCEPTION_IF_NULL(prim); MS_EXCEPTION_IF_NULL(prim);
if (prim->GetAttr(kAttrDynInputSizes) != nullptr) { if (prim->GetAttr(kAttrDynInputSizes) != nullptr) {
dyn_input_sizes = GetValue<const std::vector<int>>(prim->GetAttr(kAttrDynInputSizes)); dyn_input_sizes = GetValue<const std::vector<int64_t>>(prim->GetAttr(kAttrDynInputSizes));
} }
if (dyn_input_sizes.empty()) { if (dyn_input_sizes.empty()) {
@ -764,28 +764,26 @@ bool IsWeightBoundary(const AnfNodePtr &node) {
return false; return false;
} }
std::vector<int> GetReduceAttrAxis(const CNodePtr &cnode) { std::vector<int64_t> GetReduceAttrAxis(const CNodePtr &cnode) {
if (AnfAlgo::GetInputTensorNum(cnode) != AnfAlgo::GetOutputTensorNum(cnode) && if (AnfAlgo::GetInputTensorNum(cnode) != AnfAlgo::GetOutputTensorNum(cnode) &&
AnfAlgo::GetInputTensorNum(cnode) != 1) { AnfAlgo::GetInputTensorNum(cnode) != 1) {
MS_LOG(EXCEPTION) << "the kind of reduce node [" << cnode->DebugString() MS_LOG(EXCEPTION) << "the kind of reduce node [" << cnode->DebugString()
<< "] is not single input or single output "; << "] is not single input or single output ";
} }
std::vector<int> axis; std::vector<int64_t> axis;
auto input_shape = AnfAlgo::GetPrevNodeOutputInferShape(cnode, 0); auto input_shape = AnfAlgo::GetPrevNodeOutputInferShape(cnode, 0);
auto primitive = AnfAlgo::GetCNodePrimitive(cnode); auto primitive = AnfAlgo::GetCNodePrimitive(cnode);
MS_EXCEPTION_IF_NULL(primitive); MS_EXCEPTION_IF_NULL(primitive);
auto axis_attr = primitive->GetAttr(kAxis); auto axis_attr = primitive->GetAttr(kAxis);
if (axis_attr == nullptr) { if (axis_attr == nullptr) {
MS_LOG(ERROR) << "This node does't have axie attr."; MS_LOG(ERROR) << "This node does't have axie attr.";
return std::vector<int>(); return std::vector<int64_t>();
} }
auto type = axis_attr->type(); std::vector<int64_t> axis_list;
MS_EXCEPTION_IF_NULL(type); if (axis_attr->isa<Int64Imm>()) {
std::vector<int> axis_list; axis_list.emplace_back(GetValue<int64_t>(axis_attr));
if (type->ToString() == kTypeInt32) {
axis_list.emplace_back(GetValue<int>(axis_attr));
} else { } else {
axis_list = GetValue<std::vector<int>>(axis_attr); axis_list = GetValue<std::vector<int64_t>>(axis_attr);
} }
for (const auto &elem : axis_list) { for (const auto &elem : axis_list) {
if (elem < 0) { if (elem < 0) {

2
mindspore/ccsrc/backend/kernel_compiler/common_utils.h
Unescape View File

@ -100,7 +100,7 @@ void GetFuncGraphOutputNodes(const FuncGraphPtr &func_graph, std::vector<AnfNode
bool GetInputTensorValue(const AnfNodePtr &anf_node, size_t input_idx, nlohmann::json *const node_json); bool GetInputTensorValue(const AnfNodePtr &anf_node, size_t input_idx, nlohmann::json *const node_json);
void GetGraphRealOutput(const FuncGraphPtr &func_graph, std::vector<std::pair<AnfNodePtr, size_t>> *node_list); void GetGraphRealOutput(const FuncGraphPtr &func_graph, std::vector<std::pair<AnfNodePtr, size_t>> *node_list);
bool IsWeightBoundary(const AnfNodePtr &node); bool IsWeightBoundary(const AnfNodePtr &node);
std::vector<int> GetReduceAttrAxis(const CNodePtr &cnode); std::vector<int64_t> GetReduceAttrAxis(const CNodePtr &cnode);
std::string GetProcessorStr(const AnfNodePtr &anf_node); std::string GetProcessorStr(const AnfNodePtr &anf_node);
template <typename T> template <typename T>

2
mindspore/ccsrc/backend/kernel_compiler/cpu/argmax_cpu_kernel.cc
Unescape View File

@ -27,7 +27,7 @@ void ArgmaxCPUKernel::InitKernel(const CNodePtr &kernel_node) {
batch_size_ = shape[0]; batch_size_ = shape[0];
class_num_ = shape[1]; class_num_ = shape[1];
int axis = AnfAlgo::GetNodeAttr<int>(kernel_node, AXIS); int64_t axis = AnfAlgo::GetNodeAttr<int64_t>(kernel_node, AXIS);
if (axis != -1 && axis != 1) { if (axis != -1 && axis != 1) {
MS_LOG(EXCEPTION) << "argmax kernel not support axis " << axis; MS_LOG(EXCEPTION) << "argmax kernel not support axis " << axis;
} }

6
mindspore/ccsrc/backend/kernel_compiler/cpu/concat_cpu_kernel.cc
Unescape View File

@ -22,12 +22,12 @@ namespace kernel {
void ConcatCPUKernel::InitKernel(const CNodePtr &kernel_node) { void ConcatCPUKernel::InitKernel(const CNodePtr &kernel_node) {
CheckParam(kernel_node); CheckParam(kernel_node);
axis_ = AnfAlgo::GetNodeAttr<int>(kernel_node, AXIS); axis_ = AnfAlgo::GetNodeAttr<int64_t>(kernel_node, AXIS);
auto input_1_shape = AnfAlgo::GetPrevNodeOutputInferShape(kernel_node, 0); auto input_1_shape = AnfAlgo::GetPrevNodeOutputInferShape(kernel_node, 0);
if (axis_ < 0) { if (axis_ < 0) {
axis_ = axis_ + SizeToInt(input_1_shape.size()); axis_ = axis_ + SizeToLong(input_1_shape.size());
} }
axis_ += 4 - input_1_shape.size(); axis_ += 4 - SizeToLong(input_1_shape.size());
auto input_num = AnfAlgo::GetInputTensorNum(kernel_node); auto input_num = AnfAlgo::GetInputTensorNum(kernel_node);
for (size_t i = 0; i < input_num; i++) { for (size_t i = 0; i < input_num; i++) {

2
mindspore/ccsrc/backend/kernel_compiler/cpu/concat_cpu_kernel.h
Unescape View File

@ -36,7 +36,7 @@ class ConcatCPUKernel : public CPUKernel {
void CheckParam(const CNodePtr &kernel_node); void CheckParam(const CNodePtr &kernel_node);
void CopyDataToOutput(const std::vector<kernel::AddressPtr> &inputs, size_t dim0, size_t dim1, size_t dim2, void CopyDataToOutput(const std::vector<kernel::AddressPtr> &inputs, size_t dim0, size_t dim1, size_t dim2,
float **output_addr, size_t *buff_size); float **output_addr, size_t *buff_size);
int axis_; int64_t axis_;
std::vector<std::vector<size_t>> input_shape_list_; std::vector<std::vector<size_t>> input_shape_list_;
std::vector<size_t> output_shape_; std::vector<size_t> output_shape_;
}; };

8
mindspore/ccsrc/backend/kernel_compiler/cpu/embedding_look_up_comm_grad_cpu_kernel.cc
Unescape View File

@ -22,7 +22,7 @@ namespace mindspore {
namespace kernel { namespace kernel {
void EmbeddingLookUpCommGradCPUKernel::InitKernel(const CNodePtr &kernel_node) { void EmbeddingLookUpCommGradCPUKernel::InitKernel(const CNodePtr &kernel_node) {
CheckParam(kernel_node); CheckParam(kernel_node);
split_num_ = AnfAlgo::GetNodeAttr<int>(kernel_node, "split_num"); split_num_ = AnfAlgo::GetNodeAttr<int64_t>(kernel_node, "split_num");
MS_LOG(INFO) << "split_num: " << split_num_; MS_LOG(INFO) << "split_num: " << split_num_;
auto input_shape = AnfAlgo::GetPrevNodeOutputInferShape(kernel_node, 0); auto input_shape = AnfAlgo::GetPrevNodeOutputInferShape(kernel_node, 0);
if (input_shape[0] % split_num_ != 0) { if (input_shape[0] % split_num_ != 0) {
@ -47,9 +47,9 @@ bool EmbeddingLookUpCommGradCPUKernel::Launch(const std::vector<kernel::AddressP
MS_LOG(DEBUG) << "output addr: " << output_addr << "output size: " << output_size; MS_LOG(DEBUG) << "output addr: " << output_addr << "output size: " << output_size;
memset_s(output_addr, output_size, 0, output_size); memset_s(output_addr, output_size, 0, output_size);
const std::vector<int> &rank_group = {0, 1, 2, 3, 4, 5, 6, 7}; const std::vector<int> &rank_group = {0, 1, 2, 3, 4, 5, 6, 7};
size_t input_split_lens = input_size / split_num_ / sizeof(float_t); size_t input_split_lens = input_size / LongToSize(split_num_) / sizeof(float_t);
size_t output_split_lens = output_size / split_num_ / sizeof(float_t); size_t output_split_lens = output_size / LongToSize(split_num_) / sizeof(float_t);
for (int i = 0; i < split_num_; i++) { for (int64_t i = 0; i < split_num_; i++) {
MPIAllGather(input_addr + i * input_split_lens, output_addr + i * output_split_lens, rank_group, input_split_lens); MPIAllGather(input_addr + i * input_split_lens, output_addr + i * output_split_lens, rank_group, input_split_lens);
} }
#if defined(_WIN32) || defined(_WIN64) #if defined(_WIN32) || defined(_WIN64)

2
mindspore/ccsrc/backend/kernel_compiler/cpu/embedding_look_up_comm_grad_cpu_kernel.h
Unescape View File

@ -34,7 +34,7 @@ class EmbeddingLookUpCommGradCPUKernel : public CPUKernel {
private: private:
void CheckParam(const CNodePtr &kernel_node); void CheckParam(const CNodePtr &kernel_node);
int split_num_; int64_t split_num_;
}; };
MS_REG_CPU_KERNEL(EmbeddingLookupCommGrad, MS_REG_CPU_KERNEL(EmbeddingLookupCommGrad,

2
mindspore/ccsrc/backend/kernel_compiler/cpu/embedding_look_up_cpu_kernel.cc
Unescape View File

@ -61,7 +61,7 @@ void EmbeddingLookUpCPUKernel::InitKernel(const CNodePtr &kernel_node) {
indices_lens_ *= shape; indices_lens_ *= shape;
} }
if (AnfAlgo::HasNodeAttr(kAttrOffset, kernel_node)) { if (AnfAlgo::HasNodeAttr(kAttrOffset, kernel_node)) {
offset_ = AnfAlgo::GetNodeAttr<int>(kernel_node, kAttrOffset); offset_ = AnfAlgo::GetNodeAttr<int64_t>(kernel_node, kAttrOffset);
} }
indices_data_type_ = AnfAlgo::GetInputDeviceDataType(kernel_node, 1); indices_data_type_ = AnfAlgo::GetInputDeviceDataType(kernel_node, 1);
} }

2
mindspore/ccsrc/backend/kernel_compiler/cpu/embedding_look_up_cpu_kernel.h
Unescape View File

@ -37,7 +37,7 @@ class EmbeddingLookUpCPUKernel : public CPUKernel {
protected: protected:
void CheckParam(const CNodePtr &kernel_node); void CheckParam(const CNodePtr &kernel_node);
int offset_{0}; int64_t offset_{0};
size_t indices_lens_{1}; size_t indices_lens_{1};
size_t first_dim_size_{1}; size_t first_dim_size_{1};
size_t outer_dim_size_{1}; size_t outer_dim_size_{1};

6
mindspore/ccsrc/backend/kernel_compiler/cpu/gather_cpu_kernel.cc
Unescape View File

@ -23,9 +23,9 @@ void GatherV2CPUKernel::InitKernel(const CNodePtr &kernel_node) {
input_shape_ = AnfAlgo::GetPrevNodeOutputInferShape(kernel_node, 0); input_shape_ = AnfAlgo::GetPrevNodeOutputInferShape(kernel_node, 0);
indices_shape_ = AnfAlgo::GetPrevNodeOutputInferShape(kernel_node, 1); indices_shape_ = AnfAlgo::GetPrevNodeOutputInferShape(kernel_node, 1);
output_shape_ = AnfAlgo::GetOutputInferShape(kernel_node, 0); output_shape_ = AnfAlgo::GetOutputInferShape(kernel_node, 0);
axis_ = AnfAlgo::GetNodeAttr<int>(kernel_node, AXIS); axis_ = AnfAlgo::GetNodeAttr<int64_t>(kernel_node, AXIS);
if (axis_ < 0) { if (axis_ < 0) {
axis_ = axis_ + SizeToInt(input_shape_.size()); axis_ = axis_ + SizeToLong(input_shape_.size());
} }
axis_ += 4 - input_shape_.size(); axis_ += 4 - input_shape_.size();
CPUKernelUtils::ExpandDimsTo4(&input_shape_); CPUKernelUtils::ExpandDimsTo4(&input_shape_);
@ -75,7 +75,7 @@ void GatherV2CPUKernel::CopyDataToOutput(const std::vector<kernel::AddressPtr> &
MS_LOG(EXCEPTION) << "The indices value is less than 0."; MS_LOG(EXCEPTION) << "The indices value is less than 0.";
} }
size_t index = IntToSize(indices_addr[i]); size_t index = IntToSize(indices_addr[i]);
if (index >= input_shape_[IntToSize(axis_)]) { if (index >= input_shape_[LongToSize(axis_)]) {
auto ret = memset_s(*output_addr, *buff_size, 0., num * sizeof(float)); auto ret = memset_s(*output_addr, *buff_size, 0., num * sizeof(float));
if (ret != EOK) { if (ret != EOK) {
MS_LOG(EXCEPTION) << "memset failed."; MS_LOG(EXCEPTION) << "memset failed.";

2
mindspore/ccsrc/backend/kernel_compiler/cpu/gather_cpu_kernel.h
Unescape View File

@ -39,7 +39,7 @@ class GatherV2CPUKernel : public CPUKernel {
std::vector<size_t> input_shape_; std::vector<size_t> input_shape_;
std::vector<size_t> indices_shape_; std::vector<size_t> indices_shape_;
std::vector<size_t> output_shape_; std::vector<size_t> output_shape_;
int axis_; int64_t axis_;
}; };
MS_REG_CPU_KERNEL( MS_REG_CPU_KERNEL(

13
mindspore/ccsrc/backend/kernel_compiler/cpu/mkldnn/conv2d_cpu_kernel.cc
Unescape View File

@ -15,6 +15,7 @@
*/ */
#include "backend/kernel_compiler/cpu/mkldnn/conv2d_cpu_kernel.h" #include "backend/kernel_compiler/cpu/mkldnn/conv2d_cpu_kernel.h"
#include <string> #include <string>
#include <algorithm>
#include "utils/ms_utils.h" #include "utils/ms_utils.h"
#include "backend/kernel_compiler/cpu/mkldnn/mkl_kernel_engine.h" #include "backend/kernel_compiler/cpu/mkldnn/mkl_kernel_engine.h"
#include "runtime/device/cpu/cpu_device_address.h" #include "runtime/device/cpu/cpu_device_address.h"
@ -30,7 +31,7 @@ void Conv2dCPUKernel::InitKernel(const CNodePtr &kernel_node) {
MS_LOG(EXCEPTION) << "conv2d only support nchw input!"; MS_LOG(EXCEPTION) << "conv2d only support nchw input!";
} }
std::vector<size_t> kernel_size({weight_shape[2], weight_shape[3]}); std::vector<size_t> kernel_size({weight_shape[2], weight_shape[3]});
size_t group = IntToSize(AnfAlgo::GetNodeAttr<int>(kernel_node, GROUP)); size_t group = LongToSize(AnfAlgo::GetNodeAttr<int64_t>(kernel_node, GROUP));
if (group != 1) { if (group != 1) {
if (src_shape[1] % group != 0) { if (src_shape[1] % group != 0) {
MS_LOG(EXCEPTION) << "conv2d channels should be divided by group!"; MS_LOG(EXCEPTION) << "conv2d channels should be divided by group!";
@ -41,8 +42,14 @@ void Conv2dCPUKernel::InitKernel(const CNodePtr &kernel_node) {
dnnl::memory::desc src_desc = GetDefaultMemDesc(src_shape); dnnl::memory::desc src_desc = GetDefaultMemDesc(src_shape);
dnnl::memory::desc weights_desc = GetDefaultMemDesc(weight_shape); dnnl::memory::desc weights_desc = GetDefaultMemDesc(weight_shape);
dnnl::memory::desc dst_desc = GetDefaultMemDesc(dst_shape); dnnl::memory::desc dst_desc = GetDefaultMemDesc(dst_shape);
auto stride_ori = AnfAlgo::GetNodeAttr<std::vector<int>>(kernel_node, STRIDE); std::vector<int> stride_ori;
auto dilation_ori = AnfAlgo::GetNodeAttr<std::vector<int>>(kernel_node, DILATION); std::vector<int> dilation_ori;
auto stride_me = AnfAlgo::GetNodeAttr<std::vector<int64_t>>(kernel_node, STRIDE);
auto dilation_me = AnfAlgo::GetNodeAttr<std::vector<int64_t>>(kernel_node, DILATION);
(void)std::transform(stride_me.begin(), stride_me.end(), std::back_inserter(stride_ori),
[](const int64_t &value) { return static_cast<int>(value); });
(void)std::transform(dilation_me.begin(), dilation_me.end(), std::back_inserter(dilation_ori),
[](const int64_t &value) { return static_cast<int>(value); });
if (stride_ori.size() != 4 || stride_ori[2] != stride_ori[3]) { if (stride_ori.size() != 4 || stride_ori[2] != stride_ori[3]) {
MS_LOG(EXCEPTION) << "conv2d only support equal stride, and stride must be 4d!"; MS_LOG(EXCEPTION) << "conv2d only support equal stride, and stride must be 4d!";
} }

13
mindspore/ccsrc/backend/kernel_compiler/cpu/mkldnn/conv2d_grad_filter_cpu_kernel.cc
Unescape View File

@ -15,6 +15,7 @@
*/ */
#include "backend/kernel_compiler/cpu/mkldnn/conv2d_grad_filter_cpu_kernel.h" #include "backend/kernel_compiler/cpu/mkldnn/conv2d_grad_filter_cpu_kernel.h"
#include <string> #include <string>
#include <algorithm>
#include "utils/ms_utils.h" #include "utils/ms_utils.h"
#include "backend/kernel_compiler/cpu/mkldnn/mkl_kernel_engine.h" #include "backend/kernel_compiler/cpu/mkldnn/mkl_kernel_engine.h"
#include "runtime/device/cpu/cpu_device_address.h" #include "runtime/device/cpu/cpu_device_address.h"
@ -30,7 +31,7 @@ void Conv2dGradFilterCPUKernel::InitKernel(const CNodePtr &kernel_node) {
MS_LOG(EXCEPTION) << ("conv2d grad filter only support nchw input!"); MS_LOG(EXCEPTION) << ("conv2d grad filter only support nchw input!");
} }
std::vector<size_t> kernel_size({weight_shape[2], weight_shape[3]}); std::vector<size_t> kernel_size({weight_shape[2], weight_shape[3]});
size_t group = IntToSize(AnfAlgo::GetNodeAttr<int>(kernel_node, GROUP)); size_t group = LongToSize(AnfAlgo::GetNodeAttr<int64_t>(kernel_node, GROUP));
if (group != 1) { if (group != 1) {
if (src_shape[1] % group != 0) { if (src_shape[1] % group != 0) {
MS_LOG(EXCEPTION) << "conv2d channels should be divided by group!"; MS_LOG(EXCEPTION) << "conv2d channels should be divided by group!";
@ -41,8 +42,14 @@ void Conv2dGradFilterCPUKernel::InitKernel(const CNodePtr &kernel_node) {
dnnl::memory::desc src_desc = GetDefaultMemDesc(src_shape); dnnl::memory::desc src_desc = GetDefaultMemDesc(src_shape);
dnnl::memory::desc weights_desc = GetDefaultMemDesc(weight_shape); dnnl::memory::desc weights_desc = GetDefaultMemDesc(weight_shape);
dnnl::memory::desc dst_desc = GetDefaultMemDesc(dst_shape); dnnl::memory::desc dst_desc = GetDefaultMemDesc(dst_shape);
auto stride_ori = AnfAlgo::GetNodeAttr<std::vector<int>>(kernel_node, STRIDE); std::vector<int> stride_ori;
auto dilation_ori = AnfAlgo::GetNodeAttr<std::vector<int>>(kernel_node, DILATION); std::vector<int> dilation_ori;
auto stride_me = AnfAlgo::GetNodeAttr<std::vector<int64_t>>(kernel_node, STRIDE);
auto dilation_me = AnfAlgo::GetNodeAttr<std::vector<int64_t>>(kernel_node, DILATION);
(void)std::transform(stride_me.begin(), stride_me.end(), std::back_inserter(stride_ori),
[](const int64_t &value) { return static_cast<int>(value); });
(void)std::transform(dilation_me.begin(), dilation_me.end(), std::back_inserter(dilation_ori),
[](const int64_t &value) { return static_cast<int>(value); });
if (stride_ori.size() != 2 || stride_ori[0] != stride_ori[1]) { if (stride_ori.size() != 2 || stride_ori[0] != stride_ori[1]) {
MS_LOG(EXCEPTION) << "Conv2dGradFilterCPUKernel only support equal stride, and stride must be 2d!"; MS_LOG(EXCEPTION) << "Conv2dGradFilterCPUKernel only support equal stride, and stride must be 2d!";
} }

13
mindspore/ccsrc/backend/kernel_compiler/cpu/mkldnn/conv2d_grad_input_cpu_kernel.cc
Unescape View File

@ -15,6 +15,7 @@
*/ */
#include "backend/kernel_compiler/cpu/mkldnn/conv2d_grad_input_cpu_kernel.h" #include "backend/kernel_compiler/cpu/mkldnn/conv2d_grad_input_cpu_kernel.h"
#include <string> #include <string>
#include <algorithm>
#include "backend/kernel_compiler/cpu/mkldnn/mkl_kernel_engine.h" #include "backend/kernel_compiler/cpu/mkldnn/mkl_kernel_engine.h"
#include "runtime/device/cpu/cpu_device_address.h" #include "runtime/device/cpu/cpu_device_address.h"
#include "utils/ms_utils.h" #include "utils/ms_utils.h"
@ -30,7 +31,7 @@ void Conv2dGradInputCPUKernel::InitKernel(const CNodePtr &kernel_node) {
MS_LOG(EXCEPTION) << "conv2d grad filter only support nchw input!"; MS_LOG(EXCEPTION) << "conv2d grad filter only support nchw input!";
} }
std::vector<size_t> kernel_size({weight_shape[2], weight_shape[3]}); std::vector<size_t> kernel_size({weight_shape[2], weight_shape[3]});
size_t group = IntToSize(AnfAlgo::GetNodeAttr<int>(kernel_node, GROUP)); size_t group = LongToSize(AnfAlgo::GetNodeAttr<int64_t>(kernel_node, GROUP));
if (group != 1) { if (group != 1) {
if (src_shape[1] % group != 0) { if (src_shape[1] % group != 0) {
MS_LOG(EXCEPTION) << "conv2d channels should be divided by group!"; MS_LOG(EXCEPTION) << "conv2d channels should be divided by group!";
@ -42,8 +43,14 @@ void Conv2dGradInputCPUKernel::InitKernel(const CNodePtr &kernel_node) {
dnnl::memory::desc weights_desc = GetDefaultMemDesc(weight_shape); dnnl::memory::desc weights_desc = GetDefaultMemDesc(weight_shape);
dnnl::memory::desc dst_desc = GetDefaultMemDesc(dst_shape); dnnl::memory::desc dst_desc = GetDefaultMemDesc(dst_shape);
auto stride_ori = AnfAlgo::GetNodeAttr<std::vector<int>>(kernel_node, STRIDE); std::vector<int> stride_ori;
auto dilation_ori = AnfAlgo::GetNodeAttr<std::vector<int>>(kernel_node, DILATION); std::vector<int> dilation_ori;
auto stride_me = AnfAlgo::GetNodeAttr<std::vector<int64_t>>(kernel_node, STRIDE);
auto dilation_me = AnfAlgo::GetNodeAttr<std::vector<int64_t>>(kernel_node, DILATION);
(void)std::transform(stride_me.begin(), stride_me.end(), std::back_inserter(stride_ori),
[](const int64_t &value) { return static_cast<int>(value); });
(void)std::transform(dilation_me.begin(), dilation_me.end(), std::back_inserter(dilation_ori),
[](const int64_t &value) { return static_cast<int>(value); });
if (stride_ori.size() != 2 || stride_ori[0] != stride_ori[1]) { if (stride_ori.size() != 2 || stride_ori[0] != stride_ori[1]) {
MS_LOG(EXCEPTION) << "Conv2dGradInputCPUKernel only support equal stride, and stride must be 2d!"; MS_LOG(EXCEPTION) << "Conv2dGradInputCPUKernel only support equal stride, and stride must be 2d!";
} }

6
mindspore/ccsrc/backend/kernel_compiler/cpu/mkldnn/lstm_cpu_kernel.cc
Unescape View File

@ -76,9 +76,9 @@ void LstmCPUKernel::CheckParam(const CNodePtr &kernel_node) {
std::vector<size_t> src_h_shape = AnfAlgo::GetInputDeviceShape(kernel_node, 1); std::vector<size_t> src_h_shape = AnfAlgo::GetInputDeviceShape(kernel_node, 1);
std::vector<size_t> src_c_shape = AnfAlgo::GetInputDeviceShape(kernel_node, 2); std::vector<size_t> src_c_shape = AnfAlgo::GetInputDeviceShape(kernel_node, 2);
bidirectional_ = AnfAlgo::GetNodeAttr<bool>(kernel_node, "bidirectional"); bidirectional_ = AnfAlgo::GetNodeAttr<bool>(kernel_node, "bidirectional");
input_size_ = AnfAlgo::GetNodeAttr<int>(kernel_node, "input_size"); input_size_ = static_cast<int>(AnfAlgo::GetNodeAttr<int64_t>(kernel_node, "input_size"));
hidden_size_ = AnfAlgo::GetNodeAttr<int>(kernel_node, "hidden_size"); hidden_size_ = static_cast<int>(AnfAlgo::GetNodeAttr<int64_t>(kernel_node, "hidden_size"));
num_layers_ = AnfAlgo::GetNodeAttr<int>(kernel_node, "num_layers"); num_layers_ = static_cast<int>(AnfAlgo::GetNodeAttr<int64_t>(kernel_node, "num_layers"));
has_bias_ = AnfAlgo::GetNodeAttr<bool>(kernel_node, "has_bias"); has_bias_ = AnfAlgo::GetNodeAttr<bool>(kernel_node, "has_bias");
batch_size_ = SizeToInt(src_shape[1]); batch_size_ = SizeToInt(src_shape[1]);
seq_len_ = SizeToInt(src_shape[0]); seq_len_ = SizeToInt(src_shape[0]);

12
mindspore/ccsrc/backend/kernel_compiler/cpu/mkldnn/lstm_grad_cpu_kernel.cc
Unescape View File

@ -94,9 +94,9 @@ void LSTMGradCPUKernel::CheckParam(const CNodePtr &kernel_node) {
std::vector<size_t> src_h_shape = AnfAlgo::GetInputDeviceShape(kernel_node, 1); std::vector<size_t> src_h_shape = AnfAlgo::GetInputDeviceShape(kernel_node, 1);
std::vector<size_t> src_c_shape = AnfAlgo::GetInputDeviceShape(kernel_node, 2); std::vector<size_t> src_c_shape = AnfAlgo::GetInputDeviceShape(kernel_node, 2);
bidirectional_ = AnfAlgo::GetNodeAttr<bool>(kernel_node, "bidirectional"); bidirectional_ = AnfAlgo::GetNodeAttr<bool>(kernel_node, "bidirectional");
input_size_ = AnfAlgo::GetNodeAttr<int>(kernel_node, "input_size"); input_size_ = AnfAlgo::GetNodeAttr<int64_t>(kernel_node, "input_size");
hidden_size_ = AnfAlgo::GetNodeAttr<int>(kernel_node, "hidden_size"); hidden_size_ = AnfAlgo::GetNodeAttr<int64_t>(kernel_node, "hidden_size");
num_layers_ = AnfAlgo::GetNodeAttr<int>(kernel_node, "num_layers"); num_layers_ = AnfAlgo::GetNodeAttr<int64_t>(kernel_node, "num_layers");
has_bias_ = AnfAlgo::GetNodeAttr<bool>(kernel_node, "has_bias"); has_bias_ = AnfAlgo::GetNodeAttr<bool>(kernel_node, "has_bias");
batch_size_ = SizeToInt(src_shape[1]); batch_size_ = SizeToInt(src_shape[1]);
seq_len_ = SizeToInt(src_shape[0]); seq_len_ = SizeToInt(src_shape[0]);
@ -104,20 +104,20 @@ void LSTMGradCPUKernel::CheckParam(const CNodePtr &kernel_node) {
if (bidirectional_) { if (bidirectional_) {
num_directions_ = 2; num_directions_ = 2;
} }
const int gate_size = 4 * hidden_size_; const int64_t gate_size = 4 * hidden_size_;
if (num_layers_ <= 0) { if (num_layers_ <= 0) {
MS_LOG(EXCEPTION) << "layers must be greater than zero!"; MS_LOG(EXCEPTION) << "layers must be greater than zero!";
} }
if (num_layers_ > kMaxLSTMLayer) { if (num_layers_ > kMaxLSTMLayer) {
MS_LOG(EXCEPTION) << "layers must be lower than 100!"; MS_LOG(EXCEPTION) << "layers must be lower than 100!";
} }
for (int i = 0; i < num_layers_; ++i) { for (int64_t i = 0; i < num_layers_; ++i) {
weight_size_ += gate_size * (i == 0 ? input_size_ : hidden_size_ * num_directions_); weight_size_ += gate_size * (i == 0 ? input_size_ : hidden_size_ * num_directions_);
weight_h_size_ += gate_size * hidden_size_; weight_h_size_ += gate_size * hidden_size_;
} }
weight_size_ = weight_size_ * num_directions_; weight_size_ = weight_size_ * num_directions_;
weight_h_size_ = weight_h_size_ * num_directions_; weight_h_size_ = weight_h_size_ * num_directions_;
if (num_directions_ * num_layers_ != SizeToInt(src_h_shape[0])) { if (num_directions_ * num_layers_ != SizeToLong(src_h_shape[0])) {
MS_LOG(EXCEPTION) << "error iteration shape!"; MS_LOG(EXCEPTION) << "error iteration shape!";
} }
if (src_shape.size() != 3 || src_h_shape.size() != 3 || src_c_shape.size() != 3) { if (src_shape.size() != 3 || src_h_shape.size() != 3 || src_c_shape.size() != 3) {

14
mindspore/ccsrc/backend/kernel_compiler/cpu/mkldnn/lstm_grad_cpu_kernel.h
Unescape View File

@ -44,13 +44,13 @@ class LSTMGradCPUKernel : public MKLCPUKernel {
const dnnl::memory &diff_bias_memory); const dnnl::memory &diff_bias_memory);
void ResetMemory(const dnnl::memory &mem, string name); void ResetMemory(const dnnl::memory &mem, string name);
void CheckParam(const CNodePtr &kernel_node); void CheckParam(const CNodePtr &kernel_node);
int weight_size_ = 0; int64_t weight_size_ = 0;
int weight_h_size_ = 0; int64_t weight_h_size_ = 0;
int input_size_; int64_t input_size_;
int hidden_size_; int64_t hidden_size_;
int num_layers_; int64_t num_layers_;
int batch_size_; int64_t batch_size_;
int seq_len_; int64_t seq_len_;
int num_directions_; int num_directions_;
bool bidirectional_; bool bidirectional_;
bool has_bias_; bool has_bias_;

5
mindspore/ccsrc/backend/kernel_compiler/cpu/mkldnn/mkl_cpu_kernel.cc
Unescape View File

@ -56,7 +56,10 @@ void MKLCPUKernel::GetPadding(const CNodePtr &kernel_node, const std::string &pa
padding_r->emplace_back(0); padding_r->emplace_back(0);
padding_r->emplace_back(0); padding_r->emplace_back(0);
} else { } else {
std::vector<int> pad = AnfAlgo::GetNodeAttr<std::vector<int>>(kernel_node, PAD_LIST); std::vector<int> pad;
std::vector<int64_t> pad_me = AnfAlgo::GetNodeAttr<std::vector<int64_t>>(kernel_node, PAD_LIST);
(void)std::transform(pad_me.begin(), pad_me.end(), std::back_inserter(pad),
[](const int64_t &value) { return static_cast<int>(value); });
padding_l->emplace_back(pad[0]); padding_l->emplace_back(pad[0]);
padding_l->emplace_back(pad[1]); padding_l->emplace_back(pad[1]);
padding_r->emplace_back(pad[2]); padding_r->emplace_back(pad[2]);

10
mindspore/ccsrc/backend/kernel_compiler/cpu/mkldnn/pooling_cpu_kernel.cc
Unescape View File

@ -28,8 +28,14 @@ void PoolingCPUKernel::InitKernel(const CNodePtr &kernel_node) {
std::vector<size_t> dst_shape = AnfAlgo::GetOutputDeviceShape(kernel_node, 0); std::vector<size_t> dst_shape = AnfAlgo::GetOutputDeviceShape(kernel_node, 0);
dnnl::memory::desc src_desc = GetDefaultMemDesc(src_shape); dnnl::memory::desc src_desc = GetDefaultMemDesc(src_shape);
dnnl::memory::desc dst_desc = GetDefaultMemDesc(dst_shape); dnnl::memory::desc dst_desc = GetDefaultMemDesc(dst_shape);
std::vector<int> origin_kernel_sizes = AnfAlgo::GetNodeAttr<std::vector<int>>(kernel_node, KSIZE); std::vector<int> origin_kernel_sizes;
std::vector<int> strides = AnfAlgo::GetNodeAttr<std::vector<int>>(kernel_node, STRIDES); std::vector<int> strides;
std::vector<int64_t> kernel_sizes_me = AnfAlgo::GetNodeAttr<std::vector<int64_t>>(kernel_node, KSIZE);
std::vector<int64_t> strides_me = AnfAlgo::GetNodeAttr<std::vector<int64_t>>(kernel_node, STRIDES);
(void)std::transform(kernel_sizes_me.begin(), kernel_sizes_me.end(), std::back_inserter(origin_kernel_sizes),
[](const int64_t &value) { return static_cast<int>(value); });
(void)std::transform(strides_me.begin(), strides_me.end(), std::back_inserter(strides),
[](const int64_t &value) { return static_cast<int>(value); });
if (origin_kernel_sizes.size() != 4 || strides.size() != 4) { if (origin_kernel_sizes.size() != 4 || strides.size() != 4) {
MS_LOG(EXCEPTION) << "invalid kernel size " << origin_kernel_sizes.size() << " or stride size " << strides.size(); MS_LOG(EXCEPTION) << "invalid kernel size " << origin_kernel_sizes.size() << " or stride size " << strides.size();
} }

10
mindspore/ccsrc/backend/kernel_compiler/cpu/mkldnn/pooling_grad_cpu_kernel.cc
Unescape View File

@ -27,8 +27,14 @@ void PoolingGradCPUKernel::InitKernel(const CNodePtr &kernel_node) {
MS_EXCEPTION_IF_NULL(kernel_node); MS_EXCEPTION_IF_NULL(kernel_node);
src_shape_ = AnfAlgo::GetInputDeviceShape(kernel_node, 0); src_shape_ = AnfAlgo::GetInputDeviceShape(kernel_node, 0);
dst_shape_ = AnfAlgo::GetInputDeviceShape(kernel_node, 1); dst_shape_ = AnfAlgo::GetInputDeviceShape(kernel_node, 1);
std::vector<int> kernel_sizes = AnfAlgo::GetNodeAttr<std::vector<int>>(kernel_node, KSIZE); std::vector<int> kernel_sizes;
std::vector<int> strides = AnfAlgo::GetNodeAttr<std::vector<int>>(kernel_node, STRIDES); std::vector<int> strides;
auto kernel_sizes_me = AnfAlgo::GetNodeAttr<std::vector<int64_t>>(kernel_node, KSIZE);
auto strides_me = AnfAlgo::GetNodeAttr<std::vector<int64_t>>(kernel_node, STRIDES);
(void)std::transform(kernel_sizes_me.begin(), kernel_sizes_me.end(), std::back_inserter(kernel_sizes),
[](const int64_t &value) { return static_cast<int>(value); });
(void)std::transform(strides_me.begin(), strides_me.end(), std::back_inserter(strides),
[](const int64_t &value) { return static_cast<int>(value); });
if (kernel_sizes.size() != 4 || strides.size() != 4 || src_shape_.size() != 4 || dst_shape_.size() != 4) { if (kernel_sizes.size() != 4 || strides.size() != 4 || src_shape_.size() != 4 || dst_shape_.size() != 4) {
MS_LOG(EXCEPTION) << "pooling grad invalid input size"; MS_LOG(EXCEPTION) << "pooling grad invalid input size";
} }

6
mindspore/ccsrc/backend/kernel_compiler/cpu/mkldnn/softmax_cpu_kernel.cc
Unescape View File

@ -14,6 +14,7 @@
* limitations under the License. * limitations under the License.
*/ */
#include "backend/kernel_compiler/cpu/mkldnn/softmax_cpu_kernel.h" #include "backend/kernel_compiler/cpu/mkldnn/softmax_cpu_kernel.h"
#include <algorithm>
#include "backend/kernel_compiler/cpu/mkldnn/mkl_kernel_engine.h" #include "backend/kernel_compiler/cpu/mkldnn/mkl_kernel_engine.h"
#include "runtime/device/cpu/cpu_device_address.h" #include "runtime/device/cpu/cpu_device_address.h"
#include "utils/ms_utils.h" #include "utils/ms_utils.h"
@ -23,7 +24,10 @@ namespace kernel {
void SoftmaxCPUKernel::InitKernel(const CNodePtr &kernel_node) { void SoftmaxCPUKernel::InitKernel(const CNodePtr &kernel_node) {
MS_EXCEPTION_IF_NULL(kernel_node); MS_EXCEPTION_IF_NULL(kernel_node);
std::vector<size_t> src_shape = AnfAlgo::GetInputDeviceShape(kernel_node, 0); std::vector<size_t> src_shape = AnfAlgo::GetInputDeviceShape(kernel_node, 0);
std::vector<int> axis_list = AnfAlgo::GetNodeAttr<std::vector<int>>(kernel_node, AXIS); std::vector<int> axis_list;
std::vector<int64_t> axis_list_me = AnfAlgo::GetNodeAttr<std::vector<int64_t>>(kernel_node, AXIS);
(void)std::transform(axis_list_me.begin(), axis_list_me.end(), std::back_inserter(axis_list),
[](const int64_t &value) { return static_cast<int>(value); });
if (axis_list.size() != 1) { if (axis_list.size() != 1) {
MS_LOG(EXCEPTION) << "cpu softmax only support input axis size 1"; MS_LOG(EXCEPTION) << "cpu softmax only support input axis size 1";
} }

6
mindspore/ccsrc/backend/kernel_compiler/cpu/one_hot_cpu_kernel.cc
Unescape View File

@ -24,14 +24,14 @@ void OneHotCPUKernel::InitKernel(const CNodePtr &kernel_node) {
if (output_shape.size() < 2) { if (output_shape.size() < 2) {
MS_LOG(EXCEPTION) << "invalid output shape size: " << output_shape.size(); MS_LOG(EXCEPTION) << "invalid output shape size: " << output_shape.size();
} }
int axis = AnfAlgo::GetNodeAttr<int>(kernel_node, AXIS); int64_t axis = AnfAlgo::GetNodeAttr<int64_t>(kernel_node, AXIS);
if (axis != -1 && IntToSize(axis) >= output_shape.size()) { if (axis != -1 && LongToSize(axis) >= output_shape.size()) {
MS_LOG(EXCEPTION) << "invalid axis: " << axis; MS_LOG(EXCEPTION) << "invalid axis: " << axis;
} }
if (axis == -1) { if (axis == -1) {
axis_ = output_shape.size() - 1; axis_ = output_shape.size() - 1;
} else { } else {
axis_ = IntToSize(axis); axis_ = LongToSize(axis);
} }
depth_ = output_shape[axis_]; depth_ = output_shape[axis_];
stride_ = 1; stride_ = 1;

Some files were not shown because too many files have changed in this diff Show More

Write Preview
Loading…
Cancel
Save