search

iree-org / iree

A retargetable MLIR-based machine learning compiler and runtime toolkit.
http://iree.dev/
Apache License 2.0
2.47k stars 548 forks source link

Revert "[LinalgExt] Add online_attention op" #17658

Closed ScottTodd closed 1 week ago

ScottTodd commented 1 week ago

Reverts iree-org/iree#17536

This caused sdxl-scheduled-unet-3-tank to hit timeouts when compiling for cpu: https://github.com/iree-org/iree/actions/runs/9484305572/job/26134004282

ScottTodd commented 1 week ago

test_models :: cpu_llvm_task passed, bazel build flaked, and other builds are still pending. Going to merge without waiting for the other checks.

MaheshRavishankar commented 1 week ago

Cc @Groverkss