From 45674786e8b26e6fbd2dd9f0d81959fe4e7ae607 Mon Sep 17 00:00:00 2001 From: lucylq Date: Tue, 11 Nov 2025 15:44:41 -0800 Subject: [PATCH] Expect at least one value in the tensor, to extract scalar from Differential Revision: D86363818 Pull Request resolved: https://github.com/pytorch/executorch/pull/15627 (cherry picked from commit 47aca6912550d977c883424604dbc12d3a10f9de) --- 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 8607c36204d..4254188d6ee 100644 --- a/kernels/prim_ops/register_prim_ops.cpp +++ b/kernels/prim_ops/register_prim_ops.cpp @@ -124,6 +124,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(),