Skip to content

Commit 1b67b31

Browse files
Calibrate clamp.Tensor
ghstack-source-id: 556a7a0 Pull-Request: #32
1 parent 1ec292f commit 1b67b31

File tree

1 file changed

+8
-0
lines changed

1 file changed

+8
-0
lines changed

facto/specdb/db.py

Lines changed: 8 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -1120,6 +1120,14 @@
11201120
deps=[0, 1],
11211121
constraints=[
11221122
cp.Optional.Eq(lambda deps: False if deps[1] is None else True),
1123+
cp.Dtype.Ne(
1124+
lambda deps: (
1125+
torch.bool
1126+
if deps[0].dtype == torch.bool
1127+
and (deps[1] is None or deps[1].dtype == torch.bool)
1128+
else None
1129+
)
1130+
),
11231131
cp.Size.In(
11241132
lambda deps, r, d: fn.broadcast_with(
11251133
(

0 commit comments

Comments
 (0)