Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

[LLVMGPU] Add a verifier for tile sizes. #19906

Open
wants to merge 7 commits into
base: main
Choose a base branch
from

Conversation

erman-gurses
Copy link
Member

@erman-gurses erman-gurses commented Feb 4, 2025

Add a verifier for tile sizes.

Signed-off-by: erman-gurses <[email protected]>
Signed-off-by: erman-gurses <[email protected]>
@erman-gurses erman-gurses changed the title Add verifier for tile sizes. [LLVMGPU] Add verifier for tile sizes. Feb 5, 2025
Signed-off-by: erman-gurses <[email protected]>
@erman-gurses erman-gurses changed the title [LLVMGPU] Add verifier for tile sizes. [LLVMGPU] Add a verifier for tile sizes. Feb 5, 2025
Signed-off-by: erman-gurses <[email protected]>
Signed-off-by: erman-gurses <[email protected]>
Signed-off-by: erman-gurses <[email protected]>
@erman-gurses erman-gurses force-pushed the users/erman-gurses/tilesize-verifier branch 22 times, most recently from a462b4f to ae7e5a6 Compare February 11, 2025 06:50
@erman-gurses erman-gurses force-pushed the users/erman-gurses/tilesize-verifier branch 21 times, most recently from 0a7e8fd to 8a9f54d Compare February 17, 2025 19:36
Signed-off-by: erman-gurses <[email protected]>
@erman-gurses erman-gurses force-pushed the users/erman-gurses/tilesize-verifier branch from 8a9f54d to 09f2633 Compare February 17, 2025 19:38
@erman-gurses erman-gurses marked this pull request as ready for review February 17, 2025 20:19
@erman-gurses erman-gurses linked an issue Feb 17, 2025 that may be closed by this pull request
getLoweringConfig<IREE::Codegen::LoweringConfigAttr>(op);
if (!loweringConfig)
auto gpuLoweringConfig =
getLoweringConfig<IREE::GPU::LoweringConfigAttr>(op);
Copy link
Contributor

Choose a reason for hiding this comment

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

Rather than casting and passing two lowering config attributes, it would be cleaner to add an additional template parameter to this function (template <typename F, typename ConfigTy>) and only cast to the config type that works for F. Then rather than combining the verifiers for LLVMGPUMatmulTensorCoreMmaSync and LLVMGPUVectorDistribute, we have two separate verifyGPUMatmulPipeline functions, one for the GPU lowering config and one for the Codegen lowering config.

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.

[LLVMGPU] Add a verifier for tile sizes in lowering_config
2 participants