[all-commits] [llvm/llvm-project] a8e1f8: [mlir][sparse][gpu] derive type of cuSparse op

[Aart Bik via All-commits]

  Branch: refs/heads/main
  Home:   https://github.com/llvm/llvm-project
  Commit: a8e1f80f8b68736af15708f04681e7830058e1a9
      https://github.com/llvm/llvm-project/commit/a8e1f80f8b68736af15708f04681e7830058e1a9
  Author: Aart Bik <ajcbik at google.com>
  Date:   2023-05-19 (Fri, 19 May 2023)

  Changed paths:
    M mlir/lib/Conversion/GPUCommon/GPUToLLVMConversion.cpp
    M mlir/lib/ExecutionEngine/CudaRuntimeWrappers.cpp

  Log Message:
  -----------
  [mlir][sparse][gpu] derive type of cuSparse op

This no longer assumes just F64 output.

Note, however, that it will be cleaner to carry the data type in the corresponding operation (rather than tracking operands). That will also allow for mixed type cases, where operands and result type are different

This will be done in a follow revision where the result type is carried by the SpMV/SpMM op itself (and friends).

Reviewed By: Peiming

Differential Revision: https://reviews.llvm.org/D151005