MoveNet singlepose-lightning yield weird result on with Vulkan backend on IREE.

[RechieKho]

What happened?

Hello, I'm the author of IREE.gd, a port of IREE to Godot game engine. Recently, I tried to use a pose detection model, specifically singlepose-lightning-int8 from MoveNet. It works like a charm on Metal backend but it is a total disaster on Vulkan backend.

This image is generated on Metal backend running on MacOS

This image is generated on Vulkan backend running on Windows.

The red dots are visual representation of the output coordinate resulted from running the model in IREE.

iree-gd/iree.gd#96

Steps to reproduce your issue

No response

What component(s) does this issue relate to?

Runtime

Version information

I'm using this git commit of IREE bb542eee65fa0a498963df1f2ee2f205a3dd8bd0.

IREE (https://iree.dev):
  IREE compiler version 20241031.1064 @ 2ec9017bb1de7d5ba552c1f016326088864b9cf9
  LLVM version 20.0.0git
  Optimized build

Additional context

No response

[RechieKho]

This is the TFLite, MLIR, VMFB and dump from iree-dump-module.

singlepose_lightning_int8.zip

[ScottTodd]

Aside: I still want us to add coverage for TFLite models to iree-test-suites: iree-org/iree-test-suites#5. We should have a good process for adding test cases and validating them across configurations (CPU/Vulkan/Metal/etc.) when reports like this come in.

[fire]

If it helps the singlepose-thunder works. https://www.youtube.com/watch?v=FhTPeNDZlVI