add the following code to './torch2trt/torch2trt/plugins/NvInfer.h'
https://drive.google.com/file/d/1dIAiFoU7ROkLVw5OA5ezYifA-F77z6NE/view?usp=sharing