[Mlir-commits] [mlir] [MLIR][XeGPU] Add support for subgroup_id_range (PR #148661)
Chao Chen
llvmlistbot at llvm.org
Mon Jul 21 07:44:32 PDT 2025
================
@@ -34,6 +34,26 @@ using namespace mlir;
namespace {
+// Check if there is sg id range attached to the scf.if op.
+static bool isSgIdRangeSpecified(Operation *op, int64_t &startOfRange,
+ int64_t &endOfRange) {
+ Operation *parent = op->getParentOp();
+ // Find the outermost scf::IfOp with xegpu.sg_id_range.
+ while (parent) {
+ if (auto ifOp = dyn_cast<scf::IfOp>(parent)) {
+ if (auto attr = llvm::dyn_cast_or_null<xegpu::RangeAttr>(
+ ifOp->getAttr("sg_id_range"))) {
+ startOfRange = attr.getStart().getInt();
+ endOfRange = attr.getEnd().getInt();
+ break;
+ }
+ }
+ parent = parent->getParentOp();
+ }
+ // Return false if startOfRange is 0
----------------
chencha3 wrote:
Why it returns false when startOfRange is 0?
https://github.com/llvm/llvm-project/pull/148661
More information about the Mlir-commits
mailing list