Method check_engine

tools/Polygraphy/tests/tools/test_convert.py:53–56

(self, path)

class TestConvertToTrt:
    def check_engine(self, path):
        loader = EngineFromBytes(BytesFromPath(path))
        with loader() as engine:
            assert isinstance(engine, trt.ICudaEngine)

    def test_onnx_to_trt(self, poly_convert):
        with util.NamedTemporaryFile(suffix=".engine") as outmodel: