From f29cd7853ff7a9f667baaf2e63c02875d02f0a09 Mon Sep 17 00:00:00 2001 From: Lucy Qiu Date: Wed, 5 Nov 2025 16:38:12 -0800 Subject: [PATCH] Expect at least one value in the tensor, to extract scalar from Summary: local scalar dense extracts a scalar from a tensor. Check that the tensor has at least one value Differential Revision: D86363818 --- kernels/prim_ops/register_prim_ops.cpp | 6 ++++++ 1 file changed, 6 insertions(+) diff --git a/kernels/prim_ops/register_prim_ops.cpp b/kernels/prim_ops/register_prim_ops.cpp index dc6ed9ac26f..7ff2f1f868d 100644 --- a/kernels/prim_ops/register_prim_ops.cpp +++ b/kernels/prim_ops/register_prim_ops.cpp @@ -141,6 +141,12 @@ static Kernel prim_ops[] = { EValue& out = *stack[1]; executorch::aten::Tensor self_tensor = self.to(); + ET_KERNEL_CHECK_MSG( + context, + self_tensor.numel() >= 1, + InvalidArgument, + /* void */, + "Expected tensor with at least 1 element"); ET_SWITCH_REAL_TYPES_AND( Bool, self_tensor.scalar_type(),