Fix wrong reverting

Summary: D78130587 mistakenly reverted a previous change. this diff takes it back.

Reviewed By: Sibylau

Differential Revision: D78182104

fbshipit-source-id: e0264e1849a7677192d4750a82c93f77002d9c5a

Author: FindHao

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,10 @@ 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