[Mlir-commits] [mlir] [MLIR][XeGPU] Xegpu distribution patterns for load_nd, store_nd, and create_nd_tdesc. (PR #112945)
    llvmlistbot at llvm.org 
    llvmlistbot at llvm.org
       
    Tue Oct 29 13:17:39 PDT 2024
    
    
  
================
@@ -6558,14 +6558,14 @@ static LogicalResult verifyDistributedType(Type expanded, Type distributed,
   // If the types matches there is no distribution.
   if (expanded == distributed)
     return success();
-  auto expandedVecType = llvm::dyn_cast<VectorType>(expanded);
-  auto distributedVecType = llvm::dyn_cast<VectorType>(distributed);
+  auto expandedVecType = llvm::dyn_cast<ShapedType>(expanded);
----------------
MaheshRavishankar wrote:
If you want to generalize distribution to work on more than vector types, this whole thing needs to be moved out of vector dialect and made an interface. Checking for just `ShapedType` here seems like a violation.
https://github.com/llvm/llvm-project/pull/112945
    
    
More information about the Mlir-commits
mailing list