We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent a93e716 commit b132c96Copy full SHA for b132c96
kernels/prim_ops/register_prim_ops.cpp
@@ -90,7 +90,8 @@ static Kernel prim_ops[] = {
90
EValue& self = *stack[0];
91
EValue& out = *stack[1];
92
exec_aten::Tensor self_tensor = self.to<exec_aten::Tensor>();
93
- ET_SWITCH_REAL_TYPES(
+ ET_SWITCH_REAL_TYPES_AND(
94
+ Bool,
95
self_tensor.scalar_type(),
96
context,
97
"_local_scalar_dense",
0 commit comments