saymrwulf/onnxruntime
1
0
Fork 0
mirror of https://github.com/saymrwulf/onnxruntime.git synced 2026-07-29 20:14:01 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

[ROCm] update skip_layernorm test sample (#14051)

Browse source 
### Description
<!-- Describe your changes. -->

Larger batch_size won't cover more implementations and may block CI,
remove batch_size 128.

Co-authored-by: peixuanzuo <peixuanzuo@linmif39a000004.zvflicr54joexhdgnhvmxrxygg.phxx.internal.cloudapp.net>
This commit is contained in:
PeixuanZuo 2022-12-22 21:18:10 +08:00 committed by GitHub
parent 2f5bf75e51
commit 694ba033e9
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
1 changed files with 1 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

2
onnxruntime/python/tools/kernel_explorer/kernels/skip_layer_norm_test.py
View file

@ -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)