108 lines
2.6 KiB
C++
108 lines
2.6 KiB
C++
/* Copyright 2019-2021 Canaan Inc.
|
|
*
|
|
* 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 <gsl/gsl-lite.hpp>
|
|
#include <type_traits>
|
|
|
|
#if defined(_MSC_VER)
|
|
#ifdef NNCASE_DLL
|
|
#define NNCASE_API __declspec(dllexport)
|
|
#elif defined(NNCASE_SHARED_LIBS)
|
|
#define NNCASE_API __declspec(dllimport)
|
|
#else
|
|
#define NNCASE_API
|
|
#endif
|
|
#else
|
|
#define NNCASE_API
|
|
#endif
|
|
|
|
#if defined(_MSC_VER)
|
|
#define NNCASE_UNREACHABLE() __assume(0)
|
|
#else
|
|
#define NNCASE_UNREACHABLE() __builtin_unreachable()
|
|
#endif
|
|
|
|
#if gsl_CPP17_OR_GREATER
|
|
#define NNCASE_INLINE_VAR inline
|
|
#define NNCASE_UNUSED [[maybe_unused]]
|
|
namespace nncase
|
|
{
|
|
template <class Callable, class... Args>
|
|
using invoke_result_t = std::invoke_result_t<Callable, Args...>;
|
|
}
|
|
#else
|
|
#define NNCASE_INLINE_VAR
|
|
#if defined(_MSC_VER)
|
|
#define NNCASE_UNUSED
|
|
#else
|
|
#define NNCASE_UNUSED __attribute__((unused))
|
|
#endif
|
|
namespace nncase
|
|
{
|
|
template <class Callable, class... Args>
|
|
using invoke_result_t = std::result_of_t<Callable(Args...)>;
|
|
}
|
|
#endif
|
|
|
|
#define NNCASE_LITTLE_ENDIAN 1
|
|
|
|
#define NNCASE_HAVE_STD_BYTE gsl_CPP17_OR_GREATER
|
|
#define NNCASE_NODISCARD gsl_NODISCARD
|
|
#define NNCASE_NORETURN gsl_NORETURN
|
|
|
|
#define BEGIN_NS_NNCASE_RUNTIME \
|
|
namespace nncase \
|
|
{ \
|
|
namespace runtime \
|
|
{
|
|
#define END_NS_NNCASE_RUNTIME \
|
|
} \
|
|
}
|
|
|
|
#define BEGIN_NS_NNCASE_RT_STACKVM \
|
|
namespace nncase \
|
|
{ \
|
|
namespace runtime \
|
|
{ \
|
|
namespace stackvm \
|
|
{
|
|
#define END_NS_NNCASE_RT_STACKVM \
|
|
} \
|
|
} \
|
|
}
|
|
|
|
#define BEGIN_NS_NNCASE_KERNELS \
|
|
namespace nncase \
|
|
{ \
|
|
namespace kernels \
|
|
{
|
|
|
|
#define END_NS_NNCASE_KERNELS \
|
|
} \
|
|
}
|
|
|
|
#ifndef DEFINE_ENUM_BITMASK_OPERATORS
|
|
#define DEFINE_ENUM_BITMASK_OPERATORS(ENUMTYPE) gsl_DEFINE_ENUM_BITMASK_OPERATORS(ENUMTYPE)
|
|
#endif
|
|
|
|
namespace nncase
|
|
{
|
|
struct default_init_t
|
|
{
|
|
};
|
|
|
|
NNCASE_INLINE_VAR constexpr default_init_t default_init {};
|
|
}
|