Skip to content

Commit bd1c76f

Browse files
schwarzmxfacebook-github-bot
authored andcommitted
Fix broken test
Summary: This test seems to be broken for at least a month and is blocking S343974 Specifically, the fbpkg builds are blocked by this: https://www.internalfb.com/intern/sandcastle/job/27021598739017954 Reviewed By: gunchu Differential Revision: D46848358 fbshipit-source-id: 4109c6bf909594dc42543a986e4c55398fcb2f07
1 parent f2be07b commit bd1c76f

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

multipy/runtime/example/examples.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -156,7 +156,7 @@ def forward(
156156
self, t: Tuple[Tensor, Tensor]
157157
) -> Tuple[Tuple[Tensor, Tensor], Tuple[Tensor, Tensor]]:
158158
a, b = t
159-
result = ((a.masked_fill_(b, 0.1), b), (torch.ones_like(a), b))
159+
result = ((a.masked_fill_(b.to(torch.bool), 0.1), b), (torch.ones_like(a), b))
160160
return result
161161

162162

0 commit comments

Comments
 (0)