forked from OSchip/llvm-project
[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:
parent
7dbfc4fc06
commit
67564841cb
|
@ -37,7 +37,9 @@ module {
|
|||
//
|
||||
// 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>,
|
||||
tensor<1024xf32, #SparseVector>,
|
||||
tensor<f32>) -> tensor<f32>
|
||||
|
|
Loading…
Reference in New Issue