File tree Expand file tree Collapse file tree 1 file changed +3
-3
lines changed
models/turbine_models/custom_models/torchbench Expand file tree Collapse file tree 1 file changed +3
-3
lines changed Original file line number Diff line number Diff line change @@ -61,10 +61,10 @@ cd ..
6161python ./export.py --target=gfx942 --device=rocm --compile_to=vmfb --performance --inference --precision=fp16 --float16 --external_weights=safetensors --external_weights_dir=./torchbench_weights/
6262```
6363
64- ### Example of manual benchmark using export and IREE runtime CLI (hf_Albert )
64+ ### Example of manual benchmark using export and IREE runtime CLI (mobilenet_v3_large )
6565
6666``` shell
67- python ./export.py --target=gfx942 --device=rocm --compile_to=vmfb --performance --inference --precision=fp16 --float16 --external_weights=safetensors --external_weights_dir=./torchbench_weights/ --model_id=hf_Albert
67+ python ./export.py --target=gfx942 --device=rocm --compile_to=vmfb --performance --inference --precision=fp16 --float16 --external_weights=safetensors --external_weights_dir=./torchbench_weights/ --model_id=mobilenet_v3_large
6868
69- iree-benchmark-module --module=generated/hf_Albert_32_fp16_gfx942 .vmfb --input=@generated/hf_Albert_input0 .npy --parameters=model=./torchbench_weights/hf_Albert_fp16 .irpa --device=hip://0 --device_allocator=caching --function=main --benchmark_repetitions=10
69+ iree-benchmark-module --module=generated/mobilenet_v3_large_256_fp16_gfx942 .vmfb --input=@generated/mobilenet_v3_large_input0 .npy --parameters=model=./torchbench_weights/mobilenet_v3_large_fp16 .irpa --device=hip://0 --device_allocator=caching --function=main --benchmark_repetitions=10
7070```
You can’t perform that action at this time.
0 commit comments