[all-commits] [llvm/llvm-project] ee42e2: [mlir][sparse][gpu] first implementation of the GP...

[Aart Bik via All-commits]

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

  Changed paths:
    M mlir/include/mlir/Dialect/SparseTensor/Pipelines/Passes.h
    M mlir/include/mlir/Dialect/SparseTensor/Transforms/Passes.h
    M mlir/include/mlir/Dialect/SparseTensor/Transforms/Passes.td
    M mlir/lib/Dialect/SparseTensor/Transforms/SparseGPUCodegen.cpp
    M mlir/lib/Dialect/SparseTensor/Transforms/SparseTensorPasses.cpp
    M mlir/lib/Dialect/SparseTensor/Transforms/SparsificationAndBufferizationPass.cpp
    A mlir/test/Dialect/SparseTensor/GPU/gpu_matvec_lib.mlir

  Log Message:
  -----------
  [mlir][sparse][gpu] first implementation of the GPU libgen approach

The sparse compiler now has two prototype strategies for GPU acceleration:

* CUDA codegen: this converts sparsified code to CUDA threads
* CUDA libgen: this converts pre-sparsified code to cuSPARSE library calls

This revision introduces the first steps required for the second approach.

Reviewed By: ThomasRaoux

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