From 03092ba407e25ee7aab967292dbfd82efdb1aec3 Mon Sep 17 00:00:00 2001 From: lanluo-nvidia Date: Tue, 13 Aug 2024 17:36:19 -0700 Subject: [PATCH] fix converter test error (#3083) --- tests/py/dynamo/conversion/test_argmin_aten.py | 1 + 1 file changed, 1 insertion(+) diff --git a/tests/py/dynamo/conversion/test_argmin_aten.py b/tests/py/dynamo/conversion/test_argmin_aten.py index c0290f943e..7065ca748c 100644 --- a/tests/py/dynamo/conversion/test_argmin_aten.py +++ b/tests/py/dynamo/conversion/test_argmin_aten.py @@ -2,6 +2,7 @@ import torch.nn as nn from parameterized import parameterized from torch.testing._internal.common_utils import run_tests +from torch_tensorrt import Input from .harness import DispatchTestCase