Skip to content

Commit 12b2e24

Browse files
debug flag is deprecated, remove it so that test won't complain (#3610)
1 parent e6086b7 commit 12b2e24

File tree

4 files changed

+0
-11
lines changed

4 files changed

+0
-11
lines changed

tests/py/dynamo/backend/test_specialized_models.py

Lines changed: 0 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -393,7 +393,6 @@ def forward(self, x):
393393
backend="torch_tensorrt",
394394
options={
395395
"min_block_size": 1,
396-
"debug": True,
397396
},
398397
)
399398
with torch.no_grad():

tests/py/dynamo/models/test_engine_cache.py

Lines changed: 0 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -423,7 +423,6 @@ def remove_timing_cache(path=TIMING_CACHE_PATH):
423423
options={
424424
"use_python_runtime": False,
425425
"enabled_precisions": {torch.float},
426-
"debug": False,
427426
"min_block_size": 1,
428427
"immutable_weights": False,
429428
"cache_built_engines": cache_built_engines,
@@ -490,7 +489,6 @@ def test_torch_compile_with_custom_engine_cache(self):
490489
options={
491490
"use_python_runtime": False,
492491
"enabled_precisions": {torch.float},
493-
"debug": False,
494492
"min_block_size": 1,
495493
"immutable_weights": False,
496494
"cache_built_engines": cache_built_engines,
@@ -547,7 +545,6 @@ def test_torch_trt_compile_change_input_shape(self):
547545
**{
548546
"use_python_runtime": True,
549547
"enabled_precisions": {torch.float},
550-
"debug": False,
551548
"min_block_size": 1,
552549
"immutable_weights": False,
553550
"cache_built_engines": True,
@@ -590,7 +587,6 @@ def forward(self, x):
590587
options={
591588
"use_python_runtime": True,
592589
"enabled_precisions": {torch.float},
593-
"debug": False,
594590
"min_block_size": 1,
595591
"immutable_weights": False,
596592
"cache_built_engines": True,

tests/py/dynamo/models/test_weight_stripped_engine.py

Lines changed: 0 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -35,7 +35,6 @@ def test_three_ways_to_compile(self):
3535
settings = {
3636
"use_python_runtime": False,
3737
"enabled_precisions": {torch.float},
38-
"debug": False,
3938
"min_block_size": 1,
4039
"immutable_weights": False,
4140
"strip_engine_weights": False,
@@ -83,7 +82,6 @@ def test_three_ways_to_compile_weight_stripped_engine(self):
8382
settings = {
8483
"use_python_runtime": False,
8584
"enabled_precisions": {torch.float},
86-
"debug": False,
8785
"min_block_size": 1,
8886
"immutable_weights": False,
8987
"strip_engine_weights": True,
@@ -203,7 +201,6 @@ def test_weight_stripped_engine_results(self):
203201
options={
204202
"use_python_runtime": False,
205203
"enabled_precisions": {torch.float},
206-
"debug": False,
207204
"min_block_size": 1,
208205
"immutable_weights": False,
209206
"cache_built_engines": False,
@@ -410,7 +407,6 @@ def remove_timing_cache(path=TIMING_CACHE_PATH):
410407
options={
411408
"use_python_runtime": False,
412409
"enabled_precisions": {torch.float},
413-
"debug": False,
414410
"min_block_size": 1,
415411
"immutable_weights": False,
416412
"cache_built_engines": cache_built_engines,
@@ -488,7 +484,6 @@ def forward(self, x):
488484
options={
489485
"use_python_runtime": True,
490486
"enabled_precisions": {torch.float},
491-
"debug": False,
492487
"min_block_size": 1,
493488
"immutable_weights": False,
494489
"cache_built_engines": True,

tests/py/ts/integrations/test_to_backend_api.py

Lines changed: 0 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -22,7 +22,6 @@ def setUp(self):
2222
"inputs": [torchtrt.Input([1, 3, 300, 300])],
2323
"enabled_precisions": {torch.float},
2424
"refit": False,
25-
"debug": False,
2625
"device": {
2726
"device_type": torchtrt.DeviceType.GPU,
2827
"gpu_id": 0,

0 commit comments

Comments
 (0)