Skip to content

Commit 7ebe35c

Browse files
committed
skip : weight stripped engine timing test
Signed-off-by: tejaswinp <tejaswinp@nvidia.com>
1 parent d017ed7 commit 7ebe35c

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
@@ -273,6 +273,10 @@ def test_engine_caching_saves_weight_stripped_engine(self):
273273
not importlib.util.find_spec("torchvision"),
274274
"torchvision is not installed",
275275
)
276+
@unittest.skipIf(
277+
torch_trt.ENABLED_FEATURES.tensorrt_rtx,
278+
"Engine caching compilation time assertion is unreliable with TensorRT-RTX",
279+
)
276280
def test_dynamo_compile_with_refittable_weight_stripped_engine(self):
277281
pyt_model = models.resnet18(pretrained=True).eval().to("cuda")
278282
example_inputs = (torch.randn((100, 3, 224, 224)).to("cuda"),)

0 commit comments

Comments
 (0)