TRT-10 GA Support for main branch by zewenli98 · Pull Request #2781 · pytorch/TensorRT (original) (raw)

--- /home/runner/work/TensorRT/TensorRT/tests/py/dynamo/models/test_dyn_models.py 2024-04-29 19:55:18.604129+00:00 +++ /home/runner/work/TensorRT/TensorRT/tests/py/dynamo/models/test_dyn_models.py 2024-04-29 19:57:14.310783+00:00 @@ -7,13 +7,11 @@ from torch_tensorrt.dynamo.utils import COSINE_THRESHOLD, cosine_similarity

assertions = unittest.TestCase()

-@unittest.skip(

-) +@unittest.skip("Skipping this test for now due to a bug") @pytest.mark.unit def test_base_dynamic(ir): """ Tests the model (which is fully convertible) with dynamic shapes """ @@ -67,13 +65,11 @@

with torch.no_grad():
    torch.cuda.empty_cache()

-@unittest.skip(

-) +@unittest.skip("Skipping this test for now due to a bug") @pytest.mark.unit def test_base_dynamic_fallback(ir): """ Tests the model with dynamic shapes where torch.abs op is forced to run in PyTorch """