We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent 7ebe35c commit c6497b4Copy full SHA for c6497b4
1 file changed
tests/py/dynamo/models/test_weight_stripped_engine.py
@@ -274,8 +274,8 @@ def test_engine_caching_saves_weight_stripped_engine(self):
274
"torchvision is not installed",
275
)
276
@unittest.skipIf(
277
- torch_trt.ENABLED_FEATURES.tensorrt_rtx,
278
- "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",
279
280
def test_dynamo_compile_with_refittable_weight_stripped_engine(self):
281
pyt_model = models.resnet18(pretrained=True).eval().to("cuda")
0 commit comments