Skip to content

Commit

Permalink
update tensorrt_loggin.h and config.h.in definition
Browse files Browse the repository at this point in the history
  • Loading branch information
Ar-Ray-code committed Aug 12, 2024
1 parent daf8a2d commit 374ab1b
Show file tree
Hide file tree
Showing 2 changed files with 7 additions and 7 deletions.
6 changes: 3 additions & 3 deletions yolox_ros_cpp/yolox_cpp/include/yolox_cpp/config.h.in
Original file line number Diff line number Diff line change
Expand Up @@ -12,12 +12,12 @@
// See the License for the specific language governing permissions and
// limitations under the License.

#ifndef YOLOX_CPP__CONFIG_H_IN_
#define YOLOX_CPP__CONFIG_H_IN_
#ifndef YOLOX_CPP__CONFIG_H_
#define YOLOX_CPP__CONFIG_H_

#cmakedefine ENABLE_OPENVINO
#cmakedefine ENABLE_TENSORRT
#cmakedefine ENABLE_ONNXRUNTIME
#cmakedefine ENABLE_TFLITE

#endif // YOLOX_CPP__CONFIG_H_IN_
#endif // YOLOX_CPP__CONFIG_H_
8 changes: 4 additions & 4 deletions yolox_ros_cpp/yolox_cpp/include/yolox_cpp/tensorrt_logging.h
Original file line number Diff line number Diff line change
Expand Up @@ -93,7 +93,7 @@ namespace yolox_cpp
std::ostream & mOutput;
std::string mPrefix;
bool mShouldLog {};
}; // class LogStreamConsumerBuffer
};


class LogStreamConsumerBase
Expand All @@ -105,7 +105,7 @@ namespace yolox_cpp

protected:
LogStreamConsumerBuffer mBuffer;
}; // class LogStreamConsumerBase
};

class LogStreamConsumer: protected LogStreamConsumerBase, public std::ostream
{
Expand Down Expand Up @@ -161,7 +161,7 @@ namespace yolox_cpp

bool mShouldLog;
Severity mSeverity;
}; // class LogStreamConsumer
};


class Logger: public nvinfer1::ILogger
Expand Down Expand Up @@ -324,7 +324,7 @@ namespace yolox_cpp
}

Severity mReportableSeverity;
}; // class Logger
};

inline LogStreamConsumer LOG_VERBOSE(const Logger & logger)
{
Expand Down

0 comments on commit 374ab1b

Please sign in to comment.