提交 51aaebff 编写于 作者: M mindspore-ci-bot 提交者: Gitee

!84 fix scalar rearrange bug

Merge pull request !84 from yangsijia/fix-scalar-rearrange-bug
......@@ -1373,10 +1373,11 @@ class ScalarPeel : public IRMutator {
public:
Stmt Run(const Stmt &s) {
Stmt res = Mutate(s);
if (!before_scalar_store_) {
if (!before_scalar_store_ || !find_multi_core_) {
multi_core_body_ = s;
return Stmt();
}
CHECK(multi_core_body_.defined()) << "Error, multicore body is not defined.";
return res;
}
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册