File tree Expand file tree Collapse file tree 2 files changed +3
-1
lines changed
recipes/xnnpack_optimization Expand file tree Collapse file tree 2 files changed +3
-1
lines changed Original file line number Diff line number Diff line change 43
43
def verify_xnnpack_quantizer_matching_fx_quant_model (model_name , model , example_inputs ):
44
44
"""This is a verification against fx graph mode quantization flow as a sanity check"""
45
45
46
- if model_name == "edsr" :
46
+ if model_name in [ "edsr" , "mobilebert" ] :
47
47
# EDSR has control flows that are not traceable in symbolic_trace
48
+ # mobilebert is not symbolically traceable with torch.fx.symbolic_trace
48
49
return
49
50
if model_name == "ic3" :
50
51
# we don't want to compare results of inception_v3 with fx, since mul op with Scalar
Original file line number Diff line number Diff line change @@ -27,4 +27,5 @@ class OptimizationOptions(object):
27
27
"vit" : OptimizationOptions (False , True ),
28
28
"w2l" : OptimizationOptions (False , True ),
29
29
"edsr" : OptimizationOptions (True , False ),
30
+ "mobilebert" : OptimizationOptions (True , False ),
30
31
}
You can’t perform that action at this time.
0 commit comments