Skip to content

Commit 048d538

Browse files
committed
skip : weight stripped engine timing test
Signed-off-by: tejaswinp <tejaswinp@nvidia.com>
1 parent 4bba8ae commit 048d538

1 file changed

Lines changed: 4 additions & 0 deletions

File tree

tests/py/dynamo/models/test_weight_stripped_engine.py

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -278,6 +278,10 @@ def test_engine_caching_saves_weight_stripped_engine(self):
278278
not importlib.util.find_spec("torchvision"),
279279
"torchvision is not installed",
280280
)
281+
@unittest.skipIf(
282+
torch_trt.ENABLED_FEATURES.tensorrt_rtx,
283+
"Engine caching compilation time assertion is unreliable with TensorRT-RTX",
284+
)
281285
def test_dynamo_compile_with_refittable_weight_stripped_engine(self):
282286
pyt_model = models.resnet18(pretrained=True).eval().to("cuda")
283287
example_inputs = (torch.randn((100, 3, 224, 224)).to("cuda"),)

0 commit comments

Comments
 (0)