Skip to content

Commit 148c978

Browse files
committed
Fixing macro
Signed-off-by: Boris Fomitchev <[email protected]>
1 parent 28eb55e commit 148c978

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed

py/trtorch/csrc/tensorrt_backend.cpp

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -67,7 +67,8 @@ c10::impl::GenericList TensorRTBackend::execute(c10::IValue handle, c10::impl::G
6767
namespace {
6868
c10::IValue preprocess(const torch::jit::Module& mod,
6969
const c10::Dict<c10::IValue,
70-
#ifdef EARLY_PYTORCH_19X_VERSION
70+
// this API changed between 1.9 and 1.10
71+
#if TORCH_VERSION_MAJOR < 2 && TORCH_VERSION_MINOR < 10
7172
c10::IValue>& method_compile_spec
7273
# else
7374
c10::IValue>& method_compile_spec, const torch::jit::BackendDebugHandleGenerator& generate_debug_handles

0 commit comments

Comments
 (0)