Skip to content

Use TileSwizzle for inner_tiled layout on CPU#23705

Merged
bjacob merged 1 commit intoiree-org:mainfrom
bjacob:cpu_tile_swizzle
Mar 9, 2026
Merged

Use TileSwizzle for inner_tiled layout on CPU#23705
bjacob merged 1 commit intoiree-org:mainfrom
bjacob:cpu_tile_swizzle

Conversation

@bjacob
Copy link
Collaborator

@bjacob bjacob commented Mar 9, 2026

This allows removing CPUOpaqueMmaLayout, so CPU doesn't need any CPU-specific data structure anymore.

This involved moving some code out of GPUTileSwizzle.* into IREECodegenTypes where TileSwizzle is defined.

As these TileSwizzle objects are what determines the tile shapes tested by iree_cpu_inner_tiled_ops.mlir, this test provides test coverage for this PR.

@bjacob bjacob marked this pull request as ready for review March 9, 2026 15:48
Copy link
Contributor

@krzysz00 krzysz00 left a comment

Choose a reason for hiding this comment

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

This all looks fairly NFC to me, approved

Signed-off-by: Benoit Jacob <[email protected]>
@bjacob bjacob force-pushed the cpu_tile_swizzle branch from a8534fd to 77c6b4e Compare March 9, 2026 16:40
@bjacob bjacob enabled auto-merge (squash) March 9, 2026 16:40
@bjacob bjacob merged commit 616a969 into iree-org:main Mar 9, 2026
56 checks passed
Copy link
Contributor

Choose a reason for hiding this comment

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

Can you move the declarations to IREECPUTypes.h?

When I initiated it, I only wanted to avoid numbers of includes. Some time later, there was a discussion, and I'm +1 on what Ben said: https://discord.com/channels/689900678990135345/689900680009482386/1417237795914907659

I usually just have a FooTypes.h that includes everything but the ops - the ops are often the biggest thing I want to avoid polluting scope with
(attrs/interfaces/etc are type-like, at least, in that they are C++ types... is what I tell myself lol)

Copy link
Collaborator Author

Choose a reason for hiding this comment

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

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.

3 participants