llvm / llvm/torch-mlir

Visualizing a mlir converted model

Open
#2,376 1 comment 0 reactions 0 assignees View on GitHub

Nobody has claimed this yet.

Dominant language
C++
Stars
1.9k
Forks
736
Avg merge
5d 22h
Merged PRs (30d)
15

Description

Hi, I wanted to know if there is a way I can save a model after torch_mlir compile and then visualize that model to see how the lowering of operators is happening. This will be really helpful for me to understand this.

Contributor guide

No contributing guide indexed for this repository

First steps

  1. Read the whole issue, then the project's contributing guide.
  2. Comment on the issue to say you are picking it up — it saves two people doing the same work.
  3. Fork the repository and make your change on a branch.
  4. Open a pull request that references the issue number.

Research direction

Start by investigating the torch_mlir compile flow and how lowered operators are represented after compilation. Determine whether the project already exposes a model-save or visualization entry point; done would require a documented, reproducible way to inspect operator lowering, but the issue does not name files or tests.

Written by the indexing model from the issue text.

Assessment

Tech stack
cpp, pytorch
Domain
compilers, machine-learning
Issue type
Feature
Difficulty
5/5
Estimated time
Over a week
Activity status
Stale
Clarity
Needs clarification
Newbie friendliness
25/100

Get new issues in your inbox

A short digest of beginner-friendly GitHub issues.