Guard功能介绍

当前Guard主要基于下面数据结构实现，如下图所示：

主要包括下面几个模块：

• 用于分支校验的ExpectGuard宏：

  1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35

  /* 校验表达式e0与e1是否相等， 如果e0的hint值等于e1的hint值，则宏返回true，并且生成e0 == e1的Guard， 反之，则返回false，并生成e0 != e1的Guard */ #define EXPECT_SYMBOL_EQ(e0, e1) /* 校验表达式e0与e1是否不相等， 如果e0的hint值不等于e1的hint值，则宏返回true，并且生成e0 != e1的Guard， 反之，则返回false，并生成e0 == e1的Guard */ #define EXPECT_SYMBOL_NE(e0, e1) /* 校验表达式e0是否小于e1， 如果e0的hint值小于e1的hint值，则宏返回true，并且生成e0 < e1的Guard， 反之，则返回false，并生成e1 <= e0的Guard */ #define EXPECT_SYMBOL_LT(e0, e1) /* 校验表达式e0是否小于等于e1， 如果e0的hint值小于等于e1的hint值，则宏返回true，并且生成e0 <= e1的Guard， 反之，则返回false，并生成e1 < e0的Guard */ #define EXPECT_SYMBOL_LE(e0, e1) /* 校验表达式e0是否大于e1， 如果e0的hint值大于e1的hint值，则宏返回true，并且生成e1 < e0的Guard， 反之，则返回false，并生成e0 <= e1的Guard */ #define EXPECT_SYMBOL_GT(e0, e1) /* 校验表达式e0是否大于等于e1， 如果e0的hint值大于等于e1的hint值，则宏返回true，并且生成e0 >= e1的Guard， 反之，则返回false，并生成e0 < e1的Guard */ #define EXPECT_SYMBOL_GE(e0, e1)

• 用于强校验的AssertGuard宏：

  1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35

  /* 强校验表达式e0与e1是否相等， 如果e0的hint值等于e1的hint值，则生成e0 == e1的Guard，并继续往下执行 反之，则报错并终止流程 */ #define ASSERT_SYMBOL_EQ(e0, e1) /* 强校验表达式e0与e1是否不相等， 如果e0的hint值不等于e1的hint值，则生成e0 != e1的Guard，并继续往下执行 反之，则报错并终止流程 */ #define ASSERT_SYMBOL_NE(e0, e1) /* 强校验表达式e0是否小于e1， 如果e0的hint值小于e1的hint值，则生成e0 < e1的Guard，并继续往下执行 反之，则报错并终止流程 */ #define ASSERT_SYMBOL_LT(e0, e1) /* 强校验表达式e0是否小于等于e1， 如果e0的hint值小于等于e1的hint值，则生成e0 <= e1的Guard，并继续往下执行 反之，则报错并终止流程 */ #define ASSERT_SYMBOL_LE(e0, e1) /* 强校验表达式e0是否大于e1， 如果e0的hint值大于e1的hint值，则生成e1 < e0的Guard，并继续往下执行 反之，则报错并终止流程 */ #define ASSERT_SYMBOL_GT(e0, e1) /* 强校验表达式e0是否大于等于e1， 如果e0的hint值大于等于e1的hint值，则生成e1 <= e0的Guard，并继续往下执行 反之，则报错并终止流程 */ #define ASSERT_SYMBOL_GE(e0, e1)

 1
 2
 3
 4
 5
 6
 7
 8
 9
10
11
12
13
14
15
16
17
18
19

class SymbolicUtils {
  // 基于之前生成的Guard信息判断e1与e2是否相等，仅基于已有Guard做校验，不生成新的Guard，主要用于内存优化等编译态优化时判断使用
  static bool StaticCheckEq(const Expression &e1, const Expression &e2);

  // 基于之前生成的Guard信息判断e1与e2是否不相等，仅基于已有Guard做校验，不生成新的Guard，主要用于内存优化等编译态优化时判断使用
  static bool StaticCheckNe(const Expression &e1, const Expression &e2);

  // 基于之前生成的Guard信息判断e1是否小于e2，仅基于已有Guard做校验，不生成新的Guard，主要用于内存优化等编译态优化时判断使用
  static bool StaticCheckLt(const Expression &e1, const Expression &e2);

  // 基于之前生成的Guard信息判断e1是否小于等于e2，仅基于已有Guard做校验，不生成新的Guard，主要用于内存优化等编译态优化时判断使用
  static bool StaticCheckLe(const Expression &e1, const Expression &e2);

  // 基于之前生成的Guard信息判断e1是否大于e2，仅基于已有Guard做校验，不生成新的Guard，主要用于内存优化等编译态优化时判断使用
  static bool StaticCheckGt(const Expression &e1, const Expression &e2);

  // 基于之前生成的Guard信息判断e1是否大于等于e2，仅基于已有Guard做校验，不生成新的Guard，主要用于内存优化等编译态优化时判断使用
  static bool StaticCheckGe(const Expression &e1, const Expression &e2);
}

 1
 2
 3
 4
 5
 6
 7
 8
 9
10
11
12
13
14
15
16
17
18
19
20

class ShapeEnvGuarder {
 public:
  explicit ShapeEnvGuarder(ShapeEnvAttr *shape_env_context);
  ~ShapeEnvGuarder();
  ShapeEnvGuarder(const ShapeEnvGuarder &) = delete;
  ShapeEnvGuarder(const ShapeEnvGuarder &&) = delete;
  ShapeEnvGuarder &operator=(const ShapeEnvGuarder &) = delete;
  ShapeEnvGuarder &&operator=(const ShapeEnvGuarder &&) = delete;
 private:
  ShapeEnvAttr *origin_context_; // ShapeEnvContext
};

void test(ComputeGraphPtr &graph) {
  auto root_graph = ge::GraphUtils::FindRootGraph(graph);
  GE_ASSERT_NOTNULL(root_graph);
  // 设置context
  ShapeEnvGuarder guarder(root_graph->GetAttrsGroup<ShapeEnvAttr>());
  // 需要生成Guard的逻辑
  xxxx
}