diff --git a/mlir-tensorrt/executor/lib/Runtime/Backend/Lua/Modules/TensorRT/TensorRTModule.cpp b/mlir-tensorrt/executor/lib/Runtime/Backend/Lua/Modules/TensorRT/TensorRTModule.cpp index d88ddf4ed..0d8da09a8 100644 --- a/mlir-tensorrt/executor/lib/Runtime/Backend/Lua/Modules/TensorRT/TensorRTModule.cpp +++ b/mlir-tensorrt/executor/lib/Runtime/Backend/Lua/Modules/TensorRT/TensorRTModule.cpp @@ -156,7 +156,8 @@ class OutputAllocatorImpl : public nvinfer1::IOutputAllocator { return reinterpret_cast(mOutputPtr); } - void notifyShape(const char *name, const nvinfer1::Dims &dims) override { + void notifyShape(const char *name, + const nvinfer1::Dims &dims) noexcept override { assert(name == mTensorName && "Tensor name mismatch"); mOutputDims = dims; }