[Mlir-commits] [llvm] [mlir] [mlir][mesh] Add all-scatter operation (PR #81218)
Lei Zhang
llvmlistbot at 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 Mlir-commits
mailing list