From a9b2cf949f1ac1d88c027ce28c56bf0b0e10596a Mon Sep 17 00:00:00 2001 From: Anji Liu Date: Tue, 9 Jan 2024 19:44:33 +0800 Subject: [PATCH] remove address checking --- src/pyjuice/utils/kernel_launcher.py | 4 ---- 1 file changed, 4 deletions(-) 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