[all-commits] [llvm/llvm-project] c8ae8c: [mlir][sparse][taco] Add support for float32.
Bixia Zheng via All-commits
all-commits at lists.llvm.org
Wed Feb 23 18:24:36 PST 2022
Branch: refs/heads/main
Home: https://github.com/llvm/llvm-project
Commit: c8ae8cfb5d53217f8ee11af666c3d5a51a3e3230
https://github.com/llvm/llvm-project/commit/c8ae8cfb5d53217f8ee11af666c3d5a51a3e3230
Author: Bixia Zheng <bixia at google.com>
Date: 2022-02-23 (Wed, 23 Feb 2022)
Changed paths:
M mlir/test/Integration/Dialect/SparseTensor/taco/test_MTTKRP.py
M mlir/test/Integration/Dialect/SparseTensor/taco/test_SDDMM.py
M mlir/test/Integration/Dialect/SparseTensor/taco/test_SpMV.py
M mlir/test/Integration/Dialect/SparseTensor/taco/test_true_dense_tensor_algebra.py
M mlir/test/Integration/Dialect/SparseTensor/taco/tools/mlir_pytaco.py
M mlir/test/Integration/Dialect/SparseTensor/taco/tools/mlir_pytaco_io.py
M mlir/test/Integration/Dialect/SparseTensor/taco/tools/mlir_pytaco_utils.py
M mlir/test/Integration/Dialect/SparseTensor/taco/unit_test_tensor_utils.py
Log Message:
-----------
[mlir][sparse][taco] Add support for float32.
Previously, we only support float64. We now support float32 and float64. When
constructing a tensor without providing a data type, the default is float32.
Fix the tests to data type consistency. All PyTACO application tests now use
float32 to match the default data type of TACO. Other tests may use float32 or
float64.
Reviewed By: aartbik
Differential Revision: https://reviews.llvm.org/D120356
More information about the All-commits
mailing list