Skip to content
Snippets Groups Projects
Commit cc586720 authored by Sebastian Eibl's avatar Sebastian Eibl
Browse files

increase batch size

parent 1cf8f66c
Branches
No related tags found
No related merge requests found
...@@ -122,7 +122,7 @@ void SISSOSolver::l0_regularization(const int n_dim) ...@@ -122,7 +122,7 @@ void SISSOSolver::l0_regularization(const int n_dim)
while (!feature_combinations.is_finished()) while (!feature_combinations.is_finished())
{ {
const size_t MAX_BATCH_SIZE = 128; const size_t MAX_BATCH_SIZE = 4096;
std::vector<std::vector<int>> feature_indices; std::vector<std::vector<int>> feature_indices;
for (auto counter = 0; counter < MAX_BATCH_SIZE; ++counter) for (auto counter = 0; counter < MAX_BATCH_SIZE; ++counter)
{ {
......
...@@ -42,7 +42,7 @@ public: ...@@ -42,7 +42,7 @@ public:
using PropertyView = Kokkos::View<double**, Kokkos::LayoutLeft>; using PropertyView = Kokkos::View<double**, Kokkos::LayoutLeft>;
protected: protected:
static constexpr int MAX_BATCHES = 128; static constexpr int MAX_BATCHES = 4096;
/// dim 0: material samples /// dim 0: material samples
/// dim 1: features /// dim 1: features
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment