Skip to content

Conversation

@tungld
Copy link
Member

@tungld tungld commented Dec 12, 2025

This PR improved the pytorch backend for onnx-mlir by adding these features:

  • Cache the compiled model to disk and load it when starting a new process.
  • Add config.py to allow users set configurations for the backend
  • For each constants or parameters in the FX graph, use the first 3 values (the number can be configured via config) when hashing the graph, which makes the hash unique.
  • After N times of hashing the FX graph, if the graph is unchanged, stop hashing to reduce the overhead. N can be configured via config, default is 3.

tungld added 13 commits December 8, 2025 21:52
Signed-off-by: Tung D. Le <[email protected]>
Signed-off-by: Tung D. Le <[email protected]>
Signed-off-by: Tung D. Le <[email protected]>
Signed-off-by: Tung D. Le <[email protected]>
Signed-off-by: Tung D. Le <[email protected]>
Signed-off-by: Tung D. Le <[email protected]>
Signed-off-by: Tung D. Le <[email protected]>
Signed-off-by: Tung D. Le <[email protected]>
Signed-off-by: Tung D. Le <[email protected]>
@tungld tungld requested a review from chentong319 December 12, 2025 05:32
Signed-off-by: Tung D. Le <[email protected]>
Signed-off-by: Tung D. Le <[email protected]>
Signed-off-by: Tung D. Le <[email protected]>
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.

1 participant