Skip to content

iree: add an IREE Backend and a dedicated strategy - #107

Open
qaco wants to merge 15 commits into
xtc-tools:mainfrom
qaco:hugo/iree
Open

iree: add an IREE Backend and a dedicated strategy#107
qaco wants to merge 15 commits into
xtc-tools:mainfrom
qaco:hugo/iree

Conversation

@qaco

@qaco qaco commented Jul 24, 2026

Copy link
Copy Markdown
Contributor

Motivation

Experiment with IREE

Description

Implement all the interfaces for an IREE backend + expose an IREE-focused strategy.

Discussion

The performance seems to be consistent, at least on mac (IREE-focused strategy):

Figure_1

@qaco
qaco requested a review from guillon July 24, 2026 09:48
@qaco qaco self-assigned this Jul 24, 2026
@qaco
qaco marked this pull request as draft July 24, 2026 09:48
Comment thread src/xtc/backends/iree/IREECompiler.py
Comment thread src/xtc/backends/iree/IREECompiler.py Outdated
Comment thread src/xtc/backends/iree/IREEScheduler.py Outdated

# SIMD vector width (in f32 elements) used when a vectorized axis has no
# associated tile size (e.g. an untiled base dimension).
_VECTOR_WIDTH = 4

Copy link
Copy Markdown
Member

Choose a reason for hiding this comment

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

why not 8, or 16? Doesn't it work if we give a max (16).

Copy link
Copy Markdown
Contributor Author

Choose a reason for hiding this comment

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

Ok I bump it to 16. It's visibly safe to go this high because when the actual target is narrower, IREE narrows the vectors itself, so a larger default never over-shoots.

Comment thread src/xtc/csrcs/runtimes/iree/CMakeLists.txt Outdated
Comment thread src/xtc/runtimes/iree/IREERuntime.py
@qaco
qaco force-pushed the hugo/iree branch 4 times, most recently from b07bf4d to 7c17428 Compare July 30, 2026 10:55
@qaco qaco changed the title iree: add a basic IREE Backend (draft) iree: add an IREE Backend and a dedicated strategy Jul 30, 2026
@qaco
qaco marked this pull request as ready for review July 30, 2026 11:29
@qaco
qaco requested a review from guillon July 30, 2026 11:29
Comment thread src/xtc/runtimes/iree/IREERuntime.py Outdated
@guillon
guillon self-requested a review July 30, 2026 16:14
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