leftyer/akg
1
0
Fork 0
mirror of https://gitee.com/mindspore/akg.git synced 2025-12-06 11:59:12 +08:00
Code Issues Packages Projects Releases Wiki Activity

!1003 nullptr check

Browse Source 
Merge pull request !1003 from looop5/master
This commit is contained in:
anyrenwei
2025-07-09 02:31:22 +00:00
committed by Gitee
parent fce70236ec fa71b52e71
commit 854ad047cb
2 changed files with 13 additions and 9 deletions
Download Patch File Download Diff File Expand all files Collapse all files

20
src/composite/utils/dimension_peeling.cc
View File

@@ -420,10 +420,12 @@ void DimensionPeeler::MapDimToSpace(AffinityAnalyzer &aff, Dim *dom_dim, int axi
if (visited.count(prod)) return false;
if (prod != nullptr) visited.insert(prod);
if (!this->Propagation(axis_idx, dim, prod, affinity)) return false;
for (auto &c : prod->cons) {
if (!visited.count(c.first)) {
cons_visit.emplace_back(prod);
break;
if (prod != nullptr) {
for (auto &c : prod->cons) {
if (!visited.count(c.first)) {
cons_visit.emplace_back(prod);
break;
}
}
}
return true;
@@ -432,10 +434,12 @@ void DimensionPeeler::MapDimToSpace(AffinityAnalyzer &aff, Dim *dom_dim, int axi
if (visited.count(cons)) return false;
if (cons != nullptr) visited.insert(cons);
if (!this->Propagation(axis_idx, dim, cons, affinity)) return false;
for (auto &p : cons->prod) {
if (!visited.count(p.first)) {
prod_visit.emplace_back(cons);
break;
if (cons != nullptr) {
for (auto &p : cons->prod) {
if (!visited.count(p.first)) {
prod_visit.emplace_back(cons);
break;
}
}
}
return true;

2
src/pass/reduction_factor.cc
View File

@@ -699,7 +699,7 @@ class ReduceVectorizeEnable : public IRMutator {
}
// step 2: final reduce area
if (!cur_reduce_data_->isolate_reduce_provide.empty()) {
if (!cur_reduce_data_->isolate_reduce_provide.empty() && isolate_reduce_provide != nullptr) {
isolate_value = isolate_reduce_provide->value;
}
auto value = reduce_provide->value;