[GPU] Add gather fusion tests for vector distribution

iree-org / iree

A retargetable MLIR-based machine learning compiler and runtime toolkit.

http://iree.dev/

Apache License 2.0

2.86k stars 622 forks source link

[GPU] Add gather fusion tests for vector distribution #19209

Open Groverkss opened 6 days ago

Groverkss commented 6 days ago

VectorDistribution now supports gather fusion on producers. This pr adds pipeline tests for that. There are still numerical issues being tracked seperatly,related to distribution of gather.

Groverkss commented 5 days ago

Do we need all the hal.executable boiler plate?

Yes, this is how all the tests in that file are structured. These tests are checking if a dispatch can be codegened. We could add a pass that generates all this boiler plate automatically for pipeline tests, Not for this patch though.