================ @@ -30,6 +31,61 @@ void XeGPUDialect::initialize() { >(); } +bool XeGPUDialect::isEvenlyDistributable(llvm::ArrayRef<int64_t> shape, ---------------- Garra1980 wrote: let's have the comment here as it was earlier https://github.com/llvm/llvm-project/pull/137226