Skip to content
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.

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

Explicitly give mesh shape and systemDescPath to the pipeline option in ttir->ttnn MLIR conversion. #201

Open
ajakovljevicTT opened this issue Jan 29, 2025 · 0 comments
Assignees
Labels
multichip Multichip issues

Comments

@ajakovljevicTT
Copy link
Contributor

ajakovljevicTT commented Jan 29, 2025

Currently, we rely on having the default options set when calling the createTTIRToTTNNBackendPipeline in our ModuleBuilder, which translates TTIR to TTNN mlir dialect. This will not be possible for multichip configurations, as we need to explicitly say which for which system descriptor and what shape we will compile our model. Currently, an intermediate solution is given like this:

  mlir::tt::ttnn::TTIRToTTNNBackendPipelineOptions options;
  options.systemDescPath="system_desc.ttsys"; // This is manually copied to our tt-xla repo
  options.meshShape = {1,2};
  mlir::tt::ttnn::createTTIRToTTNNBackendPipeline(ttir_to_ttnn_pm, options);

We need to find a way to get these system descriptors specifically, along with the correct mesh shape.

@ajakovljevicTT ajakovljevicTT added the multichip Multichip issues label Jan 29, 2025
@nsmithtt nsmithtt added this to the [Multi Device 1] milestone Jan 29, 2025
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
multichip Multichip issues
Projects
None yet
Development

No branches or pull requests

3 participants