diff --git a/py/setup.py b/py/setup.py index 97dde9ed9b..1e272d3d8d 100644 --- a/py/setup.py +++ b/py/setup.py @@ -239,7 +239,7 @@ def run(self): libraries=["torchtrt"], include_dirs=[ dir_path + "torch_tensorrt/csrc", dir_path + "torch_tensorrt/include", - dir_path + "/../bazel-Torch-TensorRT/external/tensorrt/include", + dir_path + "/../bazel-TRTorch/external/tensorrt/include", dir_path + "/../bazel-Torch-TensorRT/external/tensorrt/include", dir_path + "/../" ], extra_compile_args=[