| 183 | } |
| 184 | |
| 185 | IrArray::Index IrArray::Index::SourceIndexOfSlice( |
| 186 | const Shape& operand_shape, absl::Span<const int64> starts, |
| 187 | absl::Span<const int64> strides, llvm::IRBuilder<>* builder) const { |
| 188 | std::vector<llvm::Value*> source_multi_index(multidim_.size()); |
| 189 | for (int i = 0; i < multidim_.size(); ++i) { |
| 190 | int64 stride = strides[i]; |
| 191 | auto type = multidim_[i]->getType(); |
| 192 | |
| 193 | if (stride != 1) { |
| 194 | source_multi_index[i] = builder->CreateAdd( |
| 195 | builder->CreateMul(multidim_[i], |
| 196 | llvm::ConstantInt::get(type, stride)), |
| 197 | llvm::ConstantInt::get(type, starts[i])); |
| 198 | } else { |
| 199 | source_multi_index[i] = builder->CreateAdd( |
| 200 | multidim_[i], llvm::ConstantInt::get(type, starts[i])); |
| 201 | } |
| 202 | } |
| 203 | return Index(source_multi_index, operand_shape, index_type_); |
| 204 | } |
| 205 | |
| 206 | IrArray::Index IrArray::Index::SourceIndexOfTranspose( |
| 207 | const Shape& shape, const Shape& operand_shape, |
no test coverage detected