Refactor TensorExp parameters into a union
authorGus Smith <gusss@google.com>
Thu, 1 Jul 2021 21:45:18 +0000 (21:45 +0000)
committerGus Smith <gusss@google.com>
Fri, 2 Jul 2021 14:45:56 +0000 (14:45 +0000)
commit4569c14ac347180d9514f43c45c6f52569ce8f8c
tree4891fc7b799ed47a77bc37c6e7d915dfc217a211
parente42bb5e35a88c6a163934080883d4c5bc1f32cfc
Refactor TensorExp parameters into a union

To make TensorExp clearer, this change refactors the e0/e1 fields into a union: e0/e1 for a binary op tensor expression, and tensor_num for a tensor-kinded tensor expression.

Reviewed By: aartbik

Differential Revision: https://reviews.llvm.org/D105303
mlir/include/mlir/Dialect/SparseTensor/Utils/Merger.h
mlir/lib/Dialect/SparseTensor/Transforms/Sparsification.cpp
mlir/lib/Dialect/SparseTensor/Utils/Merger.cpp