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/src/objective/aft_obj.cu
+++ b/src/objective/aft_obj.cu
@@ -102,7 +102,7 @@ class AFTObj : public ObjFunction {
    }
  }

-  void PredTransform(HostDeviceVector<bst_float> *io_preds) override {
+  void PredTransform(HostDeviceVector<bst_float> *io_preds) const override {
    // Trees give us a prediction in log scale, so exponentiate
    common::Transform<>::Init(
        [] XGBOOST_DEVICE(size_t _idx, common::Span<bst_float> _preds) {