================
@@ -2716,6 +2716,37 @@ void CodeGenFunction::EmitStoreThroughLValue(RValue Src,
LValue Dst,
return EmitStoreThroughGlobalRegLValue(Src, Dst);
if (Dst.isMatrixElt()) {
+ if (getLangOpts().HLSL) {
+ // HLSL allows direct access to matrix elements, so storing to
+ // individual elements of a matrix through MatrixElt is handled as
+ // separate store instructions.
+ Address DstAddr = Dst.getMatrixAddress();
+ llvm::Type *DestAddrTy = DstAddr.getElementType();
+ llvm::Type *ElemTy = DestAddrTy->getScalarType();
+ CharUnits ElemAlign = CharUnits::fromQuantity(
+ CGM.getDataLayout().getPrefTypeAlign(ElemTy));
+
+ assert(ElemTy->getScalarSizeInBits() >= 8 &&
+ "matrix element type must be at least byte-sized");
+
+ llvm::Value *Val = Src.getScalarVal();
+ if (Val->getType()->getPrimitiveSizeInBits() <
+ ElemTy->getScalarSizeInBits())
+ Val = Builder.CreateZExt(Val, ElemTy->getScalarType());
+
+ llvm::Value *Idx = Dst.getMatrixIdx();
+ if (CGM.getCodeGenOpts().OptimizationLevel > 0) {
+ const auto *const MatTy =
Dst.getType()->castAs<ConstantMatrixType>();
----------------
bob80905 wrote:
There's 2 consts here, I think the inner one should be removed?
https://github.com/llvm/llvm-project/pull/176216
_______________________________________________
cfe-commits mailing list
[email protected]
https://lists.llvm.org/cgi-bin/mailman/listinfo/cfe-commits
