Master Issue for model bringup from Torch and Jax

[tapspatel]

Torch

1. Write a model in torch
2. Compile using torch.compile

• Showcase torch backend and custom_backend
• Showcase FX graph generated

3. For custom_backend

• Showcase stablehlo IR (mlir)
• Showcase TT IR (mlir)
• Showcase TTNN IR (mlir)

4. In visualizer, have different run options

• From stablehlo IR
• From TT IR
• From TTNN IR

5. In visualizer, showcase golden accuracy information

• For stablehlo IR
• For TT IR
• For TTNN IR

6. In visualizer, showcase performance information

• For stablehlo IR
• For TT IR
• For TTNN IR

7. In visualizer, showcase memory information

• For stablehlo IR
• For TT IR
• For TTNN IR

JAX

1. Write a model in jax
2. Compile using jax.jit

• Showcase vhlo IR (mlir)
• Showcase shlo IR (mlir)
• Showcase TT IR (mlir)
• Showcase TTNN IR (mlir)

3. In visualizer, have different run options

• From vhlo IR
• From shlo IR
• From TT IR
• From TTNN IR

4. In visualizer, showcase golden accuracy information

• For vhlo IR
• For stablehlo IR
• For TT IR
• For TTNN IR

5. In visualizer, showcase performance information

• For vhlo IR
• For stablehlo IR
• For TT IR
• For TTNN IR

6. In visualizer, showcase memory information

• For vhlo IR
• For stablehlo IR
• For TT IR
• For TTNN IR

Sub-Issues to finish goals
MLIR

• Update memory report in ttrt with new memory allocator #1683
• Visualize IRs in tt-visualizer for model bringup from Torch #1706
• Enable different run options for IRs for model bringup from Torch #1707
• Visualize golden accuracy information for different IRs for model bringup from Torch #1708
• Visualize performance information for different IRs for model bringup from Torch #1709
• Visualize memory information for different IRs for model bringup from Torch #1710
• Visualize IRs in tt-visualizer for model bringup from JAX #1711
• Enable different run options for IRs for model bringup from JAX #1712
• Visualize golden accuracy information for different IRs for model bringup from JAX #1713
• Visualize performance information for different IRs for model bringup from JAX #1714
• Visualize memory information for different IRs for model bringup from JAX #1715
• Visualize ttir graph in tt-explorer #1719
• Add arrow points from parent -> child op in tt-explorer #1720
• Visualize stablehlo/vhlo graphs in tt-explorer #1721
• Enable running of a model from vhlo, shlo, and ttnn dialects #1722
• Showcase golden accuracy information in tt-explorer for vhlo, shlo, ttir, ttnn #1723
• Showcase performance information in tt-explorer for vhlo, shlo, ttir, ttnn #1724
• Showcase memory information in tt-explorer for vhlo, shlo, ttir, ttnn #1725

tt-xla

• Master Issue for model bringup for tt-mlir feature verification tt-xla#150
• Tests running on CPU tt-xla#152

tt-torch

• Master Issue for model bringup for tt-mlir feature verification tt-torch#161