Skip to content

Commit e185d8b

Browse files
committed
PTQ test libraries import error
1 parent 0b4536c commit e185d8b

File tree

1 file changed

+2
-0
lines changed

1 file changed

+2
-0
lines changed

py/torch_tensorrt/__init__.py

+2
Original file line numberDiff line numberDiff line change
@@ -85,6 +85,8 @@ def _find_lib(name: str, paths: List[str]) -> str:
8585
from torch_tensorrt._Device import Device # noqa: F401
8686
from torch_tensorrt._enums import * # noqa: F403
8787
from torch_tensorrt._Input import Input # noqa: F401
88+
from torch_tensorrt.logging import *
89+
from torch_tensorrt.ptq import *
8890
from torch_tensorrt._utils import * # noqa: F403
8991
from torch_tensorrt._utils import sanitized_torch_version
9092

0 commit comments

Comments
 (0)