@@ -13,117 +13,111 @@ across various devices by running comprehensive GitHub workflows.
13
13
14
14
<!-- Torchbenchmark start -->
15
15
16
- | | [ torch_npu] [ 1 ] |
17
- | ---------------------------------| ----------------|
18
- | simple_gpt | ❌ |
19
- | detectron2_fasterrcnn_r_50_dc5 | ❌ |
20
- | LearningToPaint | ✅ |
21
- | hf_GPT2_large | ✅ |
22
- | dcgan | ✅ |
23
- | nanogpt | ✅ |
24
- | fastNLP_Bert | ✅ |
25
- | moondream | ❌ |
26
- | mobilenet_v2_quantized_qat | ❌ |
27
- | functorch_dp_cifar10 | ✅ |
28
- | simple_gpt_tp_manual | ❌ |
29
- | speech_transformer | ✅ |
30
- | yolov3 | ✅ |
31
- | resnet50_quantized_qat | ❌ |
32
- | sam_fast | ❌ |
33
- | alexnet | ✅ |
34
- | timm_efficientnet | ✅ |
35
- | pyhpc_isoneutral_mixing | ✅ |
36
- | basic_gnn_edgecnn | ✅ |
37
- | nvidia_deeprecommender | ❌ |
38
- | opacus_cifar10 | ✅ |
39
- | dlrm | ✅ |
40
- | hf_Bert | ✅ |
41
- | hf_T5_generate | ✅ |
42
- | resnet50 | ✅ |
43
- | hf_BigBird | ✅ |
44
- | resnext50_32x4d | ✅ |
45
- | pyhpc_turbulent_kinetic_energy | ✅ |
46
- | llama | ✅ |
47
- | detectron2_maskrcnn_r_50_c4 | ❌ |
48
- | Super_SloMo | ✅ |
49
- | moco | ❌ |
50
- | stable_diffusion_unet | ❌ |
51
- | microbench_unbacked_tolist_sum | ✅ |
52
- | detectron2_maskrcnn_r_101_c4 | ❌ |
53
- | hf_distil_whisper | ✅ |
54
- | mnasnet1_0 | ✅ |
55
- | detectron2_fasterrcnn_r_50_fpn | ❌ |
56
- | timm_resnest | ✅ |
57
- | hf_GPT2 | ✅ |
58
- | squeezenet1_1 | ✅ |
59
- | basic_gnn_gin | ✅ |
60
- | hf_clip | ✅ |
61
- | mobilenet_v2 | ✅ |
62
- | drq | ✅ |
63
- | hf_Roberta_base | ✅ |
64
- | detectron2_maskrcnn_r_50_fpn | ❌ |
65
- | timm_nfnet | ✅ |
66
- | timm_vovnet | ✅ |
67
- | doctr_det_predictor | ✅ |
68
- | sam | ✅ |
69
- | hf_T5_large | ✅ |
70
- | mobilenet_v3_large | ✅ |
71
- | detectron2_fcos_r_50_fpn | ❌ |
72
- | soft_actor_critic | ✅ |
73
- | llava | ❌ |
74
- | timm_regnet | ✅ |
75
- | functorch_maml_omniglot | ✅ |
76
- | detectron2_fasterrcnn_r_101_c4 | ❌ |
77
- | hf_DistilBert | ✅ |
78
- | tts_angular | ✅ |
79
- | detectron2_maskrcnn | ❌ |
80
- | basic_gnn_sage | ✅ |
81
- | tacotron2 | ❌ |
82
- | detectron2_maskrcnn_r_101_fpn | ❌ |
83
- | lennard_jones | ✅ |
84
- | pytorch_unet | ✅ |
85
- | vgg16 | ✅ |
86
- | BERT_pytorch | ✅ |
87
- | timm_efficientdet | ❌ |
88
- | pyhpc_equation_of_state | ✅ |
89
- | maml | ✅ |
90
- | detectron2_fasterrcnn_r_50_c4 | ❌ |
91
- | resnet152 | ✅ |
92
- | phlippe_densenet | ✅ |
93
- | maml_omniglot | ✅ |
94
- | phlippe_resnet | ✅ |
95
- | pytorch_CycleGAN_and_pix2pix | ✅ |
96
- | hf_Whisper | ✅ |
97
- | hf_T5 | ✅ |
98
- | densenet121 | ✅ |
99
- | cm3leon_generate | ✅ |
100
- | detectron2_fasterrcnn_r_101_fpn | ❌ |
101
- | hf_Bert_large | ✅ |
102
- | stable_diffusion_text_encoder | ❌ |
103
- | hf_Reformer | ❌ |
104
- | detectron2_fasterrcnn_r_101_dc5 | ❌ |
105
- | demucs | ✅ |
106
- | pytorch_stargan | ✅ |
107
- | hf_T5_base | ✅ |
108
- | torch_multimodal_clip | ✅ |
109
- | vision_maskrcnn | ❌ |
110
- | timm_vision_transformer_large | ✅ |
111
- | hf_Bart | ✅ |
112
- | shufflenet_v2_x1_0 | ✅ |
113
- | llama_v2_7b_16h | ❌ |
114
- | basic_gnn_gcn | ✅ |
115
- | resnet18 | ✅ |
116
- | Background_Matting | ✅ |
117
- | doctr_reco_predictor | ✅ |
118
- | timm_vision_transformer | ✅ |
119
- | hf_Albert | ✅ |
120
- | hf_Longformer | ✅ |
121
-
122
- [ 1 ] : https://github.com/ascend/pytorch
123
-
124
- [ 2 ] : https://github.com/cosdt/pytorch-integration-tests/actions/workflows/ascend_npu_test.yml/badge.svg
125
-
126
- [ 3 ] : https://github.com/cosdt/pytorch-integration-tests/actions/workflows/ascend_npu_test.yml
16
+ | | npu |
17
+ | ---------------------------------| -------|
18
+ | alexnet | ✅ |
19
+ | Background_Matting | ✅ |
20
+ | basic_gnn_edgecnn | ✅ |
21
+ | basic_gnn_gcn | ✅ |
22
+ | basic_gnn_gin | ✅ |
23
+ | basic_gnn_sage | ✅ |
24
+ | BERT_pytorch | ✅ |
25
+ | cm3leon_generate | ✅ |
26
+ | dcgan | ✅ |
27
+ | demucs | ✅ |
28
+ | densenet121 | ✅ |
29
+ | detectron2_fasterrcnn_r_101_c4 | ❌ |
30
+ | detectron2_fasterrcnn_r_101_dc5 | ❌ |
31
+ | detectron2_fasterrcnn_r_101_fpn | ❌ |
32
+ | detectron2_fasterrcnn_r_50_c4 | ❌ |
33
+ | detectron2_fasterrcnn_r_50_dc5 | ❌ |
34
+ | detectron2_fasterrcnn_r_50_fpn | ❌ |
35
+ | detectron2_fcos_r_50_fpn | ❌ |
36
+ | detectron2_maskrcnn | ❌ |
37
+ | detectron2_maskrcnn_r_101_c4 | ❌ |
38
+ | detectron2_maskrcnn_r_101_fpn | ❌ |
39
+ | detectron2_maskrcnn_r_50_c4 | ❌ |
40
+ | detectron2_maskrcnn_r_50_fpn | ❌ |
41
+ | dlrm | ✅ |
42
+ | doctr_det_predictor | ✅ |
43
+ | doctr_reco_predictor | ✅ |
44
+ | drq | ✅ |
45
+ | fastNLP_Bert | ✅ |
46
+ | functorch_dp_cifar10 | ✅ |
47
+ | functorch_maml_omniglot | ✅ |
48
+ | hf_Albert | ✅ |
49
+ | hf_Bart | ✅ |
50
+ | hf_Bert | ✅ |
51
+ | hf_Bert_large | ✅ |
52
+ | hf_BigBird | ✅ |
53
+ | hf_clip | ✅ |
54
+ | hf_distil_whisper | ✅ |
55
+ | hf_DistilBert | ✅ |
56
+ | hf_GPT2 | ✅ |
57
+ | hf_GPT2_large | ✅ |
58
+ | hf_Longformer | ✅ |
59
+ | hf_Reformer | ❌ |
60
+ | hf_Roberta_base | ✅ |
61
+ | hf_T5 | ✅ |
62
+ | hf_T5_base | ✅ |
63
+ | hf_T5_generate | ✅ |
64
+ | hf_T5_large | ✅ |
65
+ | hf_Whisper | ✅ |
66
+ | LearningToPaint | ✅ |
67
+ | lennard_jones | ✅ |
68
+ | llama | ✅ |
69
+ | llama_v2_7b_16h | ❌ |
70
+ | llava | ❌ |
71
+ | maml | ✅ |
72
+ | maml_omniglot | ✅ |
73
+ | microbench_unbacked_tolist_sum | ✅ |
74
+ | mnasnet1_0 | ✅ |
75
+ | mobilenet_v2 | ✅ |
76
+ | mobilenet_v2_quantized_qat | ❌ |
77
+ | mobilenet_v3_large | ✅ |
78
+ | moco | ❌ |
79
+ | moondream | ❌ |
80
+ | nanogpt | ✅ |
81
+ | nvidia_deeprecommender | ❌ |
82
+ | opacus_cifar10 | ✅ |
83
+ | phlippe_densenet | ✅ |
84
+ | phlippe_resnet | ✅ |
85
+ | pyhpc_equation_of_state | ✅ |
86
+ | pyhpc_isoneutral_mixing | ✅ |
87
+ | pyhpc_turbulent_kinetic_energy | ✅ |
88
+ | pytorch_CycleGAN_and_pix2pix | ✅ |
89
+ | pytorch_stargan | ✅ |
90
+ | pytorch_unet | ✅ |
91
+ | resnet152 | ✅ |
92
+ | resnet18 | ✅ |
93
+ | resnet50 | ✅ |
94
+ | resnet50_quantized_qat | ❌ |
95
+ | resnext50_32x4d | ✅ |
96
+ | sam | ✅ |
97
+ | sam_fast | ❌ |
98
+ | shufflenet_v2_x1_0 | ✅ |
99
+ | simple_gpt | ❌ |
100
+ | simple_gpt_tp_manual | ❌ |
101
+ | soft_actor_critic | ✅ |
102
+ | speech_transformer | ✅ |
103
+ | squeezenet1_1 | ✅ |
104
+ | stable_diffusion_text_encoder | ❌ |
105
+ | stable_diffusion_unet | ❌ |
106
+ | Super_SloMo | ✅ |
107
+ | tacotron2 | ❌ |
108
+ | timm_efficientdet | ❌ |
109
+ | timm_efficientnet | ✅ |
110
+ | timm_nfnet | ✅ |
111
+ | timm_regnet | ✅ |
112
+ | timm_resnest | ✅ |
113
+ | timm_vision_transformer | ✅ |
114
+ | timm_vision_transformer_large | ✅ |
115
+ | timm_vovnet | ✅ |
116
+ | torch_multimodal_clip | ✅ |
117
+ | tts_angular | ✅ |
118
+ | vgg16 | ✅ |
119
+ | vision_maskrcnn | ❌ |
120
+ | yolov3 | ✅ |
127
121
128
122
<!-- Torchbenchmark end -->
129
123
0 commit comments