From ef686a0bd76d45d7c0c5ff59c07a70ba759d47e6 Mon Sep 17 00:00:00 2001 From: Arun Thangamani Date: Wed, 29 Jan 2025 01:54:31 -0800 Subject: [PATCH] avx512* -> avx512f in benchmark json extension --- benchmarks/config/base/base.json | 12 +++---- .../omp/mlir-fp32-vector-to-kernel.json | 32 +++++++++---------- 2 files changed, 22 insertions(+), 22 deletions(-) diff --git a/benchmarks/config/base/base.json b/benchmarks/config/base/base.json index f13b43b8f..a1a4e8163 100644 --- a/benchmarks/config/base/base.json +++ b/benchmarks/config/base/base.json @@ -41,7 +41,7 @@ "benchmark": [ "mlir-gen", "--kernel=const --float-type=f32 --batch=256 --layers=1024,1024,1024,1024 --tiles=32,32,32" ], "environment": {}, "flags": [ "-n", "100", "-run-args='--vector-to-kernels --registerBlocking=8,32 '" ], - "extensions": ["avx512*"] + "extensions": ["avx512f"] }, "gemm_fp32_mlir_vector_avx2": { "type": "IR-GEN", @@ -83,7 +83,7 @@ "benchmark": [ "mlir-gen", "--kernel=const --bias --relu --float-type=f32 --batch=256 --layers=1024,1024,1024,1024 --tiles=32,32,32" ], "environment": {}, "flags": [ "-n", "100", "-run-args='--def-parallel --vector-to-kernels --registerBlocking=8,32 '" ], - "extensions": ["avx512*"] + "extensions": ["avx512f"] }, "mlp_fp32_mlir_vector_avx2": { "type": "IR-GEN", @@ -128,7 +128,7 @@ "benchmark": [ "mlir-gen", "--kernel=const --float-type=f32 --batch=256 --layers=1024,1024,1024,1024" ], "environment": {}, "flags": [ "-n", "100", "-run-args='--vector-to-kernels --registerBlocking=8,32 '" ], - "extensions": [ "avx512*" ] + "extensions": [ "avx512f" ] }, "fp32_3x1024_args_mlir": { "type": "IR-GEN", @@ -142,7 +142,7 @@ "benchmark": [ "mlir-gen", "--kernel=args --float-type=f32 --batch=256 --layers=1024,1024,1024,1024" ], "environment": {}, "flags": [ "-n", "100", "-run-args='--vector-to-kernels --registerBlocking=8,32 '" ], - "extensions": [ "avx512*" ] + "extensions": [ "avx512f" ] }, "bf16_3x1024_const_mlir": { "type": "IR-GEN", @@ -173,7 +173,7 @@ "benchmark": [ "mlir-gen", "--kernel=const --bias --relu --float-type=f32 --batch=256 --layers=1024,1024,1024,1024" ], "environment": {}, "flags": [ "-n", "100", "-run-args='--def-parallel --vector-to-kernels --registerBlocking=8,32 '" ], - "extensions": [ "avx512*" ] + "extensions": [ "avx512f" ] }, "fp32_3x1024_args_mlir": { "type": "IR-GEN", @@ -187,7 +187,7 @@ "benchmark": [ "mlir-gen", "--kernel=args --bias --relu --float-type=f32 --batch=256 --layers=1024,1024,1024,1024" ], "environment": {}, "flags": [ "-n", "100", "-run-args=' --def-parallel --vector-to-kernels --registerBlocking=8,32 '" ], - "extensions": [ "avx512*" ] + "extensions": [ "avx512f" ] }, "bf16_3x1024_const_mlir": { "type": "IR-GEN", diff --git a/benchmarks/config/omp/mlir-fp32-vector-to-kernel.json b/benchmarks/config/omp/mlir-fp32-vector-to-kernel.json index 96d9efa23..e78318fdf 100644 --- a/benchmarks/config/omp/mlir-fp32-vector-to-kernel.json +++ b/benchmarks/config/omp/mlir-fp32-vector-to-kernel.json @@ -6,28 +6,28 @@ "benchmark": [ "mlir-gen", "--kernel=const --float-type=f32 --batch=256 --layers=1024,1024,1024,1024 --tiles=32,32,32" ], "environment": { "OMP_NUM_THREADS": "2", "KMP_AFFINITY": "granularity=fine,verbose,compact,1,0" }, "flags": [ "-n", "100", "-run-args='--def-parallel --parallel-task-grid=8,16 --vector-to-kernels --registerBlocking=8,32 '" ], - "extensions": [ "avx512*" ] + "extensions": [ "avx512f" ] }, "fp32_3x1024_omp_4_mlir": { "type": "IR-GEN", "benchmark": [ "mlir-gen", "--kernel=const --float-type=f32 --batch=256 --layers=1024,1024,1024,1024 --tiles=32,32,32" ], "environment": { "OMP_NUM_THREADS": "4", "KMP_AFFINITY": "granularity=fine,verbose,compact,1,0" }, "flags": [ "-n", "100", "-run-args='--def-parallel --parallel-task-grid=8,8 --vector-to-kernels --registerBlocking=8,32 '" ], - "extensions": [ "avx512*" ] + "extensions": [ "avx512f" ] }, "fp32_3x1024_omp_8_mlir": { "type": "IR-GEN", "benchmark": [ "mlir-gen", "--kernel=const --float-type=f32 --batch=256 --layers=1024,1024,1024,1024 --tiles=32,32,32" ], "environment": { "OMP_NUM_THREADS": "8", "KMP_AFFINITY": "granularity=fine,verbose,compact,1,0" }, "flags": [ "-n", "100", "-run-args='--def-parallel --parallel-task-grid=4,8 --vector-to-kernels --registerBlocking=8,32 '" ], - "extensions": [ "avx512*" ] + "extensions": [ "avx512f" ] }, "fp32_3x1024_omp_16_mlir": { "type": "IR-GEN", "benchmark": [ "mlir-gen", "--kernel=const --float-type=f32 --batch=256 --layers=1024,1024,1024,1024 --tiles=32,32,32" ], "environment": { "OMP_NUM_THREADS": "16", "KMP_AFFINITY": "granularity=fine,verbose,compact,1,0" }, "flags": [ "-n", "100", "-run-args='--def-parallel --parallel-task-grid=2,8 --vector-to-kernels --registerBlocking=8,32 '" ], - "extensions": [ "avx512*" ] + "extensions": [ "avx512f" ] } }}, { @@ -37,28 +37,28 @@ "benchmark": [ "mlir-gen", "--kernel=const --bias --relu --float-type=f32 --batch=256 --layers=1024,1024,1024,1024 --tiles=32,32,32" ], "environment": { "OMP_NUM_THREADS": "2", "KMP_AFFINITY": "granularity=fine,verbose,compact,1,0" }, "flags": [ "-n", "100", "-run-args='--def-parallel --parallel-task-grid=8,16 --vector-to-kernels --registerBlocking=8,32 '" ], - "extensions": [ "avx512*" ] + "extensions": [ "avx512f" ] }, "fp32_3x1024_omp_4_mlir": { "type": "IR-GEN", "benchmark": [ "mlir-gen", "--kernel=const --bias --relu --float-type=f32 --batch=256 --layers=1024,1024,1024,1024 --tiles=32,32,32" ], "environment": { "OMP_NUM_THREADS": "4", "KMP_AFFINITY": "granularity=fine,verbose,compact,1,0" }, "flags": [ "-n", "100", "-run-args='--def-parallel --parallel-task-grid=8,8 --vector-to-kernels --registerBlocking=8,32 '" ], - "extensions": [ "avx512*" ] + "extensions": [ "avx512f" ] }, "fp32_3x1024_omp_8_mlir": { "type": "IR-GEN", "benchmark": [ "mlir-gen", "--kernel=const --bias --relu --float-type=f32 --batch=256 --layers=1024,1024,1024,1024 --tiles=32,32,32" ], "environment": { "OMP_NUM_THREADS": "8", "KMP_AFFINITY": "granularity=fine,verbose,compact,1,0" }, "flags": [ "-n", "100", "-run-args='--def-parallel --parallel-task-grid=4,8 --vector-to-kernels --registerBlocking=8,32 '" ], - "extensions": [ "avx512*" ] + "extensions": [ "avx512f" ] }, "fp32_3x1024_omp_16_mlir": { "type": "IR-GEN", "benchmark": [ "mlir-gen", "--kernel=const --bias --relu --float-type=f32 --batch=256 --layers=1024,1024,1024,1024 --tiles=32,32,32" ], "environment": { "OMP_NUM_THREADS": "16", "KMP_AFFINITY": "granularity=fine,verbose,compact,1,0" }, "flags": [ "-n", "100", "-run-args='--def-parallel --parallel-task-grid=2,8 --vector-to-kernels --registerBlocking=8,32 '" ], - "extensions": [ "avx512*" ] + "extensions": [ "avx512f" ] } }}, { @@ -68,28 +68,28 @@ "benchmark": [ "mlir-gen", "--kernel=const --float-type=f32 --batch=256 --layers=1024,1024,1024,1024 --tiles=64,64,64" ], "environment": { "OMP_NUM_THREADS": "2", "KMP_AFFINITY": "granularity=fine,verbose,compact,1,0" }, "flags": [ "-n", "100", "-run-args='--def-parallel --parallel-task-grid=4,8 --vector-to-kernels --registerBlocking=4,64 '" ], - "extensions": [ "avx512*" ] + "extensions": [ "avx512f" ] }, "fp32_3x1024_omp_4_mlir": { "type": "IR-GEN", "benchmark": [ "mlir-gen", "--kernel=const --float-type=f32 --batch=256 --layers=1024,1024,1024,1024 --tiles=64,64,64" ], "environment": { "OMP_NUM_THREADS": "4", "KMP_AFFINITY": "granularity=fine,verbose,compact,1,0" }, "flags": [ "-n", "100", "-run-args='--def-parallel --parallel-task-grid=4,4 --vector-to-kernels --registerBlocking=4,64 '" ], - "extensions": [ "avx512*" ] + "extensions": [ "avx512f" ] }, "fp32_3x1024_omp_8_mlir": { "type": "IR-GEN", "benchmark": [ "mlir-gen", "--kernel=const --float-type=f32 --batch=256 --layers=1024,1024,1024,1024 --tiles=64,64,64" ], "environment": { "OMP_NUM_THREADS": "8", "KMP_AFFINITY": "granularity=fine,verbose,compact,1,0" }, "flags": [ "-n", "100", "-run-args='--def-parallel --parallel-task-grid=2,4 --vector-to-kernels --registerBlocking=4,64 '" ], - "extensions": [ "avx512*" ] + "extensions": [ "avx512f" ] }, "fp32_3x1024_omp_16_mlir": { "type": "IR-GEN", "benchmark": [ "mlir-gen", "--kernel=const --float-type=f32 --batch=256 --layers=1024,1024,1024,1024 --tiles=64,64,64" ], "environment": { "OMP_NUM_THREADS": "16", "KMP_AFFINITY": "granularity=fine,verbose,compact,1,0" }, "flags": [ "-n", "100", "-run-args='--def-parallel --parallel-task-grid=1,4 --vector-to-kernels --registerBlocking=4,64 '" ], - "extensions": [ "avx512*" ] + "extensions": [ "avx512f" ] } }}, { @@ -99,28 +99,28 @@ "benchmark": [ "mlir-gen", "--kernel=const --bias --relu --float-type=f32 --batch=256 --layers=1024,1024,1024,1024 --tiles=64,64,64" ], "environment": { "OMP_NUM_THREADS": "2", "KMP_AFFINITY": "granularity=fine,verbose,compact,1,0" }, "flags": [ "-n", "100", "-run-args='--def-parallel --parallel-task-grid=4,8 --vector-to-kernels --registerBlocking=4,64 '" ], - "extensions": [ "avx512*" ] + "extensions": [ "avx512f" ] }, "fp32_3x1024_omp_4_mlir": { "type": "IR-GEN", "benchmark": [ "mlir-gen", "--kernel=const --bias --relu --float-type=f32 --batch=256 --layers=1024,1024,1024,1024 --tiles=64,64,64" ], "environment": { "OMP_NUM_THREADS": "4", "KMP_AFFINITY": "granularity=fine,verbose,compact,1,0" }, "flags": [ "-n", "100", "-run-args='--def-parallel --parallel-task-grid=4,4 --vector-to-kernels --registerBlocking=4,64 '" ], - "extensions": [ "avx512*" ] + "extensions": [ "avx512f" ] }, "fp32_3x1024_omp_8_mlir": { "type": "IR-GEN", "benchmark": [ "mlir-gen", "--kernel=const --bias --relu --float-type=f32 --batch=256 --layers=1024,1024,1024,1024 --tiles=64,64,64" ], "environment": { "OMP_NUM_THREADS": "8", "KMP_AFFINITY": "granularity=fine,verbose,compact,1,0" }, "flags": [ "-n", "100", "-run-args='--def-parallel --parallel-task-grid=2,4 --vector-to-kernels --registerBlocking=4,64 '" ], - "extensions": [ "avx512*" ] + "extensions": [ "avx512f" ] }, "fp32_3x1024_omp_16_mlir": { "type": "IR-GEN", "benchmark": [ "mlir-gen", "--kernel=const --bias --relu --float-type=f32 --batch=256 --layers=1024,1024,1024,1024 --tiles=64,64,64" ], "environment": { "OMP_NUM_THREADS": "16", "KMP_AFFINITY": "granularity=fine,verbose,compact,1,0" }, "flags": [ "-n", "100", "-run-args='--def-parallel --parallel-task-grid=1,4 --vector-to-kernels --registerBlocking=4,64 '" ], - "extensions": [ "avx512*" ] + "extensions": [ "avx512f" ] } }}, {