Skip to content

Commit

Permalink
Update base for Update on "[innductor] make inductor work with new tr…
Browse files Browse the repository at this point in the history
…iton compile interface"


Recent 2 triton PRs (triton-lang/triton#2701, triton-lang/triton#2756) change the interface for triton.compile, this PR added the necessary change on inductor side to work with both old and new compile API.

Also there is some simplification between compilation call in subprocess and the one in main process
- previously we pass warm_cache_only=True if the compilation happens in subprocess. But triton never use that argument in the currently used pin. So I removed that
- previously we only pass compute_capability if compilation happens in subprocess. The PR change that to always passing compute_capability to triton.compile no matter if the compilation happens in main or sub process.


Updated:
There are more interface change from triton side. E.g.
- tl.math.{min, max} now requires a propagate_nan argument

cc voznesenskym penguinwu EikanWang jgong5 Guobing-Chen XiaobingSuper zhuhaozhe blzheng wenzhe-nrv jiayisunx peterbell10 ipiszy yf225 chenyang78 kadeng muchulee8 aakhundov ColinPeppler

[ghstack-poisoned]
  • Loading branch information
shunting314 committed Dec 19, 2023
1 parent 645ac26 commit c915953
Showing 0 changed files with 0 additions and 0 deletions.

0 comments on commit c915953

Please sign in to comment.