iree-run-module get incorrect result

[FlintWangacc]

I use iree to compile vgg neural network.
I do it in the following step.

1. Transform vgg to mlir.

2. Transform the mlir file to vmfb file through iree-compile

iree-compile --iree-hal-target-backends=rocm --iree-hip-target=gfx1100 vgg.mlir -o vgg.vmfb

3. Prepare npy file for iree-run-module
4. Run with iree-run-module

iree-run-module --device=hip --module=vgg.vmfb --function=main --input="1x3x224x224xf32=@./vgg_input.npy" --output="@./iree_output.npy"

5. load the iree_output.npy

import torch
import numpy as np

output = np.load('iree_output.npy')
tensor = torch.from_numpy(output)

I find the tensor in the above python code is different from the output of the vgg model.

Is there any wrong with my steps?