Skip to content

Commit

Permalink
[torch-mlir][sparse] inline sparse helper methods (#2918)
Browse files Browse the repository at this point in the history
Even though the reference compiler is not about performance, inlining
the generated sparse helper methods has a rather big positive impact on
performance, leaving a much better first impression. Therefore, we added
this inlining pass (which leaves all other PyTorch modules unaffected,
since they tend to be one big main() method to start with).

testing:

$./tools/e2e_test.sh --config linalg

Summary:
    Passed: 1164
    Expectedly Failed: 8

$ python -m e2e_testing.main --config=torchdynamo

Summary:
    Passed: 976
    Expectedly Failed: 162
  • Loading branch information
aartbik authored Feb 17, 2024
1 parent d65925a commit 78e10ff
Showing 1 changed file with 1 addition and 0 deletions.
Original file line number Diff line number Diff line change
Expand Up @@ -139,6 +139,7 @@ def invoke(*args):
"sparse-assembler",
"sparsification-and-bufferization",
"sparse-storage-specifier-to-llvm",
"inline", # inline sparse helper methods where useful
# Bufferize.
"func.func(scf-bufferize)",
"func.func(tm-tensor-bufferize)",
Expand Down

0 comments on commit 78e10ff

Please sign in to comment.