[all-commits] [llvm/llvm-project] 70e227: [mlir][sparse] recognize ReLu operation during spa...

Aart Bik via All-commits all-commits at lists.llvm.org
Mon May 13 14:02:53 PDT 2024


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

  Changed paths:
    M mlir/include/mlir/Dialect/SparseTensor/Utils/Merger.h
    M mlir/lib/Dialect/SparseTensor/Utils/Merger.cpp
    A mlir/test/Dialect/SparseTensor/sparse_relu.mlir
    M mlir/unittests/Dialect/SparseTensor/MergerTest.cpp

  Log Message:
  -----------
  [mlir][sparse] recognize ReLu operation during sparsification (#92016)

This is a proof of concept recognition of the most basic forms of ReLu
operations, used to show-case sparsification of end-to-end PyTorch
models. In the long run, we must avoid lowering such constructs too
early (with this need for raising them back).

See discussion at

https://discourse.llvm.org/t/min-max-abs-relu-recognition-starter-project/78918



To unsubscribe from these emails, change your notification settings at https://github.com/llvm/llvm-project/settings/notifications


More information about the All-commits mailing list