File tree 1 file changed +4
-2
lines changed
1 file changed +4
-2
lines changed Original file line number Diff line number Diff line change @@ -418,6 +418,7 @@ def extend_bindings_with_shape_inference(knl: lp.LoopKernel,
418
418
from loopy .kernel .array import ArrayBase
419
419
from loopy .symbolic import get_dependencies as lpy_get_deps
420
420
from pymbolic .mapper .substitutor import make_subst_func
421
+ from pymbolic .primitives import is_valid_operand
421
422
422
423
from pytato .transform import SizeParamGatherer
423
424
@@ -426,8 +427,9 @@ def extend_bindings_with_shape_inference(knl: lp.LoopKernel,
426
427
lp_size_params : frozenset [str ] = reduce (frozenset .union ,
427
428
(lpy_get_deps (not_none (arg .shape ))
428
429
for arg in knl .args
429
- if isinstance (arg , ArrayBase )),
430
- frozenset ())
430
+ if isinstance (arg , ArrayBase )
431
+ and is_valid_operand (arg .shape )
432
+ ), frozenset ())
431
433
432
434
pt_size_params : frozenset [SizeParam ] = reduce (frozenset .union ,
433
435
(get_size_param_deps (bnd )
You can’t perform that action at this time.
0 commit comments