mindspore2022/mindspore/ccsrc/include/common/visible.h

/**
 * Copyright 2022 Huawei Technologies Co., Ltd
 *
 * 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.
 */

#ifndef MINDSPORE_CCSRC_INCLUDE_COMMON_VISIBLE_H_
#define MINDSPORE_CCSRC_INCLUDE_COMMON_VISIBLE_H_

#include <utility>

#if (defined(_WIN32) || defined(__WIN32__) || defined(WIN32) || defined(__CYGWIN__))
#ifdef COMMON_DLL
#define COMMON_EXPORT __declspec(dllexport)
#else
#define COMMON_EXPORT __declspec(dllimport)
#endif
#define COMMON_LOCAL
#else
#define COMMON_EXPORT __attribute__((visibility("default")))
#define COMMON_LOCAL __attribute__((visibility("hidden")))
#endif

#define HANDLER_DEFINE(return_type, name, args...)                                                \
 public:                                                                                          \
  template <typename... Args>                                                                     \
  static return_type name(const Args &... argss) {                                                \
    if (name##_handler_ == nullptr) {                                                             \
      return return_type();                                                                       \
    }                                                                                             \
    return name##_handler_(argss...);                                                             \
  }                                                                                               \
  using name##Handler = std::function<decltype(name<args>)>;                                      \
  static void Set##name##Handler(name##Handler handler) { name##_handler_ = std::move(handler); } \
                                                                                                  \
 private:                                                                                         \
  inline static name##Handler name##_handler_;
#endif  // MINDSPORE_CCSRC_INCLUDE_COMMON_VISIBLE_H_