Skip to content

Commit 357d18d

Browse files
committed
fix: Remove remnants of max_num_trt_engines
- Update default `min_block_size`
1 parent 2579ccd commit 357d18d

File tree

3 files changed

+2
-6
lines changed

3 files changed

+2
-6
lines changed

py/torch_tensorrt/dynamo/backend/_defaults.py

+1-1
Original file line numberDiff line numberDiff line change
@@ -4,4 +4,4 @@
44
PRECISION = LowerPrecision.FP32
55
DEBUG = False
66
MAX_WORKSPACE_SIZE = 20 << 30
7-
MIN_BLOCK_SIZE = 3
7+
MIN_BLOCK_SIZE = 5

py/torch_tensorrt/dynamo/backend/test/test_partitioning.py

+1-1
Original file line numberDiff line numberDiff line change
@@ -36,7 +36,7 @@ def forward(self, x, y):
3636
return pow_
3737

3838
fx_graph = torch.fx.symbolic_trace(FullySupportedMultiOp())
39-
partitioned_graph = partition(deepcopy(fx_graph))
39+
partitioned_graph = partition(deepcopy(fx_graph), min_block_size=2)
4040
self.assertEquals(
4141
len(list(partitioned_graph.named_children())),
4242
1,

py/torch_tensorrt/dynamo/test/test_dynamo_backend.py

-4
Original file line numberDiff line numberDiff line change
@@ -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

Comments
 (0)