From 694ba033e91f71d3585a92c6e48d09e7f3181083 Mon Sep 17 00:00:00 2001 From: PeixuanZuo <94887879+PeixuanZuo@users.noreply.github.com> Date: Thu, 22 Dec 2022 21:18:10 +0800 Subject: [PATCH] [ROCm] update skip_layernorm test sample (#14051) ### Description Larger batch_size won't cover more implementations and may block CI, remove batch_size 128. Co-authored-by: peixuanzuo --- .../tools/kernel_explorer/kernels/skip_layer_norm_test.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/onnxruntime/python/tools/kernel_explorer/kernels/skip_layer_norm_test.py b/onnxruntime/python/tools/kernel_explorer/kernels/skip_layer_norm_test.py index c1f1a4201a..d35965513d 100644 --- a/onnxruntime/python/tools/kernel_explorer/kernels/skip_layer_norm_test.py +++ b/onnxruntime/python/tools/kernel_explorer/kernels/skip_layer_norm_test.py @@ -15,7 +15,7 @@ from utils import dtype_to_bytes def get_bert_sizes_test(): - batch_sizes = [1, 8, 128] + batch_sizes = [1, 8] seq_lens = [64, 256] hidden_sizes = [1, 2, 3, 4, 5, 7, 8, 9, 13, 32, 63, 64, 65, 127, 128, 129, 177, 256, 1023, 1024] return product(batch_sizes, seq_lens, hidden_sizes)