File tree 1 file changed +3
-4
lines changed
paddle/cinn/ir/group_schedule/config
1 file changed +3
-4
lines changed Original file line number Diff line number Diff line change @@ -184,10 +184,10 @@ bool CheckBroadcastTensorIsContinuous(
184
184
if (!iter_var2value.count (iter_var)) {
185
185
return false ;
186
186
}
187
+
187
188
ir::Expr iter_value = iter_var2value.at (iter_var);
188
189
189
- if (!iter_value.as_var () && !iter_value.is_constant ()) return false ;
190
-
190
+ if (!iter_value.as_var ()) return false ;
191
191
for (; loop_idx < for_iters.size (); ++loop_idx) {
192
192
if (for_iters[loop_idx] == iter_value.as_var_ref ()) {
193
193
break ;
@@ -212,14 +212,13 @@ bool CheckTensorIsContinuous(
212
212
for (int i = 0 ; i < indices.size (); ++i) {
213
213
ir::Expr index = indices[i];
214
214
index = optim::ArithSimplify (index );
215
- if (index .is_constant ()) return false ;
216
215
if (!index .is_var ()) return false ;
217
216
ir::Var iter_var = index .as_var_ref ();
218
217
if (!iter_var2value.count (iter_var)) {
219
218
return false ;
220
219
}
221
220
ir::Expr iter_value = iter_var2value.at (iter_var);
222
- if (!iter_value.as_var () && !iter_value. is_constant () ) return false ;
221
+ if (!iter_value.as_var ()) return false ;
223
222
if (for_iters[i] != iter_value.as_var_ref ()) {
224
223
return false ;
225
224
}
You can’t perform that action at this time.
0 commit comments