try fix Tile(block_id) error

Author: yf225

benchmarks/run.py

@@ -325,6 +325,23 @@ def run_kernel_variants(
     operator_args: dict[str, Any] | None = None,
 ) -> None:
     """Run kernel variants in the same benchmark run."""
+    
+    # Configure Helion to use fewer generations for faster autotuning during benchmarks
+    import helion
+    from helion.autotuner import DifferentialEvolutionSearch, LocalAutotuneCache
+    from helion.runtime.kernel import BoundKernel
+    from typing import Sequence
+    
+    def fast_autotuner_fn(
+        bound_kernel: BoundKernel, args: Sequence[object], **kwargs: object
+    ) -> LocalAutotuneCache:
+        # Use only 1 generation instead of default 20 for faster benchmarking
+        return LocalAutotuneCache(
+            DifferentialEvolutionSearch(bound_kernel, args, num_generations=1, **kwargs)
+        )
+    
+    # Set the custom autotuner function
+    helion.set_default_settings(helion.Settings(autotuner_fn=fast_autotuner_fn))
 
     # Configure Helion to use fewer generations for faster autotuning during benchmarks
     import helion

helion/_compiler/type_propagation.py

@@ -990,12 +990,19 @@ def __init__(self, origin: Origin, block_id: int) -> None:
         self.block_id = block_id
 
     def proxy(self) -> object:
+        from ..language.tile_proxy import Tile as TileClass
+        
         with proxy_tensor.disable_proxy_modes_tracing():
             fake_mode = torch._C._unset_dispatch_mode(  # pyright: ignore[reportAttributeAccessIssue]
                 torch._C._TorchDispatchModeKey.FAKE  # pyright: ignore[reportAttributeAccessIssue]
             )
             try:
-                return Tile(self.block_id)
+                # Create a Tile instance using torch.as_subclass to properly handle tensor subclassing
+                # This avoids the "already associated to a python object" error
+                base_tensor = torch.empty([], dtype=torch.int64, device='meta')
+                tile = base_tensor.as_subclass(TileClass)
+                tile.block_id = self.block_id
+                return tile
             finally:
                 assert fake_mode is not None
                 torch._C._set_dispatch_mode(fake_mode)  # pyright: ignore[reportAttributeAccessIssue]