Skip to content

Commit

Permalink
try sym link instead
Browse files Browse the repository at this point in the history
  • Loading branch information
@saienduri
saienduri committed Apr 16, 2024
1 parent f2dc94b commit 4bac11f
Show file tree
Hide file tree
Showing 10 changed files with 127 additions and 125 deletions.
2 changes: 2 additions & 0 deletions iree_tests/download_remote_files.py
View file
Original file line number Diff line number Diff line change
Expand Up @@ -98,6 +98,8 @@ def download_azure_remote_file(test_dir: Path, remote_file: str):
with open(local_file_path, mode="wb") as local_blob:
download_stream = blob_client.download_blob(max_concurrency=4)
local_blob.write(download_stream.readall())
if cache_location != REPO_ROOT:
os.symlink(local_file_path, test_dir / remote_file_name)


def download_generic_remote_file(test_dir: Path, remote_file: str):
Expand Down
134 changes: 67 additions & 67 deletions iree_tests/future-pytorch-models/llama-tank/real_weights_data_flags.txt
View file
Original file line number Diff line number Diff line change
@@ -1,67 +1,67 @@
--parameters=${IREE_TEST_FILES}/iree_tests/pytorch/models/llama-tank/real_weights.irpa
--input=1x93xi64=@${IREE_TEST_FILES}/iree_tests/pytorch/models/llama-tank/inference_input.0.bin
--expected_output=1x93x32000xf32=@${IREE_TEST_FILES}/iree_tests/pytorch/models/llama-tank/inference_output.0.bin
--expected_output=1x32x93x128xf32=@${IREE_TEST_FILES}/iree_tests/pytorch/models/llama-tank/inference_output.1.bin
--expected_output=1x32x93x128xf32=@${IREE_TEST_FILES}/iree_tests/pytorch/models/llama-tank/inference_output.2.bin
--expected_output=1x32x93x128xf32=@${IREE_TEST_FILES}/iree_tests/pytorch/models/llama-tank/inference_output.3.bin
--expected_output=1x32x93x128xf32=@${IREE_TEST_FILES}/iree_tests/pytorch/models/llama-tank/inference_output.4.bin
--expected_output=1x32x93x128xf32=@${IREE_TEST_FILES}/iree_tests/pytorch/models/llama-tank/inference_output.5.bin
--expected_output=1x32x93x128xf32=@${IREE_TEST_FILES}/iree_tests/pytorch/models/llama-tank/inference_output.6.bin
--expected_output=1x32x93x128xf32=@${IREE_TEST_FILES}/iree_tests/pytorch/models/llama-tank/inference_output.7.bin
--expected_output=1x32x93x128xf32=@${IREE_TEST_FILES}/iree_tests/pytorch/models/llama-tank/inference_output.8.bin
--expected_output=1x32x93x128xf32=@${IREE_TEST_FILES}/iree_tests/pytorch/models/llama-tank/inference_output.9.bin
--expected_output=1x32x93x128xf32=@${IREE_TEST_FILES}/iree_tests/pytorch/models/llama-tank/inference_output.10.bin
--expected_output=1x32x93x128xf32=@${IREE_TEST_FILES}/iree_tests/pytorch/models/llama-tank/inference_output.11.bin
--expected_output=1x32x93x128xf32=@${IREE_TEST_FILES}/iree_tests/pytorch/models/llama-tank/inference_output.12.bin
--expected_output=1x32x93x128xf32=@${IREE_TEST_FILES}/iree_tests/pytorch/models/llama-tank/inference_output.13.bin
--expected_output=1x32x93x128xf32=@${IREE_TEST_FILES}/iree_tests/pytorch/models/llama-tank/inference_output.14.bin
--expected_output=1x32x93x128xf32=@${IREE_TEST_FILES}/iree_tests/pytorch/models/llama-tank/inference_output.15.bin
--expected_output=1x32x93x128xf32=@${IREE_TEST_FILES}/iree_tests/pytorch/models/llama-tank/inference_output.16.bin
--expected_output=1x32x93x128xf32=@${IREE_TEST_FILES}/iree_tests/pytorch/models/llama-tank/inference_output.17.bin
--expected_output=1x32x93x128xf32=@${IREE_TEST_FILES}/iree_tests/pytorch/models/llama-tank/inference_output.18.bin
--expected_output=1x32x93x128xf32=@${IREE_TEST_FILES}/iree_tests/pytorch/models/llama-tank/inference_output.19.bin
--expected_output=1x32x93x128xf32=@${IREE_TEST_FILES}/iree_tests/pytorch/models/llama-tank/inference_output.20.bin
--expected_output=1x32x93x128xf32=@${IREE_TEST_FILES}/iree_tests/pytorch/models/llama-tank/inference_output.21.bin
--expected_output=1x32x93x128xf32=@${IREE_TEST_FILES}/iree_tests/pytorch/models/llama-tank/inference_output.22.bin
--expected_output=1x32x93x128xf32=@${IREE_TEST_FILES}/iree_tests/pytorch/models/llama-tank/inference_output.23.bin
--expected_output=1x32x93x128xf32=@${IREE_TEST_FILES}/iree_tests/pytorch/models/llama-tank/inference_output.24.bin
--expected_output=1x32x93x128xf32=@${IREE_TEST_FILES}/iree_tests/pytorch/models/llama-tank/inference_output.25.bin
--expected_output=1x32x93x128xf32=@${IREE_TEST_FILES}/iree_tests/pytorch/models/llama-tank/inference_output.26.bin
--expected_output=1x32x93x128xf32=@${IREE_TEST_FILES}/iree_tests/pytorch/models/llama-tank/inference_output.27.bin
--expected_output=1x32x93x128xf32=@${IREE_TEST_FILES}/iree_tests/pytorch/models/llama-tank/inference_output.28.bin
--expected_output=1x32x93x128xf32=@${IREE_TEST_FILES}/iree_tests/pytorch/models/llama-tank/inference_output.29.bin
--expected_output=1x32x93x128xf32=@${IREE_TEST_FILES}/iree_tests/pytorch/models/llama-tank/inference_output.30.bin
--expected_output=1x32x93x128xf32=@${IREE_TEST_FILES}/iree_tests/pytorch/models/llama-tank/inference_output.31.bin
--expected_output=1x32x93x128xf32=@${IREE_TEST_FILES}/iree_tests/pytorch/models/llama-tank/inference_output.32.bin
--expected_output=1x32x93x128xf32=@${IREE_TEST_FILES}/iree_tests/pytorch/models/llama-tank/inference_output.33.bin
--expected_output=1x32x93x128xf32=@${IREE_TEST_FILES}/iree_tests/pytorch/models/llama-tank/inference_output.34.bin
--expected_output=1x32x93x128xf32=@${IREE_TEST_FILES}/iree_tests/pytorch/models/llama-tank/inference_output.35.bin
--expected_output=1x32x93x128xf32=@${IREE_TEST_FILES}/iree_tests/pytorch/models/llama-tank/inference_output.36.bin
--expected_output=1x32x93x128xf32=@${IREE_TEST_FILES}/iree_tests/pytorch/models/llama-tank/inference_output.37.bin
--expected_output=1x32x93x128xf32=@${IREE_TEST_FILES}/iree_tests/pytorch/models/llama-tank/inference_output.38.bin
--expected_output=1x32x93x128xf32=@${IREE_TEST_FILES}/iree_tests/pytorch/models/llama-tank/inference_output.39.bin
--expected_output=1x32x93x128xf32=@${IREE_TEST_FILES}/iree_tests/pytorch/models/llama-tank/inference_output.40.bin
--expected_output=1x32x93x128xf32=@${IREE_TEST_FILES}/iree_tests/pytorch/models/llama-tank/inference_output.41.bin
--expected_output=1x32x93x128xf32=@${IREE_TEST_FILES}/iree_tests/pytorch/models/llama-tank/inference_output.42.bin
--expected_output=1x32x93x128xf32=@${IREE_TEST_FILES}/iree_tests/pytorch/models/llama-tank/inference_output.43.bin
--expected_output=1x32x93x128xf32=@${IREE_TEST_FILES}/iree_tests/pytorch/models/llama-tank/inference_output.44.bin
--expected_output=1x32x93x128xf32=@${IREE_TEST_FILES}/iree_tests/pytorch/models/llama-tank/inference_output.45.bin
--expected_output=1x32x93x128xf32=@${IREE_TEST_FILES}/iree_tests/pytorch/models/llama-tank/inference_output.46.bin
--expected_output=1x32x93x128xf32=@${IREE_TEST_FILES}/iree_tests/pytorch/models/llama-tank/inference_output.47.bin
--expected_output=1x32x93x128xf32=@${IREE_TEST_FILES}/iree_tests/pytorch/models/llama-tank/inference_output.48.bin
--expected_output=1x32x93x128xf32=@${IREE_TEST_FILES}/iree_tests/pytorch/models/llama-tank/inference_output.49.bin
--expected_output=1x32x93x128xf32=@${IREE_TEST_FILES}/iree_tests/pytorch/models/llama-tank/inference_output.50.bin
--expected_output=1x32x93x128xf32=@${IREE_TEST_FILES}/iree_tests/pytorch/models/llama-tank/inference_output.51.bin
--expected_output=1x32x93x128xf32=@${IREE_TEST_FILES}/iree_tests/pytorch/models/llama-tank/inference_output.52.bin
--expected_output=1x32x93x128xf32=@${IREE_TEST_FILES}/iree_tests/pytorch/models/llama-tank/inference_output.53.bin
--expected_output=1x32x93x128xf32=@${IREE_TEST_FILES}/iree_tests/pytorch/models/llama-tank/inference_output.54.bin
--expected_output=1x32x93x128xf32=@${IREE_TEST_FILES}/iree_tests/pytorch/models/llama-tank/inference_output.55.bin
--expected_output=1x32x93x128xf32=@${IREE_TEST_FILES}/iree_tests/pytorch/models/llama-tank/inference_output.56.bin
--expected_output=1x32x93x128xf32=@${IREE_TEST_FILES}/iree_tests/pytorch/models/llama-tank/inference_output.57.bin
--expected_output=1x32x93x128xf32=@${IREE_TEST_FILES}/iree_tests/pytorch/models/llama-tank/inference_output.58.bin
--expected_output=1x32x93x128xf32=@${IREE_TEST_FILES}/iree_tests/pytorch/models/llama-tank/inference_output.59.bin
--expected_output=1x32x93x128xf32=@${IREE_TEST_FILES}/iree_tests/pytorch/models/llama-tank/inference_output.60.bin
--expected_output=1x32x93x128xf32=@${IREE_TEST_FILES}/iree_tests/pytorch/models/llama-tank/inference_output.61.bin
--expected_output=1x32x93x128xf32=@${IREE_TEST_FILES}/iree_tests/pytorch/models/llama-tank/inference_output.62.bin
--expected_output=1x32x93x128xf32=@${IREE_TEST_FILES}/iree_tests/pytorch/models/llama-tank/inference_output.63.bin
--expected_output=1x32x93x128xf32=@${IREE_TEST_FILES}/iree_tests/pytorch/models/llama-tank/inference_output.64.bin
--parameters=real_weights.irpa
--input=1x93xi64=@inference_input.0.bin
--expected_output=1x93x32000xf32=@inference_output.0.bin
--expected_output=1x32x93x128xf32=@inference_output.1.bin
--expected_output=1x32x93x128xf32=@inference_output.2.bin
--expected_output=1x32x93x128xf32=@inference_output.3.bin
--expected_output=1x32x93x128xf32=@inference_output.4.bin
--expected_output=1x32x93x128xf32=@inference_output.5.bin
--expected_output=1x32x93x128xf32=@inference_output.6.bin
--expected_output=1x32x93x128xf32=@inference_output.7.bin
--expected_output=1x32x93x128xf32=@inference_output.8.bin
--expected_output=1x32x93x128xf32=@inference_output.9.bin
--expected_output=1x32x93x128xf32=@inference_output.10.bin
--expected_output=1x32x93x128xf32=@inference_output.11.bin
--expected_output=1x32x93x128xf32=@inference_output.12.bin
--expected_output=1x32x93x128xf32=@inference_output.13.bin
--expected_output=1x32x93x128xf32=@inference_output.14.bin
--expected_output=1x32x93x128xf32=@inference_output.15.bin
--expected_output=1x32x93x128xf32=@inference_output.16.bin
--expected_output=1x32x93x128xf32=@inference_output.17.bin
--expected_output=1x32x93x128xf32=@inference_output.18.bin
--expected_output=1x32x93x128xf32=@inference_output.19.bin
--expected_output=1x32x93x128xf32=@inference_output.20.bin
--expected_output=1x32x93x128xf32=@inference_output.21.bin
--expected_output=1x32x93x128xf32=@inference_output.22.bin
--expected_output=1x32x93x128xf32=@inference_output.23.bin
--expected_output=1x32x93x128xf32=@inference_output.24.bin
--expected_output=1x32x93x128xf32=@inference_output.25.bin
--expected_output=1x32x93x128xf32=@inference_output.26.bin
--expected_output=1x32x93x128xf32=@inference_output.27.bin
--expected_output=1x32x93x128xf32=@inference_output.28.bin
--expected_output=1x32x93x128xf32=@inference_output.29.bin
--expected_output=1x32x93x128xf32=@inference_output.30.bin
--expected_output=1x32x93x128xf32=@inference_output.31.bin
--expected_output=1x32x93x128xf32=@inference_output.32.bin
--expected_output=1x32x93x128xf32=@inference_output.33.bin
--expected_output=1x32x93x128xf32=@inference_output.34.bin
--expected_output=1x32x93x128xf32=@inference_output.35.bin
--expected_output=1x32x93x128xf32=@inference_output.36.bin
--expected_output=1x32x93x128xf32=@inference_output.37.bin
--expected_output=1x32x93x128xf32=@inference_output.38.bin
--expected_output=1x32x93x128xf32=@inference_output.39.bin
--expected_output=1x32x93x128xf32=@inference_output.40.bin
--expected_output=1x32x93x128xf32=@inference_output.41.bin
--expected_output=1x32x93x128xf32=@inference_output.42.bin
--expected_output=1x32x93x128xf32=@inference_output.43.bin
--expected_output=1x32x93x128xf32=@inference_output.44.bin
--expected_output=1x32x93x128xf32=@inference_output.45.bin
--expected_output=1x32x93x128xf32=@inference_output.46.bin
--expected_output=1x32x93x128xf32=@inference_output.47.bin
--expected_output=1x32x93x128xf32=@inference_output.48.bin
--expected_output=1x32x93x128xf32=@inference_output.49.bin
--expected_output=1x32x93x128xf32=@inference_output.50.bin
--expected_output=1x32x93x128xf32=@inference_output.51.bin
--expected_output=1x32x93x128xf32=@inference_output.52.bin
--expected_output=1x32x93x128xf32=@inference_output.53.bin
--expected_output=1x32x93x128xf32=@inference_output.54.bin
--expected_output=1x32x93x128xf32=@inference_output.55.bin
--expected_output=1x32x93x128xf32=@inference_output.56.bin
--expected_output=1x32x93x128xf32=@inference_output.57.bin
--expected_output=1x32x93x128xf32=@inference_output.58.bin
--expected_output=1x32x93x128xf32=@inference_output.59.bin
--expected_output=1x32x93x128xf32=@inference_output.60.bin
--expected_output=1x32x93x128xf32=@inference_output.61.bin
--expected_output=1x32x93x128xf32=@inference_output.62.bin
--expected_output=1x32x93x128xf32=@inference_output.63.bin
--expected_output=1x32x93x128xf32=@inference_output.64.bin
8 changes: 4 additions & 4 deletions iree_tests/future-pytorch-models/sd-clip-tank/real_weights_data_flags.txt
View file
Original file line number Diff line number Diff line change
@@ -1,4 +1,4 @@
--parameters=model=${IREE_TEST_FILES}/iree_tests/pytorch/models/sd-clip-tank/real_weights.irpa
--input=1x77xi64=@${IREE_TEST_FILES}/iree_tests/pytorch/models/sd-clip-tank/inference_input.0.bin
--expected_output=1x77x768xf32=@${IREE_TEST_FILES}/iree_tests/pytorch/models/sd-clip-tank/inference_output.0.bin
--expected_output=1x768xf32=@${IREE_TEST_FILES}/iree_tests/pytorch/models/sd-clip-tank/inference_output.1.bin
--parameters=model=real_weights.irpa
--input=1x77xi64=@inference_input.0.bin
--expected_output=1x77x768xf32=@inference_output.0.bin
--expected_output=1x768xf32=@inference_output.1.bin
10 changes: 5 additions & 5 deletions iree_tests/future-pytorch-models/sd-unet-tank/real_weights_data_flags.txt
View file
Original file line number Diff line number Diff line change
@@ -1,5 +1,5 @@
--parameters=model=${IREE_TEST_FILES}/iree_tests/pytorch/models/sd-unet-tank/real_weights.irpa
--input=1x4x64x64xf32=@${IREE_TEST_FILES}/iree_tests/pytorch/models/sd-unet-tank/inference_input.0.bin
--input=1xf32=@${IREE_TEST_FILES}/iree_tests/pytorch/models/sd-unet-tank/inference_input.1.bin
--input=2x77x768xf32=@${IREE_TEST_FILES}/iree_tests/pytorch/models/sd-unet-tank/inference_input.2.bin
--expected_output=1x4x64x64xf32=@${IREE_TEST_FILES}/iree_tests/pytorch/models/sd-unet-tank/inference_output.0.bin
--parameters=model=real_weights.irpa
--input=1x4x64x64xf32=@inference_input.0.bin
--input=1xf32=@inference_input.1.bin
--input=2x77x768xf32=@inference_input.2.bin
--expected_output=1x4x64x64xf32=@inference_output.0.bin
6 changes: 3 additions & 3 deletions iree_tests/future-pytorch-models/sd-vae-decode-tank/real_weights_data_flags.txt
View file
Original file line number Diff line number Diff line change
@@ -1,3 +1,3 @@
--parameters=model=${IREE_TEST_FILES}/iree_tests/pytorch/models/sd-vae-decode-tank/real_weights.irpa
--input=1x4x64x64xf32=@${IREE_TEST_FILES}/iree_tests/pytorch/models/sd-vae-decode-tank/inference_input.0.bin
--expected_output=1x3x512x512xf32=@${IREE_TEST_FILES}/iree_tests/pytorch/models/sd-vae-decode-tank/inference_output.0.bin
--parameters=model=real_weights.irpa
--input=1x4x64x64xf32=@inference_input.0.bin
--expected_output=1x3x512x512xf32=@inference_output.0.bin
54 changes: 27 additions & 27 deletions iree_tests/pytorch/models/opt-125M/real_weights_data_flags.txt
View file
Original file line number Diff line number Diff line change
@@ -1,27 +1,27 @@
--parameters=${IREE_TEST_FILES}/iree_tests/pytorch/models/opt-125M/real_weights.irpa
--input=1x8xi64=@${IREE_TEST_FILES}/iree_tests/pytorch/models/opt-125M/inference_input.0.bin
--expected_output=1x8x50272xf32=@${IREE_TEST_FILES}/iree_tests/pytorch/models/opt-125M/inference_output.0.bin
--expected_output=1x12x8x64xf32=@${IREE_TEST_FILES}/iree_tests/pytorch/models/opt-125M/inference_output.1.bin
--expected_output=1x12x8x64xf32=@${IREE_TEST_FILES}/iree_tests/pytorch/models/opt-125M/inference_output.2.bin
--expected_output=1x12x8x64xf32=@${IREE_TEST_FILES}/iree_tests/pytorch/models/opt-125M/inference_output.3.bin
--expected_output=1x12x8x64xf32=@${IREE_TEST_FILES}/iree_tests/pytorch/models/opt-125M/inference_output.4.bin
--expected_output=1x12x8x64xf32=@${IREE_TEST_FILES}/iree_tests/pytorch/models/opt-125M/inference_output.5.bin
--expected_output=1x12x8x64xf32=@${IREE_TEST_FILES}/iree_tests/pytorch/models/opt-125M/inference_output.6.bin
--expected_output=1x12x8x64xf32=@${IREE_TEST_FILES}/iree_tests/pytorch/models/opt-125M/inference_output.7.bin
--expected_output=1x12x8x64xf32=@${IREE_TEST_FILES}/iree_tests/pytorch/models/opt-125M/inference_output.8.bin
--expected_output=1x12x8x64xf32=@${IREE_TEST_FILES}/iree_tests/pytorch/models/opt-125M/inference_output.9.bin
--expected_output=1x12x8x64xf32=@${IREE_TEST_FILES}/iree_tests/pytorch/models/opt-125M/inference_output.10.bin
--expected_output=1x12x8x64xf32=@${IREE_TEST_FILES}/iree_tests/pytorch/models/opt-125M/inference_output.11.bin
--expected_output=1x12x8x64xf32=@${IREE_TEST_FILES}/iree_tests/pytorch/models/opt-125M/inference_output.12.bin
--expected_output=1x12x8x64xf32=@${IREE_TEST_FILES}/iree_tests/pytorch/models/opt-125M/inference_output.13.bin
--expected_output=1x12x8x64xf32=@${IREE_TEST_FILES}/iree_tests/pytorch/models/opt-125M/inference_output.14.bin
--expected_output=1x12x8x64xf32=@${IREE_TEST_FILES}/iree_tests/pytorch/models/opt-125M/inference_output.15.bin
--expected_output=1x12x8x64xf32=@${IREE_TEST_FILES}/iree_tests/pytorch/models/opt-125M/inference_output.16.bin
--expected_output=1x12x8x64xf32=@${IREE_TEST_FILES}/iree_tests/pytorch/models/opt-125M/inference_output.17.bin
--expected_output=1x12x8x64xf32=@${IREE_TEST_FILES}/iree_tests/pytorch/models/opt-125M/inference_output.18.bin
--expected_output=1x12x8x64xf32=@${IREE_TEST_FILES}/iree_tests/pytorch/models/opt-125M/inference_output.19.bin
--expected_output=1x12x8x64xf32=@${IREE_TEST_FILES}/iree_tests/pytorch/models/opt-125M/inference_output.20.bin
--expected_output=1x12x8x64xf32=@${IREE_TEST_FILES}/iree_tests/pytorch/models/opt-125M/inference_output.21.bin
--expected_output=1x12x8x64xf32=@${IREE_TEST_FILES}/iree_tests/pytorch/models/opt-125M/inference_output.22.bin
--expected_output=1x12x8x64xf32=@${IREE_TEST_FILES}/iree_tests/pytorch/models/opt-125M/inference_output.23.bin
--expected_output=1x12x8x64xf32=@${IREE_TEST_FILES}/iree_tests/pytorch/models/opt-125M/inference_output.24.bin
--parameters=real_weights.irpa
--input=1x8xi64=@inference_input.0.bin
--expected_output=1x8x50272xf32=@inference_output.0.bin
--expected_output=1x12x8x64xf32=@inference_output.1.bin
--expected_output=1x12x8x64xf32=@inference_output.2.bin
--expected_output=1x12x8x64xf32=@inference_output.3.bin
--expected_output=1x12x8x64xf32=@inference_output.4.bin
--expected_output=1x12x8x64xf32=@inference_output.5.bin
--expected_output=1x12x8x64xf32=@inference_output.6.bin
--expected_output=1x12x8x64xf32=@inference_output.7.bin
--expected_output=1x12x8x64xf32=@inference_output.8.bin
--expected_output=1x12x8x64xf32=@inference_output.9.bin
--expected_output=1x12x8x64xf32=@inference_output.10.bin
--expected_output=1x12x8x64xf32=@inference_output.11.bin
--expected_output=1x12x8x64xf32=@inference_output.12.bin
--expected_output=1x12x8x64xf32=@inference_output.13.bin
--expected_output=1x12x8x64xf32=@inference_output.14.bin
--expected_output=1x12x8x64xf32=@inference_output.15.bin
--expected_output=1x12x8x64xf32=@inference_output.16.bin
--expected_output=1x12x8x64xf32=@inference_output.17.bin
--expected_output=1x12x8x64xf32=@inference_output.18.bin
--expected_output=1x12x8x64xf32=@inference_output.19.bin
--expected_output=1x12x8x64xf32=@inference_output.20.bin
--expected_output=1x12x8x64xf32=@inference_output.21.bin
--expected_output=1x12x8x64xf32=@inference_output.22.bin
--expected_output=1x12x8x64xf32=@inference_output.23.bin
--expected_output=1x12x8x64xf32=@inference_output.24.bin
6 changes: 3 additions & 3 deletions iree_tests/pytorch/models/resnet50/real_weights_data_flags.txt
View file
Original file line number Diff line number Diff line change
@@ -1,3 +1,3 @@
--parameters=${IREE_TEST_FILES}/iree_tests/pytorch/models/resnet50/real_weights.irpa
--input=1x3x224x224xf32=@${IREE_TEST_FILES}/iree_tests/pytorch/models/resnet50/inference_input.0.bin
--expected_output=1x1000xf32=@${IREE_TEST_FILES}/iree_tests/pytorch/models/resnet50/inference_output.0.bin
--parameters=real_weights.irpa
--input=1x3x224x224xf32=@inference_input.0.bin
--expected_output=1x1000xf32=@inference_output.0.bin
14 changes: 7 additions & 7 deletions iree_tests/pytorch/models/sdxl-prompt-encoder-tank/real_weights_data_flags.txt
View file
Original file line number Diff line number Diff line change
@@ -1,8 +1,8 @@
--parameters=model=${IREE_TEST_FILES}/iree_tests/pytorch/models/sdxl-prompt-encoder-tank/real_weights.irpa
--input=1x64xi64=@${IREE_TEST_FILES}/iree_tests/pytorch/models/sdxl-prompt-encoder-tank/inference_input.0.bin
--input=1x64xi64=@${IREE_TEST_FILES}/iree_tests/pytorch/models/sdxl-prompt-encoder-tank/inference_input.1.bin
--input=1x64xi64=@${IREE_TEST_FILES}/iree_tests/pytorch/models/sdxl-prompt-encoder-tank/inference_input.2.bin
--input=1x64xi64=@${IREE_TEST_FILES}/iree_tests/pytorch/models/sdxl-prompt-encoder-tank/inference_input.3.bin
--expected_output=2x64x2048xf16=@${IREE_TEST_FILES}/iree_tests/pytorch/models/sdxl-prompt-encoder-tank/inference_output.0.bin
--expected_output=2x1280xf16=@${IREE_TEST_FILES}/iree_tests/pytorch/models/sdxl-prompt-encoder-tank/inference_output.1.bin
--parameters=model=real_weights.irpa
--input=1x64xi64=@inference_input.0.bin
--input=1x64xi64=@inference_input.1.bin
--input=1x64xi64=@inference_input.2.bin
--input=1x64xi64=@inference_input.3.bin
--expected_output=2x64x2048xf16=@inference_output.0.bin
--expected_output=2x1280xf16=@inference_output.1.bin
--expected_f16_threshold=1.0f
Loading

0 comments on commit 4bac11f

Please sign in to comment.