cuda_utils.h

Go to the documentation of this file.

// ----------------------------------------------------------------------------
// -                        CloudViewer: www.cloudViewer.org                  -
// ----------------------------------------------------------------------------
// Copyright (c) 2018-2024 www.cloudViewer.org
// SPDX-License-Identifier: MIT
// ----------------------------------------------------------------------------
 
#pragma once
 
#include <cmath>
 
namespace cloudViewer {
namespace ml {
namespace contrib {
 
#define TOTAL_THREADS 1024
#define THREADS_PER_BLOCK 256
#define DIVUP(m, n) ((m) / (n) + ((m) % (n) > 0))
 
inline int OptNumThreads(int work_size) {
    const int pow_2 = std::log(static_cast<double>(work_size)) / std::log(2.0);
 
    return max(min(1 << pow_2, TOTAL_THREADS), 1);
}
 
inline dim3 OptBlockConfig(int x, int y) {
    const int x_threads = OptNumThreads(x);
    const int y_threads =
            max(min(OptNumThreads(y), TOTAL_THREADS / x_threads), 1);
    dim3 block_config(x_threads, y_threads, 1);
    return block_config;
}
 
}  // namespace contrib
}  // namespace ml
}  // namespace cloudViewer