[llvm] [mlir] [mlir][mesh] Add all-scatter operation (PR #81218)

Lei Zhang via llvm-commits llvm-commits at lists.llvm.org
Tue Feb 13 12:02:00 PST 2024


================
@@ -590,6 +596,22 @@ static LogicalResult verifyScatterOperandAndResultShape(
   return success();
 }
 
+static RankedTensorType scatterResultType(Type operandType, MeshOp mesh,
+                                          ArrayRef<MeshAxis> meshAxes,
+                                          int64_t scatterAxis) {
+  RankedTensorType operandRankedTensorType =
+      operandType.cast<RankedTensorType>();
----------------
antiagainst wrote:

This style of casting is deprecated? Prefer to use `cast<RankedTensorType>(operandType)`.

https://github.com/llvm/llvm-project/pull/81218


More information about the llvm-commits mailing list