[Enhancement] Include Metadata (LayoutMap etc.) into hashing (#570)

- Modified the serialization of function scripts in both KernelCache and AutoTunerCache to include metadata by setting `show_meta=True` in `cloudpickle.dumps()`. This change enhances the hash key generation for kernel configurations, improving cache accuracy and consistency.

testing/python/transform/test_tilelang_transform_frontend_legalize.py

@@ -8,7 +8,6 @@ def _check(original, transformed):
     func = original
     mod = tvm.IRModule.from_expr(func.with_attr("global_symbol", "main"))
     mod = tl.transform.FrontendLegalize()(mod)
-    print(mod.script())
     tvm.ir.assert_structural_equal(mod["main"], transformed.with_attr("global_symbol", "main"),
                                    True)
 

tilelang/cache/kernel_cache.py

@@ -91,7 +91,7 @@ class KernelCache:
             str: SHA256 hash key for the kernel configuration.
         """
         self.execution_backend = execution_backend
-        func_binary = cloudpickle.dumps(func.script())
+        func_binary = cloudpickle.dumps(func.script(show_meta=True))
         key_data = {
             "version": __version__,
             "func": sha256(func_binary).hexdigest(),  # Use SHA256 to generate hash key

tilelang/cache/tuner_cache.py

@@ -86,7 +86,7 @@ class AutoTunerCache:
         Returns:
             str: SHA256 hash key for the kernel configuration.
         """
-        func_binary = cloudpickle.dumps(func.script())
+        func_binary = cloudpickle.dumps(func.script(show_meta=True))
         key_data = {
             "version": __version__,
             "func": sha256(func_binary).hexdigest(),  # Use SHA256 to generate hash key