diff --git a/src/pyjuice/utils/kernel_launcher.py b/src/pyjuice/utils/kernel_launcher.py
index fe03b765..2d535025 100644
--- a/src/pyjuice/utils/kernel_launcher.py
+++ b/src/pyjuice/utils/kernel_launcher.py
@@ -37,10 +37,6 @@ def wrapper(*args, **kwargs):
             if "batch_size" in kwargs:
                 signature_list.append(("batch_size", kwargs["batch_size"]))
 
-            for i, arg in enumerate(args):
-                if isinstance(arg, torch.Tensor):
-                    signature_list.append((i, id(arg)))
-
             grid_length = len(grid)
             grid0, grid1, grid2 = grid[0], grid[1] if grid_length > 1 else 1, grid[2] if grid_length > 2 else 1