Skip to content

Commit 6e3c0c9

Browse files
committed
totensor tomemref assembly format in catalyst conversion lit test
1 parent b10f0e8 commit 6e3c0c9

File tree

1 file changed

+3
-3
lines changed

1 file changed

+3
-3
lines changed

mlir/test/Catalyst/ConversionTest.mlir

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -160,13 +160,13 @@ module @test1 {
160160

161161
// call @callback_1([[ptr0]], [[ptr1]])
162162

163-
%0 = bufferization.to_memref %arg0 : memref<f64>
163+
%0 = bufferization.to_memref %arg0 : tensor<f64> to memref<f64>
164164
%1 = bufferization.alloc_tensor() {memory_space = 0 : i64} : tensor<f64>
165-
%2 = bufferization.to_memref %1 : memref<f64>
165+
%2 = bufferization.to_memref %1 : tensor<f64> to memref<f64>
166166

167167

168168
catalyst.callback_call @callback_1(%0, %2) : (memref<f64>, memref<f64>) -> ()
169-
%3 = bufferization.to_tensor %2 : memref<f64>
169+
%3 = bufferization.to_tensor %2 : memref<f64> to tensor<f64>
170170
return %3 : tensor<f64>
171171
}
172172
}

0 commit comments

Comments
 (0)