File tree Expand file tree Collapse file tree 3 files changed +6
-2
lines changed
userbenchmark/dynamo/dynamobench Expand file tree Collapse file tree 3 files changed +6
-2
lines changed Original file line number Diff line number Diff line change @@ -2861,7 +2861,7 @@ def get_multiplier():
28612861 elif use_larger_multiplier_for_smaller_tensor and (
28622862 fp64_ref .numel () <= 500
28632863 ):
2864- multiplier = 5 .0
2864+ multiplier = 8 .0
28652865 elif (
28662866 fp64_ref .numel () < 1000
28672867 or (ref .ndim == 4 and ref .shape [- 1 ] == ref .shape [- 2 ] == 1 )
Original file line number Diff line number Diff line change @@ -369,7 +369,10 @@ def skip_models_due_to_control_flow(self):
369369 return self ._skip ["control_flow" ]
370370
371371 def use_larger_multiplier_for_smaller_tensor (self , name ):
372- return name in ["ElectraForQuestionAnswering" ]
372+ return name in [
373+ "ElectraForQuestionAnswering" ,
374+ "MegatronBertForQuestionAnswering" ,
375+ ]
373376
374377 def _get_model_cls_and_config (self , model_name ):
375378 if model_name not in EXTRA_MODELS :
Original file line number Diff line number Diff line change @@ -127,6 +127,7 @@ def pip_install(package):
127127 "inception_v3" ,
128128 "mobilenetv3_large_100" ,
129129 "cspdarknet53" ,
130+ "gluon_inception_v3" ,
130131}
131132
132133
You can’t perform that action at this time.
0 commit comments