Skip to content

Commit

Permalink
NFC: Fix build of TensorRT Lua module
Browse files Browse the repository at this point in the history
  • Loading branch information
parthchadha authored Oct 25, 2024
1 parent 5085987 commit 790bf78
Showing 1 changed file with 2 additions and 1 deletion.
Original file line number Diff line number Diff line change
Expand Up @@ -156,7 +156,8 @@ class OutputAllocatorImpl : public nvinfer1::IOutputAllocator {
return reinterpret_cast<void *>(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;
}
Expand Down

0 comments on commit 790bf78

Please sign in to comment.