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

[Stablehlo]index type use i64 (#3354)

pull/3085/head
Xinyu Yang 2024-05-16 15:33:23 +08:00 committed by GitHub
parent 7faba75696
commit 28193fd985
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 1 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
lib/Conversion/TorchToStablehlo/GatherScatter.cpp
View File

@ -247,8 +247,7 @@ FailureOr<Value> broadcastAndConcatIndices(Operation *op,
concatShape.push_back(indexTensors.size()); concatShape.push_back(indexTensors.size());
SmallVector<Value> broadcastedIndices; SmallVector<Value> broadcastedIndices;
Type indexElemTy = Type indexElemTy = rewriter.getI64Type();
cast<RankedTensorType>(indexTensors[0].getType()).getElementType();
RankedTensorType bcastIndexType = RankedTensorType bcastIndexType =
RankedTensorType::get(indicesShape, indexElemTy); RankedTensorType::get(indicesShape, indexElemTy);
for (auto indexTensor : indexTensors) { for (auto indexTensor : indexTensors) {