diff --git a/backends/cadence/utils/facto_util.py b/backends/cadence/utils/facto_util.py index e49cf412c19..c93ef345ff3 100644 --- a/backends/cadence/utils/facto_util.py +++ b/backends/cadence/utils/facto_util.py @@ -373,6 +373,9 @@ def random_size_constraint(deps: object, r: int, d: int) -> int: cp.Dtype.In(lambda deps: [torch.int64, torch.int32, torch.float32]), cp.Value.Ge(lambda deps, dtype, struct: -(2**4)), cp.Value.Le(lambda deps, dtype, struct: 2**4), + cp.Value.Ne( + lambda deps, dtype, struct: 0 + ), # Prevent division by zero cp.Rank.Ge(lambda deps: 1), cp.Rank.Eq(lambda deps: deps[0].dim()), cp.Size.Eq(lambda deps, r, d: fn.safe_size(deps[0], d)),