Skip to content

Commit 9cbfe51

Browse files
author
Zonglin Peng
committed
jarvis-nightly-operators-test-aten-flip-out
Differential Revision: [D85364548](https://our.internmc.facebook.com/intern/diff/D85364548/) [ghstack-poisoned]
1 parent 30be5ff commit 9cbfe51

File tree

1 file changed

+6
-0
lines changed

1 file changed

+6
-0
lines changed

backends/cadence/utils/facto_util.py

Lines changed: 6 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -440,6 +440,12 @@ def random_size_constraint(deps: object, r: int, d: int) -> int:
440440
cp.Size.Le(lambda deps, r, d: 2**2),
441441
]
442442
)
443+
case "flip.default":
444+
tensor_constraints.extend(
445+
[
446+
cp.Dtype.In(lambda deps: [torch.float32]),
447+
]
448+
)
443449
case _:
444450
pass
445451
return tensor_constraints

0 commit comments

Comments
 (0)