context.hpp

/*
 * Copyright (c) 2025, NVIDIA CORPORATION.
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 *     http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */
 
#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,
  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::INIT_JIT_CACHE);
 
void deinitialize();
 
}  // namespace CUDF_EXPORT cudf