Update TritonParse to disable warning on Triton language element type arguments

Summary: Disables a warning generated if you pass a dtype (e.g. `tl.float32`) into a TritonBench kernel as a constexpr. This is needed for expanding the warp spec examples to allow for different data types.

Reviewed By: FindHao

Differential Revision: D77948317

fbshipit-source-id: 68bca8669e34618033ea37ad173452114a180d59

Author: njriasan

tritonparse/structured_logging.py

@@ -142,6 +142,8 @@ def convert(obj):
     Returns:
         A serializable version of the input object where dataclasses are converted to dictionaries
     """
+    from triton.language.core import dtype
+
     # 1. primitives that JSON already supports  -------------------------------
     if obj is None or isinstance(obj, (bool, int, str)):
         return obj
@@ -176,6 +178,11 @@ def convert(obj):
         return convert(
             asdict(obj)
         )  # Convert dataclass to dict and then process that dict
+
+    # 4. Common Triton constexpr objects
+    if isinstance(obj, dtype):
+        return f"triton.language.core.dtype('{str(obj)}')"
+
     log.warning(f"Unknown type: {type(obj)}")
     return str(obj)  # Return primitive types as-is