forked from mindspore-Ecosystem/mindspore
fix the codex and bot warnings
This commit is contained in:
parent
de4aca2025
commit
ec043fcd56
|
@ -26,7 +26,6 @@
|
|||
|
||||
namespace mindspore {
|
||||
namespace parallel {
|
||||
|
||||
using RankList = std::vector<int32_t>;
|
||||
using Shape = std::vector<int32_t>;
|
||||
|
||||
|
|
|
@ -1620,7 +1620,6 @@ CNodePtr FindLossCNode(const FuncGraphPtr &func_graph) {
|
|||
auto pre_cnode = pre_node->cast<CNodePtr>();
|
||||
MS_EXCEPTION_IF_NULL(pre_cnode);
|
||||
auto current_prim = GetValueNode<PrimitivePtr>(pre_cnode->input(0));
|
||||
|
||||
// return -> cast
|
||||
if (current_prim->name() == CAST && pre_cnode->operator_info() == nullptr) {
|
||||
pre_cnode = pre_cnode->input(1)->cast<CNodePtr>();
|
||||
|
|
|
@ -39,7 +39,8 @@ using OperatorList = std::vector<OperatorC>;
|
|||
class RedistributionOperatorInfer {
|
||||
public:
|
||||
const int NONE = -1;
|
||||
explicit RedistributionOperatorInfer(bool construct_op_flag = true) : construct_op_flag_(construct_op_flag) {}
|
||||
explicit RedistributionOperatorInfer(bool construct_op_flag = true)
|
||||
: construct_op_flag_(construct_op_flag), is_cost_model_(false) {}
|
||||
Status Init(const TensorLayout &tensor_layout, const Map &out_tensor_map, RankList dev_list,
|
||||
bool is_cost_model = false);
|
||||
~RedistributionOperatorInfer() = default;
|
||||
|
|
Loading…
Reference in New Issue