mirrors/mmdeploy
1
0
Fork 0
mirror of https://github.com/open-mmlab/mmdeploy.git synced 2025-01-14 08:09:43 +08:00
Code Issues Projects Releases Wiki Activity

Disable scatternd in TRT8+ (#277)

Browse Source
This commit is contained in:
q.yao 2021-12-13 16:43:22 +08:00 committed by GitHub
parent 25a5838080
commit bd2867178d
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 5 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
csrc/backend_ops/tensorrt/scatternd/trt_scatternd.cpp
View File

@ -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