Function binaryOpParamSameShapeCheck

kernels/binary_op/binary_op_host.cpp:236–264

}

mluOpStatus_t binaryOpParamSameShapeCheck(
    const std::string &op_name, const mluOpTensorDescriptor_t input1_desc,
    const mluOpTensorDescriptor_t input2_desc,
    const mluOpTensorDescriptor_t output_desc) {
  // check dim
  PARAM_CHECK_EQ("[" + op_name + "]", input1_desc->getDim(),
                 input2_desc->getDim());
  PARAM_CHECK_EQ("[" + op_name + "]", input1_desc->getDim(),
                 output_desc->getDim());

  // check shape
  for (int i = 0; i < input1_desc->getDim(); i++) {
    if (input1_desc->getDimIndex(i) != input2_desc->getDimIndex(i)) {
      LOG(ERROR) << op_name << ":The shape of input1 should be equal to input2"
                 << ". But now input1's shape[" << i << "] is "
                 << input1_desc->getDimIndex(i) << ", input2's shape[" << i
                 << "] is " << input2_desc->getDimIndex(i) << ".";
      return MLUOP_STATUS_BAD_PARAM;
    }
    if (input1_desc->getDimIndex(i) != output_desc->getDimIndex(i)) {
      LOG(ERROR) << op_name << ":The shape of input1 should be equal to output"
                 << ". But now input1's shape[" << i << "] is "
                 << input1_desc->getDimIndex(i) << ", output's shape[" << i
                 << "] is " << output_desc->getDimIndex(i) << ".";
      return MLUOP_STATUS_BAD_PARAM;
    }
  }
  return MLUOP_STATUS_SUCCESS;
}

std::string array2String(int32_t dim_num, const int64_t *dims) {
  std::string res;