[MLIR] enhance get_scalar op, only requires dyn_leaf_cnt = 1

[sjfeng1999]

Motivation

Technical Details

Test Plan

Test Result

Submission Checklist

• Look over the contributing guidelines at https://github.com/ROCm/ROCm/blob/develop/CONTRIBUTING.md#pull-requests.

[Copilot]

Pull request overview

Enhances Fly dialect get_scalar handling so it can accept nested singleton IntTuple shapes (e.g. ((((42)))) / ((((?))))) rather than requiring the tuple attr to be a direct leaf, and adds coverage in the layout-lowering MLIR tests.

Changes:

• Update GetScalarOp return type inference to unwrap rank-1 nesting until a leaf is reached.
• Update fly-layout-lowering’s GetScalarOp rewrite to perform the same unwrapping.
• Add MLIR tests for nested singleton static and dynamic get_scalar.

Reviewed changes

Copilot reviewed 3 out of 3 changed files in this pull request and generated 1 comment.

File	Description
tests/mlir/Transforms/layout_lowering.mlir	Adds FileCheck tests validating nested singleton get_scalar is eliminated to constants/SSA values.
lib/Dialect/Fly/Transforms/LayoutLowering.cpp	Adjusts GetScalarOp lowering to unwrap nested singleton tuples before extracting the leaf.
lib/Dialect/Fly/IR/FlyOps.cpp	Adjusts GetScalarOp type inference to accept nested singleton tuples and improve the diagnostic for non-scalars.

---

💡 Add Copilot custom instructions for smarter, more guided reviews. Learn how to get started.