Skip to content

Commit b132c96

Browse files
authored
Enable bool type for _local_scalar_dense (#6994)
1 parent a93e716 commit b132c96

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed

kernels/prim_ops/register_prim_ops.cpp

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -90,7 +90,8 @@ static Kernel prim_ops[] = {
9090
EValue& self = *stack[0];
9191
EValue& out = *stack[1];
9292
exec_aten::Tensor self_tensor = self.to<exec_aten::Tensor>();
93-
ET_SWITCH_REAL_TYPES(
93+
ET_SWITCH_REAL_TYPES_AND(
94+
Bool,
9495
self_tensor.scalar_type(),
9596
context,
9697
"_local_scalar_dense",

0 commit comments

Comments
 (0)