We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent 048d538 commit a5fd05cCopy full SHA for a5fd05c
1 file changed
tests/py/dynamo/models/test_weight_stripped_engine.py
@@ -279,8 +279,8 @@ def test_engine_caching_saves_weight_stripped_engine(self):
279
"torchvision is not installed",
280
)
281
@unittest.skipIf(
282
- torch_trt.ENABLED_FEATURES.tensorrt_rtx,
283
- "Engine caching compilation time assertion is unreliable with TensorRT-RTX",
+ torch_trt.ENABLED_FEATURES.tensorrt_rtx,
+ "Engine caching compilation time assertion is unreliable with TensorRT-RTX",
284
285
def test_dynamo_compile_with_refittable_weight_stripped_engine(self):
286
pyt_model = models.resnet18(pretrained=True).eval().to("cuda")
0 commit comments