[Testing] Fix bf16->f32 utility to preserve shape of the input

nod-ai / iree-amd-aie

IREE plugin repository for the AMD AIE accelerator

Apache License 2.0

69 stars 30 forks source link

[Testing] Fix bf16->f32 utility to preserve shape of the input #835

Closed Abhishek-Varma closed 1 month ago

Abhishek-Varma commented 1 month ago

-- This commit adds a quick fix for bf16->f32 utility to preserve the shape of the input. -- Without this the utility is linearizing the input which causes issues during comparison.

Signed-off-by: Abhishek Varma abhvarma@amd.com

newling commented 1 month ago

Thanks.