[all-commits] [llvm/llvm-project] 9f596a: [mlir][sparse] implement simple codegen for insert...

Aart Bik via All-commits all-commits at lists.llvm.org
Mon Oct 17 18:02:23 PDT 2022


  Branch: refs/heads/main
  Home:   https://github.com/llvm/llvm-project
  Commit: 9f596a7c67cc10fbb624f8b55bf9637e1dddd1c2
      https://github.com/llvm/llvm-project/commit/9f596a7c67cc10fbb624f8b55bf9637e1dddd1c2
  Author: Aart Bik <ajcbik at google.com>
  Date:   2022-10-17 (Mon, 17 Oct 2022)

  Changed paths:
    M mlir/include/mlir/Dialect/SparseTensor/IR/SparseTensorOps.td
    M mlir/lib/Dialect/SparseTensor/Transforms/SparseTensorCodegen.cpp
    M mlir/lib/Dialect/SparseTensor/Transforms/SparseTensorConversion.cpp
    M mlir/test/Dialect/SparseTensor/codegen.mlir
    M mlir/test/Dialect/SparseTensor/conversion.mlir
    M mlir/test/Dialect/SparseTensor/roundtrip.mlir

  Log Message:
  -----------
  [mlir][sparse] implement simple codegen for insertion (and related ops)

This is a proof of concept insertion implementation that sets up
the basic framework and implements it with push backs for just
sparse vectors. It adds insertion/compression through SSA values,
so that we properly update the memref after after pushback operation.

Note that properly using SSA values in sparsification is still TBD
but I will wait until Peiming's loop emitter is in to avoid conflicts.

Reviewed By: wrengr

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




More information about the All-commits mailing list