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

Fix TensorFromElementsOp reference

pull/155/head
Aaron J Arthurs 2021-01-22 08:34:09 -06:00 committed by Sean Silva
parent fc650c9447
commit c0e14da888
2 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
lib/Dialect/TCP/Transforms/Bufferize.cpp
View File

@ -49,7 +49,7 @@ static SmallVector<Value, 6> bypassResultShapes(Operation &op) {
Value outDim = builder.create<AddIOp>(op.getLoc(), totalExpansion, operandDim); Value outDim = builder.create<AddIOp>(op.getLoc(), totalExpansion, operandDim);
outDims.push_back(outDim); outDims.push_back(outDim);
} }
Value outDimTensor = builder.create<TensorFromElementsOp>(op.getLoc(), ValueRange(outDims)); Value outDimTensor = builder.create<tensor::FromElementsOp>(op.getLoc(), ValueRange(outDims));
return {outDimTensor}; return {outDimTensor};
} }

2
test/Dialect/TCP/bufferize.mlir
View File

@ -41,7 +41,7 @@ func @tcp_splatted(%arg0: f32, %arg1: tensor<?xindex>) -> tensor<?x?xf32> {
// CHECK: %[[D1:.*]] = dim %[[TENSOR]], %[[C0_0]] : tensor<?xf32> // CHECK: %[[D1:.*]] = dim %[[TENSOR]], %[[C0_0]] : tensor<?xf32>
// CHECK: %[[D1_EXPANSION:.*]] = addi %[[LOWER_EXTENT_D1]], %[[UPPER_EXTENT_D1]] : index // CHECK: %[[D1_EXPANSION:.*]] = addi %[[LOWER_EXTENT_D1]], %[[UPPER_EXTENT_D1]] : index
// CHECK: %[[D1_OUT:.*]] = addi %[[D1_EXPANSION]], %[[D1]] : index // CHECK: %[[D1_OUT:.*]] = addi %[[D1_EXPANSION]], %[[D1]] : index
// CHECK: %[[D1_OUT_TENSOR:.*]] = tensor_from_elements %[[D1_OUT]] : tensor<1xindex> // CHECK: %[[D1_OUT_TENSOR:.*]] = tensor.from_elements %[[D1_OUT]] : tensor<1xindex>
// CHECK: %[[D1_OUT_MREF:.*]] = refback.alloc_memref %[[D1_OUT_TENSOR]] : memref<?xf32> // CHECK: %[[D1_OUT_MREF:.*]] = refback.alloc_memref %[[D1_OUT_TENSOR]] : memref<?xf32>
// CHECK: %[[C1:.*]] = constant 1 : index // CHECK: %[[C1:.*]] = constant 1 : index
// CHECK: %[[C0_1:.*]] = constant 0 : index // CHECK: %[[C0_1:.*]] = constant 0 : index