Skip to content

Commit cb0dee7

Browse files
authored
[Nightly] Sync benchmark list with pytorch for nightly tests (#2306)
disable_all
1 parent 2cced9a commit cb0dee7

File tree

3 files changed

+21
-16
lines changed

3 files changed

+21
-16
lines changed

.ci/benchmarks/huggingface_models_list.txt

Lines changed: 13 additions & 11 deletions
Original file line numberDiff line numberDiff line change
@@ -17,27 +17,36 @@ DistilBertForQuestionAnswering,512
1717
DistillGPT2,32
1818
ElectraForCausalLM,64
1919
ElectraForQuestionAnswering,128
20+
GoogleFnet,32
21+
google/gemma-2-2b,8
22+
google/gemma-3-4b-it,8
2023
GPT2ForSequenceClassification,8
2124
GPTJForCausalLM,1
2225
GPTJForQuestionAnswering,1
2326
GPTNeoForCausalLM,32
2427
GPTNeoForSequenceClassification,32
25-
GoogleFnet,32
2628
LayoutLMForMaskedLM,32
2729
LayoutLMForSequenceClassification,32
2830
M2M100ForConditionalGeneration,64
2931
MBartForCausalLM,8
3032
MBartForConditionalGeneration,4
31-
MT5ForConditionalGeneration,32
3233
MegatronBertForCausalLM,16
3334
MegatronBertForQuestionAnswering,16
35+
meta-llama/Llama-3.2-1B,8
36+
mistralai/Mistral-7B-Instruct-v0.3,8
37+
mistralai/Mistral-7B-Instruct-v0.3,8
3438
MobileBertForMaskedLM,256
3539
MobileBertForQuestionAnswering,256
40+
MT5ForConditionalGeneration,32
41+
openai/gpt-oss-20b,8
42+
openai/gpt-oss-20b,8
43+
openai/whisper-tiny,8
3644
OPTForCausalLM,4
37-
PLBartForCausalLM,16
38-
PLBartForConditionalGeneration,8
3945
PegasusForCausalLM,128
4046
PegasusForConditionalGeneration,64
47+
PLBartForCausalLM,16
48+
PLBartForConditionalGeneration,8
49+
Qwen/Qwen3-0.6B,8
4150
RobertaForCausalLM,32
4251
RobertaForQuestionAnswering,32
4352
T5ForConditionalGeneration,8
@@ -46,10 +55,3 @@ TrOCRForCausalLM,64
4655
XGLMForCausalLM,32
4756
XLNetLMHeadModel,16
4857
YituTechConvBert,32
49-
meta-llama/Llama-3.2-1B,8
50-
google/gemma-2-2b,8
51-
google/gemma-3-4b-it,8
52-
openai/whisper-tiny,8
53-
Qwen/Qwen3-0.6B,8
54-
mistralai/Mistral-7B-Instruct-v0.3,8
55-
openai/gpt-oss-20b,8

.ci/benchmarks/timm_models_list.txt

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -6,9 +6,11 @@ coat_lite_mini 128
66
convit_base 128
77
convmixer_768_32 64
88
convnext_base 128
9+
convnextv2_nano.fcmae_ft_in22k_in1k 128
910
crossvit_9_240 256
1011
cspdarknet53 128
1112
deit_base_distilled_patch16_224 128
13+
deit_tiny_patch16_224.fb_in1k 128
1214
dla102 128
1315
dm_nfnet_f0 128
1416
dpn107 64
@@ -56,10 +58,8 @@ tinynet_a 128
5658
tnt_s_patch16_224 128
5759
twins_pcpvt_base 128
5860
visformer_small 128
61+
vit_base_patch14_dinov2.lvd142m 128
5962
vit_base_patch16_224 128
63+
vit_base_patch16_siglip_256 128
6064
volo_d1_224 128
6165
xcit_large_24_p8_224 16
62-
convnextv2_nano.fcmae_ft_in22k_in1k
63-
deit_tiny_patch16_224.fb_in1k
64-
vit_base_patch14_dinov2.lvd142m
65-
vit_base_patch16_siglip_256

.ci/benchmarks/torchbench.yaml

Lines changed: 4 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -43,6 +43,7 @@ tolerance:
4343
- doctr_reco_predictor
4444
- drq
4545
- phlippe_resnet
46+
- pytorch_CycleGAN_and_pix2pix
4647

4748
higher_bf16:
4849
- doctr_reco_predictor
@@ -154,7 +155,7 @@ trt_not_yet_working:
154155
skip:
155156
all:
156157
# OOMs (A100 40G)
157-
- detectron2_maskrcnn
158+
# - detectron2_maskrcnn
158159
# TIMEOUT, https://github.com/pytorch/pytorch/issues/98467
159160
- tacotron2
160161
# Failing in eager mode
@@ -216,6 +217,8 @@ skip:
216217

217218
cuda: []
218219

220+
xpu: []
221+
219222
test:
220223
training:
221224
- *DETECTRON2_MODELS

0 commit comments

Comments
 (0)