Skip to content

Commit 37d1674

Browse files
committed
fix bug: Command '['ninja', '-v']' returned non-zero exit status 1
1 parent 51721ae commit 37d1674

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

py/torch_tensorrt/csrc/torch_tensorrt_py.cpp

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -2,7 +2,7 @@
22
#include "pybind11/stl.h"
33

44
#include "ATen/core/jit_type.h"
5-
#include "NvInferRuntimeBase.h"
5+
#include "NvInferRuntime.h"
66
#include "Python.h"
77
#include "core/compiler.h"
88
#include "core/conversion/conversion.h"

0 commit comments

Comments
 (0)