jarvis-nightly-operators-test-aten-flip-out

Zonglin Peng · Zonglin Peng · commit 564ac93bbd6f · 2025-11-01T07:44:52.000-07:00
Pull Request resolved: #15498 ghstack-source-id: 320280755 @exported-using-ghexport Differential Revision: [D85364548](https://our.internmc.facebook.com/intern/diff/D85364548/)
diff --git a/backends/cadence/utils/facto_util.py b/backends/cadence/utils/facto_util.py
@@ -444,6 +444,12 @@ def random_size_constraint(deps: object, r: int, d: int) -> int:
                     cp.Size.Le(lambda deps, r, d: 2**2),
                 ]
             )
+        case "flip.default":
+            tensor_constraints.extend(
+                [
+                    cp.Dtype.In(lambda deps: [torch.float32]),
+                ]
+            )
         case _:
             pass
     return tensor_constraints

Original file line number	Diff line number	Diff line change
`@@ -444,6 +444,12 @@ def random_size_constraint(deps: object, r: int, d: int) -> int:`
`444`	`444`	`cp.Size.Le(lambda deps, r, d: 2**2),`
`445`	`445`	`]`
`446`	`446`	`)`
	`447`	`+ case "flip.default":`
	`448`	`+ tensor_constraints.extend(`
	`449`	`+ [`
	`450`	`+ cp.Dtype.In(lambda deps: [torch.float32]),`
	`451`	`+ ]`
	`452`	`+ )`
`447`	`453`	`case _:`
`448`	`454`	`pass`
`449`	`455`	`return tensor_constraints`