Skip to content

Commit a36597c

Browse files
zonglinpengfacebook-github-bot
authored andcommitted
add mean to g3 nightly
Summary: titled expect loss in some cases Differential Revision: D68845592
1 parent d9c462c commit a36597c

File tree

1 file changed

+20
-0
lines changed

1 file changed

+20
-0
lines changed

examples/cadence/operators/facto_util.py

Lines changed: 20 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -11,6 +11,7 @@
1111
from facto.inputgen.utils.random_manager import random_manager
1212
from facto.inputgen.variable.type import ScalarDtype
1313
from facto.specdb.db import SpecDictDB
14+
import facto.specdb.function as fn
1415

1516
# seed to generate identical cases every run to reproduce from bisect
1617
random_manager.seed(1729)
@@ -86,8 +87,27 @@ def facto_testcase_gen(op_name: str) -> List[Tuple[List[str], OrderedDict[str, s
8687
cp.Value.Le(lambda deps, dtype: 2),
8788
]
8889
)
90+
elif in_spec.type.is_scalar_type():
91+
spec.inspec[index].constraints.extend(
92+
[
93+
cp.Dtype.In(lambda deps: apply_scalar_contraints(op_name)),
94+
]
95+
)
8996
elif in_spec.type.is_tensor():
9097
spec.inspec[index].constraints.extend(tensor_constraints)
98+
elif in_spec.type.is_dim_list():
99+
spec.inspec[index].constraints.extend(
100+
[
101+
cp.Length.Ge(lambda deps: 1),
102+
cp.Optional.Eq(lambda deps: False),
103+
]
104+
)
105+
elif in_spec.type.is_bool():
106+
spec.inspec[index].constraints.extend(
107+
[
108+
cp.Dtype.In(lambda deps: [torch.bool]),
109+
]
110+
)
91111

92112
return [
93113
(posargs, inkwargs)

0 commit comments

Comments
 (0)