Skip to content

Commit

Permalink
uniform names
Browse files Browse the repository at this point in the history
  • Loading branch information
@saienduri
saienduri committed Apr 4, 2024
1 parent 402109c commit d6e0582
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion iree_tests/benchmarks/benchmark_sdxl_cpu.sh
View file
Original file line number Diff line number Diff line change
Expand Up @@ -16,6 +16,6 @@ prompt_encoder_dir="$iree_root/pytorch/models/sdxl-prompt-encoder-tank"

echo "Running sdxl benchmark"

iree-benchmark-module --device=local-task --module=$prompt_encoder_dir/model_cpu_llvm_task_real_weights.vmfb --parameters=model="$prompt_encoder_dir/real_weights.irpa" --module="$scheduled_unet_dir/model_cpu_llvm_task_real_weights.vmfb" --parameters=model="$scheduled_unet_dir/real_weights.irpa" --module="$vae_decode_dir/model_cpu_llvm_task_real_weights.vmfb" --parameters=model="$vae_decode_dir/real_weights.irpa" --module="$this_dir/sdxl_full_pipeline_fp16_.vmfb" --function=tokens_to_image --input=1x4x128x128xf16 --input=1xf16 --input=1x64xi64 --input=1x64xi64 --input=1x64xi64 --input=1x64xi64
iree-benchmark-module --device=local-task --module="$prompt_encoder_dir/model_cpu_llvm_task_real_weights.vmfb" --parameters=model="$prompt_encoder_dir/real_weights.irpa" --module="$scheduled_unet_dir/model_cpu_llvm_task_real_weights.vmfb" --parameters=model="$scheduled_unet_dir/real_weights.irpa" --module="$vae_decode_dir/model_cpu_llvm_task_real_weights.vmfb" --parameters=model="$vae_decode_dir/real_weights.irpa" --module="$this_dir/sdxl_full_pipeline_fp16_.vmfb" --function=tokens_to_image --input=1x4x128x128xf16 --input=1xf16 --input=1x64xi64 --input=1x64xi64 --input=1x64xi64 --input=1x64xi64

echo "Succesfully finished sdxl pipeline benchmark"

0 comments on commit d6e0582

Please sign in to comment.