[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