Disable scatternd in TRT8+ (open-mmlab#277)

humu789 · Dec 13, 2021 · bd28671 · bd28671
1 parent 25a5838
commit bd28671
Showing 1 changed file with 5 additions and 2 deletions.
diff --git a/csrc/backend_ops/tensorrt/scatternd/trt_scatternd.cpp b/csrc/backend_ops/tensorrt/scatternd/trt_scatternd.cpp
@@ -1,11 +1,13 @@
 // Copyright (c) OpenMMLab. All rights reserved.
-#include "trt_scatternd.hpp"
-
+#include "NvInferVersion.h"
+// ScatterND is supported since TensorRT8
+#if NV_TENSORRT_MAJOR <= 7
 #include <assert.h>
 #include <stdio.h>
 
 #include <chrono>
 
+#include "trt_scatternd.hpp"
 #include "trt_scatternd_kernel.hpp"
 #include "trt_serialize.hpp"
 
@@ -151,3 +153,4 @@ nvinfer1::IPluginV2 *TRTScatterNDCreator::deserializePlugin(const char *name,
 
 REGISTER_TENSORRT_PLUGIN(TRTScatterNDCreator);
 }  // namespace mmdeploy
+#endif