diff --git a/examples/a2a3/tensormap_and_ringbuffer/scalar_data_test/test_scalar_data.py b/examples/a2a3/tensormap_and_ringbuffer/scalar_data_test/test_scalar_data.py index 590bda5cb2..49771dcd8b 100644 --- a/examples/a2a3/tensormap_and_ringbuffer/scalar_data_test/test_scalar_data.py +++ b/examples/a2a3/tensormap_and_ringbuffer/scalar_data_test/test_scalar_data.py @@ -60,7 +60,10 @@ def generate_args(self, params): Tensor("a", torch.full((SIZE,), 2.0, dtype=torch.float32)), Tensor("b", torch.arange(SIZE, dtype=torch.float32)), Tensor("result", torch.zeros(SIZE, dtype=torch.float32)), - Tensor("check", torch.zeros(10, dtype=torch.float32)), + # Exactly 9 slots: the orchestration writes check[0..8] via + # set_tensor_data. Output-tensor slots are not seeded from the host, + # so any extra slot reads undefined device memory. + Tensor("check", torch.zeros(9, dtype=torch.float32)), ) def compute_golden(self, args, params):