[Enhancement] Update kernel declaration pattern to support launch bounds in...

[Enhancement] Update kernel declaration pattern to support launch bounds in match_declare_kernel function (#366)

- Modified the regex pattern in `match_declare_kernel` to accommodate optional `__launch_bounds__` specifications, enhancing the function's ability to match kernel declarations accurately.
- This change improves the flexibility of kernel matching in the source code, allowing for more complex kernel definitions.
Co-authored-by: Zhiwen Mo <zhiwen.mo25@ic.ac.uk>

tilelang/jit/adapter/utils.py

@@ -29,7 +29,7 @@ def match_global_kernel(source: str, annotation: str = "__global__") -> int:
 
 
 def match_declare_kernel(source: str, annotation: str = "__global__") -> int:
-    pattern = r"__global__\s+void\s+\w+"
+    pattern = r"__global__\s+void\s+(?:__launch_bounds__\(\d+\)\s+)?\w+"
     for line in source.split("\n"):
         if annotation in line:
             matched = re.findall(pattern, line)