fix: Review comments incorporated

Signed-off-by: Anurag Dixit <[email protected]>
pytorch · May 2, 2022 · 962660d · 962660d
1 parent 417c096
commit 962660d
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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=[