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 ..
61
61
python ./export.py --target=gfx942 --device=rocm --compile_to=vmfb --performance --inference --precision=fp16 --float16 --external_weights=safetensors --external_weights_dir=./torchbench_weights/
62
62
```
63
63
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 )
65
65
66
66
``` 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
68
68
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
70
70
```
You can’t perform that action at this time.
0 commit comments