fastx_host_limits.cuh

Go to the documentation of this file.

#pragma once
 
#include <algorithm>
#include <cstddef>
#include <cstdint>
#include <limits>
 
#include <cusbf/detail/host_parse.hpp>
 
namespace cusbf::detail {
 
static_assert(sizeof(size_t) == sizeof(uint64_t), "cuSBF assumes size_t is 64-bit");
 
[[nodiscard]] inline size_t parse_host_chunk_max_bytes(const char* env_name) {
    const uint64_t mebibytes = parse_env_mebibytes(getenv_value(env_name));
    if (mebibytes == 0) {
        return 0;
    }
    return static_cast<size_t>(mebibytes) << 20;
}
 
[[nodiscard]] inline size_t fastx_host_chunk_max_bytes() {
    if (const size_t override_bytes = parse_host_chunk_max_bytes("CUSBF_FASTX_MAX_HOST_CHUNK_MB");
        override_bytes != 0) {
        return override_bytes;
    }
 
    if (const size_t large_bytes = parse_host_chunk_max_bytes("CUSBF_LARGE_FASTX_HOST_CHUNK_MB");
        large_bytes != 0) {
        return large_bytes;
    }
 
    return std::numeric_limits<size_t>::max();
}
 
[[nodiscard]] inline constexpr bool fastx_chunk_reached_host_byte_limit(
    size_t host_chunk_max_bytes,
    uint64_t raw_chunk_bytes
) noexcept {
    return host_chunk_max_bytes != std::numeric_limits<size_t>::max() &&
           raw_chunk_bytes >= host_chunk_max_bytes;
}
 
}  // namespace cusbf::detail