text: #ifndef _TRT_COMMON_H_
  #define _TRT_COMMON_H_
  #include "NvInfer.h"
  #include <string>
  #include <vector>
  #include <fstream>
  #include <cassert>
  #include <iostream>
  #define CHECK(status)									\
  {														\
  	if (status != 0)									\
  	{													\
  		std::cout << "Cuda failure: " << status;		\
  		abort();										\
  	}													\
  }
  
  
  // Logger for GIE info/warning/errors
  class Logger : public nvinfer1::ILogger			
  {
      public:
  	void log(nvinfer1::ILogger::Severity severity, const char* msg) override
  	{
  		// suppress info-level messages
          if (severity == Severity::kINFO) return;
  
          switch (severity)
          {
              case Severity::kINTERNAL_ERROR: std::cerr << "INTERNAL_ERROR: "; break;
              case Severity::kERROR: std::cerr << "ERROR: "; break;
              case Severity::kWARNING: std::cerr << "WARNING: "; break;
              case Severity::kINFO: std::cerr << "INFO: "; break;
              default: std::cerr << "UNKNOWN: "; break;
          }
          std::cerr << msg << std::endl;
  	}
  };
  
  std::string locateFile(const std::string& input, const std::vector<std::string> & directories);
  void readPGMFile(const std::string& fileName,  uint8_t *buffer, int inH, int inW);
  #endif // _TRT_COMMON_H_
  