gbrl 1.0.0.dev2__tar.gz → 1.0.0.dev3__tar.gz
This diff represents the content of publicly available package versions that have been released to one of the supported registries. The information contained in this diff is provided for informational purposes only and reflects changes between package versions as they appear in their respective public registries.
- {gbrl-1.0.0.dev2/gbrl.egg-info → gbrl-1.0.0.dev3}/PKG-INFO +1 -1
- {gbrl-1.0.0.dev2 → gbrl-1.0.0.dev3}/gbrl/src/cpp/utils.cpp +1 -1
- {gbrl-1.0.0.dev2 → gbrl-1.0.0.dev3/gbrl.egg-info}/PKG-INFO +1 -1
- {gbrl-1.0.0.dev2 → gbrl-1.0.0.dev3}/pyproject.toml +1 -1
- {gbrl-1.0.0.dev2 → gbrl-1.0.0.dev3}/setup.py +1 -8
- {gbrl-1.0.0.dev2 → gbrl-1.0.0.dev3}/LICENSE +0 -0
- {gbrl-1.0.0.dev2 → gbrl-1.0.0.dev3}/MANIFEST.in +0 -0
- {gbrl-1.0.0.dev2 → gbrl-1.0.0.dev3}/README.md +0 -0
- {gbrl-1.0.0.dev2 → gbrl-1.0.0.dev3}/gbrl/src/cpp/fitter.cpp +0 -0
- {gbrl-1.0.0.dev2 → gbrl-1.0.0.dev3}/gbrl/src/cpp/fitter.h +0 -0
- {gbrl-1.0.0.dev2 → gbrl-1.0.0.dev3}/gbrl/src/cpp/gbrl.cpp +0 -0
- {gbrl-1.0.0.dev2 → gbrl-1.0.0.dev3}/gbrl/src/cpp/gbrl.h +0 -0
- {gbrl-1.0.0.dev2 → gbrl-1.0.0.dev3}/gbrl/src/cpp/gbrl_binding.cpp +0 -0
- {gbrl-1.0.0.dev2 → gbrl-1.0.0.dev3}/gbrl/src/cpp/loss.cpp +0 -0
- {gbrl-1.0.0.dev2 → gbrl-1.0.0.dev3}/gbrl/src/cpp/loss.h +0 -0
- {gbrl-1.0.0.dev2 → gbrl-1.0.0.dev3}/gbrl/src/cpp/main.cpp +0 -0
- {gbrl-1.0.0.dev2 → gbrl-1.0.0.dev3}/gbrl/src/cpp/math_ops.cpp +0 -0
- {gbrl-1.0.0.dev2 → gbrl-1.0.0.dev3}/gbrl/src/cpp/math_ops.h +0 -0
- {gbrl-1.0.0.dev2 → gbrl-1.0.0.dev3}/gbrl/src/cpp/node.cpp +0 -0
- {gbrl-1.0.0.dev2 → gbrl-1.0.0.dev3}/gbrl/src/cpp/node.h +0 -0
- {gbrl-1.0.0.dev2 → gbrl-1.0.0.dev3}/gbrl/src/cpp/optimizer.cpp +0 -0
- {gbrl-1.0.0.dev2 → gbrl-1.0.0.dev3}/gbrl/src/cpp/optimizer.h +0 -0
- {gbrl-1.0.0.dev2 → gbrl-1.0.0.dev3}/gbrl/src/cpp/predictor.cpp +0 -0
- {gbrl-1.0.0.dev2 → gbrl-1.0.0.dev3}/gbrl/src/cpp/predictor.h +0 -0
- {gbrl-1.0.0.dev2 → gbrl-1.0.0.dev3}/gbrl/src/cpp/scheduler.cpp +0 -0
- {gbrl-1.0.0.dev2 → gbrl-1.0.0.dev3}/gbrl/src/cpp/scheduler.h +0 -0
- {gbrl-1.0.0.dev2 → gbrl-1.0.0.dev3}/gbrl/src/cpp/split_candidate_generator.cpp +0 -0
- {gbrl-1.0.0.dev2 → gbrl-1.0.0.dev3}/gbrl/src/cpp/split_candidate_generator.h +0 -0
- {gbrl-1.0.0.dev2 → gbrl-1.0.0.dev3}/gbrl/src/cpp/types.cpp +0 -0
- {gbrl-1.0.0.dev2 → gbrl-1.0.0.dev3}/gbrl/src/cpp/types.h +0 -0
- {gbrl-1.0.0.dev2 → gbrl-1.0.0.dev3}/gbrl/src/cpp/utils.h +0 -0
- {gbrl-1.0.0.dev2 → gbrl-1.0.0.dev3}/gbrl/src/cuda/cuda_fitter.cu +0 -0
- {gbrl-1.0.0.dev2 → gbrl-1.0.0.dev3}/gbrl/src/cuda/cuda_fitter.h +0 -0
- {gbrl-1.0.0.dev2 → gbrl-1.0.0.dev3}/gbrl/src/cuda/cuda_loss.cu +0 -0
- {gbrl-1.0.0.dev2 → gbrl-1.0.0.dev3}/gbrl/src/cuda/cuda_loss.h +0 -0
- {gbrl-1.0.0.dev2 → gbrl-1.0.0.dev3}/gbrl/src/cuda/cuda_predictor.cu +0 -0
- {gbrl-1.0.0.dev2 → gbrl-1.0.0.dev3}/gbrl/src/cuda/cuda_predictor.h +0 -0
- {gbrl-1.0.0.dev2 → gbrl-1.0.0.dev3}/gbrl/src/cuda/cuda_preprocess.cu +0 -0
- {gbrl-1.0.0.dev2 → gbrl-1.0.0.dev3}/gbrl/src/cuda/cuda_preprocess.h +0 -0
- {gbrl-1.0.0.dev2 → gbrl-1.0.0.dev3}/gbrl/src/cuda/cuda_types.cu +0 -0
- {gbrl-1.0.0.dev2 → gbrl-1.0.0.dev3}/gbrl/src/cuda/cuda_types.h +0 -0
- {gbrl-1.0.0.dev2 → gbrl-1.0.0.dev3}/gbrl/src/cuda/cuda_utils.cu +0 -0
- {gbrl-1.0.0.dev2 → gbrl-1.0.0.dev3}/gbrl/src/cuda/cuda_utils.h +0 -0
- {gbrl-1.0.0.dev2 → gbrl-1.0.0.dev3}/gbrl.egg-info/SOURCES.txt +0 -0
- {gbrl-1.0.0.dev2 → gbrl-1.0.0.dev3}/gbrl.egg-info/dependency_links.txt +0 -0
- {gbrl-1.0.0.dev2 → gbrl-1.0.0.dev3}/gbrl.egg-info/requires.txt +0 -0
- {gbrl-1.0.0.dev2 → gbrl-1.0.0.dev3}/gbrl.egg-info/top_level.txt +0 -0
- {gbrl-1.0.0.dev2 → gbrl-1.0.0.dev3}/setup.cfg +0 -0
- {gbrl-1.0.0.dev2 → gbrl-1.0.0.dev3}/tests/test_gbt_multi.py +0 -0
- {gbrl-1.0.0.dev2 → gbrl-1.0.0.dev3}/tests/test_gbt_single.py +0 -0
|
@@ -21,7 +21,7 @@ std::string VectoString(const float* vec, const int vec_size){
|
|
|
21
21
|
|
|
22
22
|
int binaryToDecimal(const BoolVector& binaryPath) {
|
|
23
23
|
int decimal = 0;
|
|
24
|
-
for (size_t i = binaryPath.size() - 1, j = 0; i
|
|
24
|
+
for (size_t i = binaryPath.size() - 1, j = 0; i-- > 0; ++j) {
|
|
25
25
|
decimal += binaryPath[i] * (1 << j);
|
|
26
26
|
}
|
|
27
27
|
return decimal + (1 << binaryPath.size()) - 1;
|
|
@@ -230,7 +230,7 @@ class CustomBuildExt(build_ext):
|
|
|
230
230
|
Path(os.path.dirname(target)).mkdir(parents=True, exist_ok=True)
|
|
231
231
|
command = [nvcc_path] + nvcc_compile_args + ['-c', source, '-o', target]
|
|
232
232
|
print(' '.join(command))
|
|
233
|
-
subprocess.check_call(command
|
|
233
|
+
subprocess.check_call(command)
|
|
234
234
|
# Manually adding the compiled CUDA object files to be linked
|
|
235
235
|
for idx, _ in enumerate(self.extensions):
|
|
236
236
|
cuda_extras = [os.path.join(build_dir, f.replace('.cu', '.o')) for f in cuda_source_files]
|
|
@@ -240,13 +240,6 @@ class CustomBuildExt(build_ext):
|
|
|
240
240
|
setup(
|
|
241
241
|
name="gbrl",
|
|
242
242
|
version="1.0.0",
|
|
243
|
-
install_requires=["pybind11==2.11.1",
|
|
244
|
-
"numpy",
|
|
245
|
-
"torch",
|
|
246
|
-
],
|
|
247
|
-
extras_require={
|
|
248
|
-
'sklearn': ['scikit_learn==1.2.2'], # Optional for unittests
|
|
249
|
-
},
|
|
250
243
|
packages=find_packages(include=["gbrl.*"], exclude=("tests*",)),
|
|
251
244
|
package_data={
|
|
252
245
|
'gbrl_cpp': ['*.so', '*.pyd'] # Include SO/PYD files
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|
|
File without changes
|