From b6d7703e02797b1dd71673aa63f84fdfc81c70b3 Mon Sep 17 00:00:00 2001
From: Pavel Belevich <belevichp@gmail.com>
Date: Fri, 2 Jan 2026 11:23:43 -0500
Subject: [PATCH] Refactor binary_op functions to remove unused result
 parameter

---
 examples/python/CuTeDSL/notebooks/tensorssa.ipynb | 10 ++++------
 1 file changed, 4 insertions(+), 6 deletions(-)

diff --git a/examples/python/CuTeDSL/notebooks/tensorssa.ipynb b/examples/python/CuTeDSL/notebooks/tensorssa.ipynb
index b082cb9a..62804a10 100644
--- a/examples/python/CuTeDSL/notebooks/tensorssa.ipynb
+++ b/examples/python/CuTeDSL/notebooks/tensorssa.ipynb
@@ -167,7 +167,7 @@
    "outputs": [],
    "source": [
     "@cute.jit\n",
-    "def binary_op_1(res: cute.Tensor, a: cute.Tensor, b: cute.Tensor):\n",
+    "def binary_op_1(a: cute.Tensor, b: cute.Tensor):\n",
     "    a_vec = a.load()\n",
     "    b_vec = b.load()\n",
     "\n",
@@ -194,8 +194,7 @@
     "a.fill(1.0)\n",
     "b = np.empty((3,), dtype=np.float32)\n",
     "b.fill(2.0)\n",
-    "res = np.empty((3,), dtype=np.float32)\n",
-    "binary_op_1(from_dlpack(res), from_dlpack(a), from_dlpack(b))"
+    "binary_op_1(from_dlpack(a), from_dlpack(b))"
    ]
   },
   {
@@ -205,7 +204,7 @@
    "outputs": [],
    "source": [
     "@cute.jit\n",
-    "def binary_op_2(res: cute.Tensor, a: cute.Tensor, c: cutlass.Constexpr):\n",
+    "def binary_op_2(a: cute.Tensor, c: cutlass.Constexpr):\n",
     "    a_vec = a.load()\n",
     "\n",
     "    add_res = a_vec + c\n",
@@ -230,8 +229,7 @@
     "a = np.empty((3,), dtype=np.float32)\n",
     "a.fill(1.0)\n",
     "c = 2.0\n",
-    "res = np.empty((3,), dtype=np.float32)\n",
-    "binary_op_2(from_dlpack(res), from_dlpack(a), c)"
+    "binary_op_2(from_dlpack(a), c)"
    ]
   },
   {