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

increase batch size

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