[mlir][sparse] support Parallel for/reduction.
authorPeiming Liu <peiming@google.com>
Tue, 18 Oct 2022 16:41:03 +0000 (16:41 +0000)
committerPeiming Liu <peiming@google.com>
Fri, 4 Nov 2022 22:47:27 +0000 (22:47 +0000)
commit838389780e56f1a198a94f66ea436359466bf5ed
tree8a5c70c048a38fbcddae241343908fd739b90076
parent03c7cd3a616b731cf041fbf164078b349223c4ef
[mlir][sparse] support Parallel for/reduction.

Reviewed By: aartbik

Differential Revision: https://reviews.llvm.org/D135927
mlir/lib/Dialect/SparseTensor/Transforms/CodegenUtils.cpp
mlir/lib/Dialect/SparseTensor/Transforms/CodegenUtils.h
mlir/lib/Dialect/SparseTensor/Transforms/Sparsification.cpp
mlir/test/Dialect/SparseTensor/sparse_parallel.mlir
mlir/test/Dialect/SparseTensor/sparse_parallel_reduce.mlir [new file with mode: 0644]
mlir/test/Integration/Dialect/SparseTensor/CPU/sparse_matmul.mlir
mlir/test/Integration/Dialect/SparseTensor/CPU/sparse_matvec.mlir