from shark.shark_inference import SharkInference
from shark.shark_downloader import download_model


mlir_model, func_name, inputs, golden_out = download_model(
    "v_diffusion", frontend="torch"
)

shark_module = SharkInference(
    mlir_model, device="vulkan", mlir_dialect="linalg"
)
shark_module.compile()
result = shark_module.forward(inputs)
print("The obtained result via shark is: ", result)
print("The golden result is:", golden_out)