diff options
Diffstat (limited to 'source')
| -rw-r--r-- | source/slang/slang-ir-autodiff-transpose.h | 7 |
1 files changed, 3 insertions, 4 deletions
diff --git a/source/slang/slang-ir-autodiff-transpose.h b/source/slang/slang-ir-autodiff-transpose.h index a5ed5814c..5e96c4e0f 100644 --- a/source/slang/slang-ir-autodiff-transpose.h +++ b/source/slang/slang-ir-autodiff-transpose.h @@ -1812,8 +1812,8 @@ struct DiffTransposePass { List<RevGradient> gradients; auto matrixType = as<IRMatrixType>(fwdMakeMatrix->getDataType()); - auto row = as<IRIntLit>(matrixType->getRowCount()); auto colCount = matrixType->getColumnCount(); + auto colCountVal = as<IRIntLit>(matrixType->getColumnCount())->getValue(); IRType* rowVectorType = nullptr; for (UIndex ii = 0; ii < fwdMakeMatrix->getOperandCount(); ii++) { @@ -1828,9 +1828,8 @@ struct DiffTransposePass } else { - SLANG_RELEASE_ASSERT(row); - UInt rowIndex = ii / (UInt)row->getValue(); - UInt colIndex = ii % (UInt)row->getValue(); + UInt rowIndex = ii / (UInt)colCountVal; + UInt colIndex = ii % (UInt)colCountVal; if (!rowVectorType) rowVectorType = builder->getVectorType(matrixType->getElementType(), colCount); auto revRow = builder->emitElementExtract( |