llvm
/
torch-mlir
mirror of https://github.com/llvm/torch-mlir
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Update.

ziheng/mhlo-index_put
Ziheng Jiang 2023-02-01 19:58:57 -08:00
parent bba254ab1c
commit 6c07704837
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
lib/Conversion/TorchToMhlo/Basic.cpp
View File

@ -1307,7 +1307,7 @@ LogicalResult ConvertAtenOp<Aten_IndexPutImplOp>::matchAndRewrite(
}
auto valuesType = values.getType().cast<RankedTensorType>();
SmallVector<int64_t> valuesShape(valuesType.getShape());
if (valueShape.size() != 3) {
if (valuesShape.size() != 3) {
return op->emitError("only support 3D values in index_put");
}