@@ -24,7 +24,6 @@ def test_resnet18(ir):
2424 "device" : torchtrt .Device ("cuda:0" ),
2525 "enabled_precisions" : {torch .float },
2626 "ir" : ir ,
27- "max_num_trt_engines" : 200 ,
2827 }
2928
3029 trt_mod = torchtrt .compile (model , ** compile_spec )
@@ -55,7 +54,6 @@ def test_mobilenet_v2(ir):
5554 "device" : torchtrt .Device ("cuda:0" ),
5655 "enabled_precisions" : {torch .float },
5756 "ir" : ir ,
58- "max_num_trt_engines" : 200 ,
5957 }
6058
6159 trt_mod = torchtrt .compile (model , ** compile_spec )
@@ -86,7 +84,6 @@ def test_efficientnet_b0(ir):
8684 "device" : torchtrt .Device ("cuda:0" ),
8785 "enabled_precisions" : {torch .float },
8886 "ir" : ir ,
89- "max_num_trt_engines" : 200 ,
9087 }
9188
9289 trt_mod = torchtrt .compile (model , ** compile_spec )
@@ -126,7 +123,6 @@ def test_bert_base_uncased(ir):
126123 "enabled_precisions" : {torch .float },
127124 "truncate_long_and_double" : True ,
128125 "ir" : ir ,
129- "max_num_trt_engines" : 200 ,
130126 }
131127 trt_mod = torchtrt .compile (model , ** compile_spec )
132128
0 commit comments