remove address checking

Tractables · Jan 9, 2024 · a9b2cf9 · a9b2cf9
1 parent fbd191b
commit a9b2cf9
Showing 1 changed file with 0 additions and 4 deletions.
diff --git 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