Skip to content
Merged
Changes from 1 commit
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
3 changes: 3 additions & 0 deletions backends/cadence/utils/facto_util.py
Original file line number Diff line number Diff line change
Expand Up @@ -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)),
Expand Down
Loading