maxjhandsome
/
llvm-project
forked from OSchip/llvm-project
1
0
Fork 0
Code Issues Pull Requests Packages Releases Wiki Activity

[mlir][sparse][bufferization] initialize reduction variable 

After recent bufferization improvement, this test
started failing due to missed zero initialization.

Reviewed By: jpienaar

Differential Revision: https://reviews.llvm.org/D129800
This commit is contained in:
Aart Bik 2022-07-14 13:18:17 -07:00
parent 7dbfc4fc06
commit 67564841cb
1 changed files with 3 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
mlir/test/Integration/Dialect/SparseTensor/CPU/sparse_dot.mlir
View File

@ -37,7 +37,9 @@ module {
// //
// CHECK: 53 // CHECK: 53
// //
%x = bufferization.alloc_tensor() : tensor<f32> %t = bufferization.alloc_tensor() : tensor<f32>
%z = arith.constant 0.0 : f32
%x = tensor.insert %z into %t[] : tensor<f32>
%0 = call @sparse_dot(%s1, %s2, %x) : (tensor<1024xf32, #SparseVector>, %0 = call @sparse_dot(%s1, %s2, %x) : (tensor<1024xf32, #SparseVector>,
tensor<1024xf32, #SparseVector>, tensor<1024xf32, #SparseVector>,
tensor<f32>) -> tensor<f32> tensor<f32>) -> tensor<f32>