Skip to content
GitLab
Menu
Projects
Groups
Snippets
Loading...
Help
Help
Support
Community forum
Keyboard shortcuts
?
Submit feedback
Contribute to GitLab
Sign in / Register
Toggle navigation
Menu
Open sidebar
tianlh
LightGBM-DCU
Commits
548cec82
Commit
548cec82
authored
Oct 21, 2025
by
Jeff Daily
Browse files
Merge branch 'master' into rocm3
parents
2f7bd8ef
5dbfcdc4
Changes
195
Hide whitespace changes
Inline
Side-by-side
Showing
20 changed files
with
51 additions
and
42 deletions
+51
-42
src/network/linkers_socket.cpp
src/network/linkers_socket.cpp
+1
-0
src/network/network.cpp
src/network/network.cpp
+3
-0
src/network/socket_wrapper.hpp
src/network/socket_wrapper.hpp
+3
-3
src/objective/binary_objective.hpp
src/objective/binary_objective.hpp
+3
-3
src/objective/cuda/cuda_binary_objective.hpp
src/objective/cuda/cuda_binary_objective.hpp
+3
-3
src/objective/cuda/cuda_multiclass_objective.hpp
src/objective/cuda/cuda_multiclass_objective.hpp
+3
-3
src/objective/cuda/cuda_rank_objective.hpp
src/objective/cuda/cuda_rank_objective.hpp
+3
-3
src/objective/cuda/cuda_regression_objective.hpp
src/objective/cuda/cuda_regression_objective.hpp
+3
-3
src/objective/multiclass_objective.hpp
src/objective/multiclass_objective.hpp
+3
-3
src/objective/objective_function.cpp
src/objective/objective_function.cpp
+2
-0
src/objective/rank_objective.hpp
src/objective/rank_objective.hpp
+3
-3
src/objective/regression_objective.hpp
src/objective/regression_objective.hpp
+3
-3
src/objective/xentropy_objective.hpp
src/objective/xentropy_objective.hpp
+3
-3
src/treelearner/col_sampler.hpp
src/treelearner/col_sampler.hpp
+3
-3
src/treelearner/cost_effective_gradient_boosting.hpp
src/treelearner/cost_effective_gradient_boosting.hpp
+3
-3
src/treelearner/cuda/cuda_best_split_finder.cpp
src/treelearner/cuda/cuda_best_split_finder.cpp
+1
-0
src/treelearner/cuda/cuda_best_split_finder.cu
src/treelearner/cuda/cuda_best_split_finder.cu
+1
-0
src/treelearner/cuda/cuda_best_split_finder.hpp
src/treelearner/cuda/cuda_best_split_finder.hpp
+3
-3
src/treelearner/cuda/cuda_data_partition.cpp
src/treelearner/cuda/cuda_data_partition.cpp
+1
-0
src/treelearner/cuda/cuda_data_partition.hpp
src/treelearner/cuda/cuda_data_partition.hpp
+3
-3
No files found.
src/network/linkers_socket.cpp
View file @
548cec82
...
...
@@ -11,6 +11,7 @@
#include <algorithm>
#include <chrono>
#include <cstring>
#include <memory>
#include <string>
#include <thread>
#include <unordered_map>
...
...
src/network/network.cpp
View file @
548cec82
...
...
@@ -6,8 +6,11 @@
#include <LightGBM/utils/common.h>
#include <algorithm>
#include <cstdlib>
#include <cstring>
#include <memory>
#include <vector>
#include "linkers.h"
...
...
src/network/socket_wrapper.hpp
View file @
548cec82
...
...
@@ -2,8 +2,8 @@
* Copyright (c) 2016 Microsoft Corporation. All rights reserved.
* Licensed under the MIT License. See LICENSE file in the project root for license information.
*/
#ifndef LIGHTGBM_NETWORK_SOCKET_WRAPPER_HPP_
#define LIGHTGBM_NETWORK_SOCKET_WRAPPER_HPP_
#ifndef LIGHTGBM_
SRC_
NETWORK_SOCKET_WRAPPER_HPP_
#define LIGHTGBM_
SRC_
NETWORK_SOCKET_WRAPPER_HPP_
#ifdef USE_SOCKET
#include <LightGBM/utils/log.h>
...
...
@@ -319,4 +319,4 @@ class TcpSocket {
}
// namespace LightGBM
#endif // USE_SOCKET
#endif // L
ightGBM
_NETWORK_SOCKET_WRAPPER_HPP_
#endif // L
IGHTGBM_SRC
_NETWORK_SOCKET_WRAPPER_HPP_
src/objective/binary_objective.hpp
View file @
548cec82
...
...
@@ -2,8 +2,8 @@
* Copyright (c) 2016 Microsoft Corporation. All rights reserved.
* Licensed under the MIT License. See LICENSE file in the project root for license information.
*/
#ifndef LIGHTGBM_OBJECTIVE_BINARY_OBJECTIVE_HPP_
#define LIGHTGBM_OBJECTIVE_BINARY_OBJECTIVE_HPP_
#ifndef LIGHTGBM_
SRC_
OBJECTIVE_BINARY_OBJECTIVE_HPP_
#define LIGHTGBM_
SRC_
OBJECTIVE_BINARY_OBJECTIVE_HPP_
#include <LightGBM/network.h>
#include <LightGBM/objective_function.h>
...
...
@@ -213,4 +213,4 @@ class BinaryLogloss: public ObjectiveFunction {
};
}
// namespace LightGBM
#endif // L
ightGBM
_OBJECTIVE_BINARY_OBJECTIVE_HPP_
#endif // L
IGHTGBM_SRC
_OBJECTIVE_BINARY_OBJECTIVE_HPP_
src/objective/cuda/cuda_binary_objective.hpp
View file @
548cec82
...
...
@@ -4,8 +4,8 @@
* license information.
*/
#ifndef LIGHTGBM_OBJECTIVE_CUDA_CUDA_BINARY_OBJECTIVE_HPP_
#define LIGHTGBM_OBJECTIVE_CUDA_CUDA_BINARY_OBJECTIVE_HPP_
#ifndef LIGHTGBM_
SRC_
OBJECTIVE_CUDA_CUDA_BINARY_OBJECTIVE_HPP_
#define LIGHTGBM_
SRC_
OBJECTIVE_CUDA_CUDA_BINARY_OBJECTIVE_HPP_
#ifdef USE_CUDA
...
...
@@ -60,4 +60,4 @@ class CUDABinaryLogloss : public CUDAObjectiveInterface<BinaryLogloss> {
#endif // USE_CUDA
#endif // LIGHTGBM_OBJECTIVE_CUDA_CUDA_BINARY_OBJECTIVE_HPP_
#endif // LIGHTGBM_
SRC_
OBJECTIVE_CUDA_CUDA_BINARY_OBJECTIVE_HPP_
src/objective/cuda/cuda_multiclass_objective.hpp
View file @
548cec82
...
...
@@ -2,8 +2,8 @@
* Copyright (c) 2021 Microsoft Corporation. All rights reserved.
* Licensed under the MIT License. See LICENSE file in the project root for license information.
*/
#ifndef LIGHTGBM_OBJECTIVE_CUDA_CUDA_MULTICLASS_OBJECTIVE_HPP_
#define LIGHTGBM_OBJECTIVE_CUDA_CUDA_MULTICLASS_OBJECTIVE_HPP_
#ifndef LIGHTGBM_
SRC_
OBJECTIVE_CUDA_CUDA_MULTICLASS_OBJECTIVE_HPP_
#define LIGHTGBM_
SRC_
OBJECTIVE_CUDA_CUDA_MULTICLASS_OBJECTIVE_HPP_
#ifdef USE_CUDA
...
...
@@ -75,4 +75,4 @@ class CUDAMulticlassOVA: public CUDAObjectiveInterface<MulticlassOVA> {
}
// namespace LightGBM
#endif // USE_CUDA
#endif // LIGHTGBM_OBJECTIVE_CUDA_CUDA_MULTICLASS_OBJECTIVE_HPP_
#endif // LIGHTGBM_
SRC_
OBJECTIVE_CUDA_CUDA_MULTICLASS_OBJECTIVE_HPP_
src/objective/cuda/cuda_rank_objective.hpp
View file @
548cec82
...
...
@@ -4,8 +4,8 @@
* license information.
*/
#ifndef LIGHTGBM_OBJECTIVE_CUDA_CUDA_RANK_OBJECTIVE_HPP_
#define LIGHTGBM_OBJECTIVE_CUDA_CUDA_RANK_OBJECTIVE_HPP_
#ifndef LIGHTGBM_
SRC_
OBJECTIVE_CUDA_CUDA_RANK_OBJECTIVE_HPP_
#define LIGHTGBM_
SRC_
OBJECTIVE_CUDA_CUDA_RANK_OBJECTIVE_HPP_
#ifdef USE_CUDA
...
...
@@ -119,4 +119,4 @@ class CUDARankXENDCG : public CUDALambdaRankObjectiveInterface<RankXENDCG> {
}
// namespace LightGBM
#endif // USE_CUDA
#endif // LIGHTGBM_OBJECTIVE_CUDA_CUDA_RANK_OBJECTIVE_HPP_
#endif // LIGHTGBM_
SRC_
OBJECTIVE_CUDA_CUDA_RANK_OBJECTIVE_HPP_
src/objective/cuda/cuda_regression_objective.hpp
View file @
548cec82
...
...
@@ -4,8 +4,8 @@
* license information.
*/
#ifndef LIGHTGBM_OBJECTIVE_CUDA_CUDA_REGRESSION_OBJECTIVE_HPP_
#define LIGHTGBM_OBJECTIVE_CUDA_CUDA_REGRESSION_OBJECTIVE_HPP_
#ifndef LIGHTGBM_
SRC_
OBJECTIVE_CUDA_CUDA_REGRESSION_OBJECTIVE_HPP_
#define LIGHTGBM_
SRC_
OBJECTIVE_CUDA_CUDA_REGRESSION_OBJECTIVE_HPP_
#ifdef USE_CUDA
...
...
@@ -164,4 +164,4 @@ class CUDARegressionQuantileloss : public CUDARegressionObjectiveInterface<Regre
}
// namespace LightGBM
#endif // USE_CUDA
#endif // LIGHTGBM_OBJECTIVE_CUDA_CUDA_REGRESSION_OBJECTIVE_HPP_
#endif // LIGHTGBM_
SRC_
OBJECTIVE_CUDA_CUDA_REGRESSION_OBJECTIVE_HPP_
src/objective/multiclass_objective.hpp
View file @
548cec82
...
...
@@ -2,8 +2,8 @@
* Copyright (c) 2016 Microsoft Corporation. All rights reserved.
* Licensed under the MIT License. See LICENSE file in the project root for license information.
*/
#ifndef LIGHTGBM_OBJECTIVE_MULTICLASS_OBJECTIVE_HPP_
#define LIGHTGBM_OBJECTIVE_MULTICLASS_OBJECTIVE_HPP_
#ifndef LIGHTGBM_
SRC_
OBJECTIVE_MULTICLASS_OBJECTIVE_HPP_
#define LIGHTGBM_
SRC_
OBJECTIVE_MULTICLASS_OBJECTIVE_HPP_
#include <LightGBM/network.h>
#include <LightGBM/objective_function.h>
...
...
@@ -276,4 +276,4 @@ class MulticlassOVA: public ObjectiveFunction {
};
}
// namespace LightGBM
#endif // L
ightGBM
_OBJECTIVE_MULTICLASS_OBJECTIVE_HPP_
#endif // L
IGHTGBM_SRC
_OBJECTIVE_MULTICLASS_OBJECTIVE_HPP_
src/objective/objective_function.cpp
View file @
548cec82
...
...
@@ -4,6 +4,8 @@
*/
#include <LightGBM/objective_function.h>
#include <string>
#include "binary_objective.hpp"
#include "multiclass_objective.hpp"
#include "rank_objective.hpp"
...
...
src/objective/rank_objective.hpp
View file @
548cec82
...
...
@@ -3,8 +3,8 @@
* Licensed under the MIT License. See LICENSE file in the project root for
* license information.
*/
#ifndef LIGHTGBM_OBJECTIVE_RANK_OBJECTIVE_HPP_
#define LIGHTGBM_OBJECTIVE_RANK_OBJECTIVE_HPP_
#ifndef LIGHTGBM_
SRC_
OBJECTIVE_RANK_OBJECTIVE_HPP_
#define LIGHTGBM_
SRC_
OBJECTIVE_RANK_OBJECTIVE_HPP_
#include <LightGBM/metric.h>
#include <LightGBM/objective_function.h>
...
...
@@ -462,4 +462,4 @@ class RankXENDCG : public RankingObjective {
};
}
// namespace LightGBM
#endif // L
ightGBM
_OBJECTIVE_RANK_OBJECTIVE_HPP_
#endif // L
IGHTGBM_SRC
_OBJECTIVE_RANK_OBJECTIVE_HPP_
src/objective/regression_objective.hpp
View file @
548cec82
...
...
@@ -2,8 +2,8 @@
* Copyright (c) 2016 Microsoft Corporation. All rights reserved.
* Licensed under the MIT License. See LICENSE file in the project root for license information.
*/
#ifndef LIGHTGBM_OBJECTIVE_REGRESSION_OBJECTIVE_HPP_
#define LIGHTGBM_OBJECTIVE_REGRESSION_OBJECTIVE_HPP_
#ifndef LIGHTGBM_
SRC_
OBJECTIVE_REGRESSION_OBJECTIVE_HPP_
#define LIGHTGBM_
SRC_
OBJECTIVE_REGRESSION_OBJECTIVE_HPP_
#include <LightGBM/meta.h>
#include <LightGBM/objective_function.h>
...
...
@@ -760,4 +760,4 @@ class RegressionTweedieLoss: public RegressionPoissonLoss {
#undef WeightedPercentileFun
}
// namespace LightGBM
#endif // L
ightGBM
_OBJECTIVE_REGRESSION_OBJECTIVE_HPP_
#endif // L
IGHTGBM_SRC
_OBJECTIVE_REGRESSION_OBJECTIVE_HPP_
src/objective/xentropy_objective.hpp
View file @
548cec82
...
...
@@ -2,8 +2,8 @@
* Copyright (c) 2017 Microsoft Corporation. All rights reserved.
* Licensed under the MIT License. See LICENSE file in the project root for license information.
*/
#ifndef LIGHTGBM_OBJECTIVE_XENTROPY_OBJECTIVE_HPP_
#define LIGHTGBM_OBJECTIVE_XENTROPY_OBJECTIVE_HPP_
#ifndef LIGHTGBM_
SRC_
OBJECTIVE_XENTROPY_OBJECTIVE_HPP_
#define LIGHTGBM_
SRC_
OBJECTIVE_XENTROPY_OBJECTIVE_HPP_
#include <LightGBM/meta.h>
#include <LightGBM/objective_function.h>
...
...
@@ -313,4 +313,4 @@ class CrossEntropyLambda: public ObjectiveFunction {
}
// end namespace LightGBM
#endif
//
end #ifndef
LIGHTGBM_OBJECTIVE_XENTROPY_OBJECTIVE_HPP_
#endif // LIGHTGBM_
SRC_
OBJECTIVE_XENTROPY_OBJECTIVE_HPP_
src/treelearner/col_sampler.hpp
View file @
548cec82
...
...
@@ -3,8 +3,8 @@
* Licensed under the MIT License. See LICENSE file in the project root for
* license information.
*/
#ifndef LIGHTGBM_TREELEARNER_COL_SAMPLER_HPP_
#define LIGHTGBM_TREELEARNER_COL_SAMPLER_HPP_
#ifndef LIGHTGBM_
SRC_
TREELEARNER_COL_SAMPLER_HPP_
#define LIGHTGBM_
SRC_
TREELEARNER_COL_SAMPLER_HPP_
#include <LightGBM/dataset.h>
#include <LightGBM/meta.h>
...
...
@@ -204,4 +204,4 @@ class ColSampler {
};
}
// namespace LightGBM
#endif // LIGHTGBM_TREELEARNER_COL_SAMPLER_HPP_
#endif // LIGHTGBM_
SRC_
TREELEARNER_COL_SAMPLER_HPP_
src/treelearner/cost_effective_gradient_boosting.hpp
View file @
548cec82
...
...
@@ -3,8 +3,8 @@
* Licensed under the MIT License. See LICENSE file in the project root for
* license information.
*/
#ifndef LIGHTGBM_TREELEARNER_COST_EFFECTIVE_GRADIENT_BOOSTING_HPP_
#define LIGHTGBM_TREELEARNER_COST_EFFECTIVE_GRADIENT_BOOSTING_HPP_
#ifndef LIGHTGBM_
SRC_
TREELEARNER_COST_EFFECTIVE_GRADIENT_BOOSTING_HPP_
#define LIGHTGBM_
SRC_
TREELEARNER_COST_EFFECTIVE_GRADIENT_BOOSTING_HPP_
#include <LightGBM/config.h>
#include <LightGBM/dataset.h>
...
...
@@ -171,4 +171,4 @@ class CostEfficientGradientBoosting {
}
// namespace LightGBM
#endif // LIGHTGBM_TREELEARNER_COST_EFFECTIVE_GRADIENT_BOOSTING_HPP_
#endif // LIGHTGBM_
SRC_
TREELEARNER_COST_EFFECTIVE_GRADIENT_BOOSTING_HPP_
src/treelearner/cuda/cuda_best_split_finder.cpp
View file @
548cec82
...
...
@@ -7,6 +7,7 @@
#ifdef USE_CUDA
#include <algorithm>
#include <vector>
#include "cuda_best_split_finder.hpp"
#include "cuda_leaf_splits.hpp"
...
...
src/treelearner/cuda/cuda_best_split_finder.cu
View file @
548cec82
...
...
@@ -13,6 +13,7 @@
#include <LightGBM/cuda/cuda_rocm_interop.h>
#include <algorithm>
#include <vector>
namespace
LightGBM
{
...
...
src/treelearner/cuda/cuda_best_split_finder.hpp
View file @
548cec82
...
...
@@ -4,8 +4,8 @@
* license information.
*/
#ifndef LIGHTGBM_TREELEARNER_CUDA_CUDA_BEST_SPLIT_FINDER_HPP_
#define LIGHTGBM_TREELEARNER_CUDA_CUDA_BEST_SPLIT_FINDER_HPP_
#ifndef LIGHTGBM_
SRC_
TREELEARNER_CUDA_CUDA_BEST_SPLIT_FINDER_HPP_
#define LIGHTGBM_
SRC_
TREELEARNER_CUDA_CUDA_BEST_SPLIT_FINDER_HPP_
#ifdef USE_CUDA
...
...
@@ -241,4 +241,4 @@ class CUDABestSplitFinder {
}
// namespace LightGBM
#endif // USE_CUDA
#endif // LIGHTGBM_TREELEARNER_CUDA_CUDA_BEST_SPLIT_FINDER_HPP_
#endif // LIGHTGBM_
SRC_
TREELEARNER_CUDA_CUDA_BEST_SPLIT_FINDER_HPP_
src/treelearner/cuda/cuda_data_partition.cpp
View file @
548cec82
...
...
@@ -8,6 +8,7 @@
#include <algorithm>
#include <memory>
#include <vector>
#include "cuda_data_partition.hpp"
...
...
src/treelearner/cuda/cuda_data_partition.hpp
View file @
548cec82
...
...
@@ -3,8 +3,8 @@
* Licensed under the MIT License. See LICENSE file in the project root for
* license information.
*/
#ifndef LIGHTGBM_TREELEARNER_CUDA_CUDA_DATA_PARTITION_HPP_
#define LIGHTGBM_TREELEARNER_CUDA_CUDA_DATA_PARTITION_HPP_
#ifndef LIGHTGBM_
SRC_
TREELEARNER_CUDA_CUDA_DATA_PARTITION_HPP_
#define LIGHTGBM_
SRC_
TREELEARNER_CUDA_CUDA_DATA_PARTITION_HPP_
#ifdef USE_CUDA
...
...
@@ -393,4 +393,4 @@ class CUDADataPartition {
}
// namespace LightGBM
#endif // USE_CUDA
#endif // LIGHTGBM_TREELEARNER_CUDA_CUDA_DATA_PARTITION_HPP_
#endif // LIGHTGBM_
SRC_
TREELEARNER_CUDA_CUDA_DATA_PARTITION_HPP_
Prev
1
…
4
5
6
7
8
9
10
Next
Write
Preview
Markdown
is supported
0%
Try again
or
attach a new file
.
Attach a file
Cancel
You are about to add
0
people
to the discussion. Proceed with caution.
Finish editing this message first!
Cancel
Please
register
or
sign in
to comment