Fix thread safety of softmax prediction. (#7104)

2021-07-16 02:06:55 +08:00
parent 2801d69fb7
commit abec3dbf6d
6 changed files with 20 additions and 23 deletions
--- a/plugin/example/custom_obj.cc
+++ b/plugin/example/custom_obj.cc
@@ -58,7 +58,7 @@ class MyLogistic : public ObjFunction {
  const char* DefaultEvalMetric() const override {
    return "logloss";
  }
-  void PredTransform(HostDeviceVector<bst_float> *io_preds) override {
+  void PredTransform(HostDeviceVector<bst_float> *io_preds) const override {
    // transform margin value to probability.
    std::vector<bst_float> &preds = io_preds->HostVector();
    for (auto& pred : preds) {