[DEV] Support TRITON_INTERPRET=1

[mark14wu]

[FIX] Trace now supports InterpretedFunction:

• class Trace constructor now accepts either JITFunction or InterpretedFunction (re-using the latter or wrapping the former)
• The trace() decorator wraps both types on first use, and the error message lists all accepted kernel types.

[Copilot]

Pull Request Overview

This PR enhances the Trace functionality by adding support for InterpretedFunction kernels. In addition to allowing InterpretedFunction in the Trace constructor, the decorator has been updated to wrap both JITFunction and InterpretedFunction and provide a clearer error message.

• Updated Trace.init to accept either JITFunction or InterpretedFunction.
• Modified the decorator to support both kernel types and adjusted error messaging.

Comments suppressed due to low confidence (1)

triton_viz/core/trace.py:37

• The constructor's type hint still restricts kernel to JITFunction even though the implementation now supports InterpretedFunction. Consider updating the signature to accept Union[JITFunction, InterpretedFunction] for clarity and consistency.

def __init__(self, kernel: JITFunction, client: Union[str, Client]) -> None:

[mark14wu]

triton_viz/core/trace.py:37

• The constructor's type hint still restricts kernel to JITFunction even though the implementation now supports InterpretedFunction. Consider updating the signature to accept Union[JITFunction, InterpretedFunction] for clarity and consistency.

def __init__(self, kernel: JITFunction, client: Union[str, Client]) -> None:

Thanks. Now fixed.