/**

 * Copyright (c) 2025 Huawei Technologies Co., Ltd.

 * This program is free software, you can redistribute it and/or modify it under the terms and conditions of

 * CANN Open Software License Agreement Version 2.0 (the "License").

 * Please refer to the License for details. You may not use this file except in compliance with the License.

 * THIS SOFTWARE IS PROVIDED ON AN "AS IS" BASIS, WITHOUT WARRANTIES OF ANY KIND, EITHER EXPRESS OR IMPLIED,

 * INCLUDING BUT NOT LIMITED TO NON-INFRINGEMENT, MERCHANTABILITY, OR FITNESS FOR A PARTICULAR PURPOSE.

 * See LICENSE in the root of the software repository for the full text of the License.

 */



/*!

 * \file mul_infershape.cpp

 * \brief

 */



#include "log/log.h"

#include "infershape_broadcast_util.h"

#include "register/op_impl_registry.h"



using namespace ge;

namespace ops {

static std::string ShapeCannotBroadcastMsg(const gert::Shape& shape1, const gert::Shape& shape2) {

  std::string res = "shape ";

  res += Ops::Base::ToString(shape1);

  res += " and ";

  res += Ops::Base::ToString(shape2);

  res += " cannot broadcast!";

  return res;

}



static ge::graphStatus InferShape4Broadcast(gert::InferShapeContext* context) {

  auto in_shape1 = context->GetInputShape(0);

  OP_CHECK_NULL_WITH_CONTEXT(context, in_shape1);

  auto in_shape2 = context->GetInputShape(1);

  OP_CHECK_NULL_WITH_CONTEXT(context, in_shape2);

  auto out_shape = context->GetOutputShape(0);

  OP_CHECK_NULL_WITH_CONTEXT(context, out_shape);



  OP_CHECK_IF((!Ops::Base::BroadcastShape(in_shape1, in_shape2, out_shape)),

           OP_LOGE(context->GetNodeName(), "%s", ShapeCannotBroadcastMsg(*in_shape2, *in_shape1).c_str()),

           return ge::GRAPH_FAILED);



  return ge::GRAPH_SUCCESS;

}



IMPL_OP_INFERSHAPE(Mul).InferShape(InferShape4Broadcast);



}