diff --git a/kernels/prim_ops/register_prim_ops.cpp b/kernels/prim_ops/register_prim_ops.cpp index 5486149bce0..1d1bdd07769 100644 --- a/kernels/prim_ops/register_prim_ops.cpp +++ b/kernels/prim_ops/register_prim_ops.cpp @@ -90,7 +90,8 @@ static Kernel prim_ops[] = { EValue& self = *stack[0]; EValue& out = *stack[1]; exec_aten::Tensor self_tensor = self.to(); - ET_SWITCH_REAL_TYPES( + ET_SWITCH_REAL_TYPES_AND( + Bool, self_tensor.scalar_type(), context, "_local_scalar_dense",