diff --git a/cpp/config.hpp b/cpp/config.hpp index e7f3db5..7ca32f2 100644 --- a/cpp/config.hpp +++ b/cpp/config.hpp @@ -14,12 +14,19 @@ __device__ constexpr const size_t M = 5; //log2(NB_THREADS_2D_Y)); #endif -// Save state to avoid recalulation on restart +// Save state to avoid recalculation on restart #define SAVE_STATE true // Redo the state even if it's already saved #define FORCE_REDO false + // Use GPU to greatly accelerate runtime #define GPU_BOOSTED true +// Depending on what you set, the output label will be as follow : +// | GPU_BOOSTED | LABEL | +// |-------------|-------| +// | true | GPU | +// | false | CPU | + // Number of weak classifiers // const size_t TS[] = { 1 }; // const size_t TS[] = { 1, 5, 10 }; diff --git a/python/config.py b/python/config.py index 5a6bbdb..2f8b871 100644 --- a/python/config.py +++ b/python/config.py @@ -9,7 +9,7 @@ NB_THREADS_2D = (32, 32) NB_THREADS_3D = (16, 16, 4) M = int(np.log2(NB_THREADS_2D[1])) -# Save state to avoid recalulation on restart +# Save state to avoid recalculation on restart SAVE_STATE = True # Redo the state even if it's already saved FORCE_REDO = False @@ -17,6 +17,14 @@ FORCE_REDO = False COMPILE_WITH_C = True # Use GPU to greatly accelerate runtime (as priority over NJIT) GPU_BOOSTED = True +# Depending on what you set, the output label will be as follow : +# | COMPILE_WITH_C | GPU_BOOSTED | LABEL | +# |----------------|-------------|-------| +# | True | True | GPU | +# | True | False | CPU | +# | False | True | PGPU | +# | False | False | PY | + # Number of weak classifiers # TS = [1] # TS = [1, 5, 10]