context.hpp

/*
 * SPDX-FileCopyrightText: Copyright (c) 2025-2026, NVIDIA CORPORATION.
 * SPDX-License-Identifier: Apache-2.0
 */
 
#pragma once
 
#include <cudf/utilities/export.hpp>
 
#include <cstdint>
#include <type_traits>
 
namespace CUDF_EXPORT cudf {
 
enum class init_flags : std::uint32_t {
  NONE = 0,
  LOAD_NVCOMP = 1 << 0,
  INIT_JIT_CACHE = 1 << 1,
  DEFAULT = INIT_JIT_CACHE,
  ALL = LOAD_NVCOMP | INIT_JIT_CACHE
};
 
constexpr init_flags operator|(init_flags lhs, init_flags rhs) noexcept
{
  using underlying_t = std::underlying_type_t<init_flags>;
  return static_cast<init_flags>(static_cast<underlying_t>(lhs) | static_cast<underlying_t>(rhs));
}
 
constexpr init_flags operator&(init_flags lhs, init_flags rhs) noexcept
{
  using underlying_t = std::underlying_type_t<init_flags>;
  return static_cast<init_flags>(static_cast<underlying_t>(lhs) & static_cast<underlying_t>(rhs));
}
 
constexpr init_flags operator~(init_flags flags) noexcept
{
  using underlying_t = std::underlying_type_t<init_flags>;
  return static_cast<init_flags>(static_cast<underlying_t>(init_flags::ALL) &
                                 ~static_cast<underlying_t>(flags));
}
 
constexpr bool has_flag(init_flags flags, init_flags flag) noexcept
{
  return (flags | flag) == flags;
}
 
void initialize(init_flags flags = init_flags::DEFAULT);
 
void teardown();
 
void enable_jit_cache(bool enable);
 
void clear_jit_cache();
 
}  // namespace CUDF_EXPORT cudf