@@ -1584,24 +1584,24 @@ PY
15841584 sh ' ls -l /dev/kfd'
15851585 sh ' ls -l /dev/dri'
15861586 // Run MLIR vs MIOpen perf benchmarks.
1587- sh """ python3 ./bin/perfRunner.py --op=conv --batch_all \
1588- --configs_file =${ convToUse} \
1589- --tuning_db =${ WORKSPACE} /build/mlir_tuning_${ CHIP} .tsv \
1590- --quick_tuning_db =${ WORKSPACE} /build/mlir_quick_tuning_${ CHIP} .tsv"""
1587+ sh """ python3 ./bin/perfRunner.py --op=conv --batch-all \
1588+ --configs-file =${ convToUse} \
1589+ --tuning-db =${ WORKSPACE} /build/mlir_tuning_${ CHIP} .tsv \
1590+ --quick-tuning-db =${ WORKSPACE} /build/mlir_quick_tuning_${ CHIP} .tsv"""
15911591 // Run MLIR vs hipBLASLt perf benchmarks
1592- sh """ python3 ./bin/perfRunner.py --op=gemm --batch_all \
1593- --configs_file =${ gemmToUse} \
1594- --tuning_db =${ WORKSPACE} /build/mlir_tuning_${ CHIP} .tsv \
1595- --quick_tuning_db =${ WORKSPACE} /build/mlir_quick_tuning_${ CHIP} .tsv"""
1592+ sh """ python3 ./bin/perfRunner.py --op=gemm --batch-all \
1593+ --configs-file =${ gemmToUse} \
1594+ --tuning-db =${ WORKSPACE} /build/mlir_tuning_${ CHIP} .tsv \
1595+ --quick-tuning-db =${ WORKSPACE} /build/mlir_quick_tuning_${ CHIP} .tsv"""
15961596 }
15971597 }
15981598
15991599 stage(" Test Fusion" ) {
16001600 dir(' build' ) {
16011601 // Run fusion resnet50 perf benchmarks
1602- sh """ python3 ./bin/perfRunner.py --op=fusion --test_dir =${ WORKSPACE} /mlir/test/fusion/resnet50-e2e/ --tuning_db =${ WORKSPACE} /build/tuning_fusion_${ CHIP} .tsv"""
1602+ sh """ python3 ./bin/perfRunner.py --op=fusion --test-dir =${ WORKSPACE} /mlir/test/fusion/resnet50-e2e/ --tuning-db =${ WORKSPACE} /build/tuning_fusion_${ CHIP} .tsv"""
16031603 // Run bert perf benchmarks
1604- sh """ python3 ./bin/perfRunner.py --op fusion --test_dir =${ WORKSPACE} /mlir/test/xmir/bert-torch-tosa-e2e/ --tuning_db =${ WORKSPACE} /build/tuning_fusion_${ CHIP} .tsv"""
1604+ sh """ python3 ./bin/perfRunner.py --op fusion --test-dir =${ WORKSPACE} /mlir/test/xmir/bert-torch-tosa-e2e/ --tuning-db =${ WORKSPACE} /build/tuning_fusion_${ CHIP} .tsv"""
16051605 }
16061606 }
16071607
16181618 }
16191619 // Run attention benchmarks
16201620 sh """ python3 ./bin/perfRunner.py --op=attention -b \
1621- --configs_file =${ attnToUse} \
1622- --tuning_db =${ WORKSPACE} /build/mlir_tuning_${ CHIP} .tsv"""
1621+ --configs-file =${ attnToUse} \
1622+ --tuning-db =${ WORKSPACE} /build/mlir_tuning_${ CHIP} .tsv"""
16231623 }
16241624 }
16251625 }
16561656 splitConfigFile(gemmInput, gemmToUse, runIndex)
16571657 }
16581658 }
1659- sh """ python3 ./bin/perfRunner.py --op=gemm --batch_all \
1660- --configs_file =${ gemmToUse} \
1661- --tuning_db =${ WORKSPACE} /build/mlir_tuning_${ CHIP} .tsv --data-type f32 f16 i8_i8 --external-gemm-library CK"""
1659+ sh """ python3 ./bin/perfRunner.py --op=gemm --batch-all \
1660+ --configs-file =${ gemmToUse} \
1661+ --tuning-db =${ WORKSPACE} /build/mlir_tuning_${ CHIP} .tsv --data-type f32 f16 i8_i8 --external-gemm-library CK"""
16621662 def ckChip = get_gpu_architecture()
16631663 sh " python3 ./bin/createPerformanceReports.py ${ ckChip} CK"
16641664 }
0 commit comments