Skip to content

[ingress][torch][runner] PyTorch backend - dump object file#181

Merged
adam-smnk merged 4 commits into
llvm:mainfrom
adam-smnk:torch-backend-dump-obj
Jun 3, 2026
Merged

[ingress][torch][runner] PyTorch backend - dump object file#181
adam-smnk merged 4 commits into
llvm:mainfrom
adam-smnk:torch-backend-dump-obj

Conversation

@adam-smnk
Copy link
Copy Markdown
Member

@adam-smnk adam-smnk commented Jun 3, 2026

Enables MLIR object file creation in the PyTorch backend.
Removes temporary file creation for the object files. Instead, users must explicitly provide a file name for the dumps.

Tools are updated accordingly.
The new functionality is used in kernel-bench tool to enable assembly dumps when torch-compile option is enabled.

Enables MLIR object file creation in the PyTorch backend.

The new functionality is used in kernel-bench tool to enable assembly
dumps when torch-compile option is enabled.
@adam-smnk adam-smnk requested a review from rengolin June 3, 2026 11:45
Copy link
Copy Markdown
Member

@rengolin rengolin left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

filename should come from the user/tool. Change the argument from boolean to string (default empty, if not, dumps to a file with that string name). This also avoids printing to stdout from inside a core module.

Comment thread tools/kernel-bench
@adam-smnk adam-smnk changed the title [ingress][torch] PyTorch backend - dump object file [ingress][torch][runner] PyTorch backend - dump object file Jun 3, 2026
@adam-smnk adam-smnk merged commit c61a06e into llvm:main Jun 3, 2026
3 checks passed
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

None yet

Projects

None yet

Development

Successfully merging this pull request may close these issues.

2 participants