New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Jump to bottom

Master Issue for model bringup for tt-mlir feature verification #150

Open

1 task done

Tracked by

tapspatel opened this issue Jan 6, 2025 · 0 comments

Open

1 task done

Tracked by

Master Issue for model bringup for tt-mlir feature verification #150

tapspatel opened this issue Jan 6, 2025 · 0 comments

Assignees

tapspatel commented Jan 6, 2025 •

edited

Loading

JAX

Write a model in jax
Compile using jax.jit

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

In visualizer, have different run options

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

In visualizer, showcase golden accuracy information

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

In visualizer, showcase performance information

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

In visualizer, showcase memory information

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

Sub-Issues to finish goals

Tests running on CPU #152

tapspatel self-assigned this

tapspatel mentioned this issue

Master Issue for model bringup from Torch and Jax tenstorrent/tt-mlir#1705

Open

21 tasks

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment