Skip to content

Commit 4d81769

Browse files
Update some additional usages of "bufferization.to_tensor|to_memref" in docs and tests
1 parent 5c457e1 commit 4d81769

File tree

2 files changed

+3
-3
lines changed

2 files changed

+3
-3
lines changed

mlir/docs/Bufferization.md

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -223,8 +223,8 @@ func.func @test_matmul(%A: memref<1x17x19xf32>,
223223
%B: memref<1x19x29xf32>,
224224
%C: memref<1x17x29xf32>) {
225225
226-
%A_tensor = bufferization.to_tensor %A restrict : memref<1x17x19xf32>
227-
%B_tensor = bufferization.to_tensor %B restrict : memref<1x19x29xf32>
226+
%A_tensor = bufferization.to_tensor %A restrict : memref<1x17x19xf32> -> tensor<1x17x19xf32>
227+
%B_tensor = bufferization.to_tensor %B restrict : memref<1x19x29xf32> -> tensor<1x19x29xf32>
228228
229229
%0 = tosa.matmul %A_tensor, %B_tensor
230230
: (tensor<1x17x19xf32>, tensor<1x19x29xf32>) ->

mlir/test/Integration/Dialect/SparseTensor/GPU/CUDA/sparse-matvec-const.mlir

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -47,7 +47,7 @@ module {
4747

4848
// Call the kernel with an vector taken from global memory.
4949
%xbuf = memref.get_global @__constant_64xf64 : memref<64xf64>
50-
%x = bufferization.to_tensor %xbuf restrict : memref<64xf64>
50+
%x = bufferization.to_tensor %xbuf restrict : memref<64xf64> -> tensor<64xf64>
5151
%0 = call @matvec(%A, %x, %y) : (tensor<1024x64xf64, #CSR>, tensor<64xf64>, tensor<1024xf64>) -> tensor<1024xf64>
5252

5353
//

0 commit comments

Comments
 (0)