mindspore lite: update header files protect macro
This commit is contained in:
parent
515e54d361
commit
511a0d64d7
|
@ -14,8 +14,8 @@
|
|||
* limitations under the License.
|
||||
*/
|
||||
|
||||
#ifndef MINDSPORE_LITE_CONVOLUTION_FP32_H_
|
||||
#define MINDSPORE_LITE_CONVOLUTION_FP32_H_
|
||||
#ifndef MINDSPORE_LITE_EXPERIMENTAL_KERNEL_CONVOLUTION_FP32_H_
|
||||
#define MINDSPORE_LITE_EXPERIMENTAL_KERNEL_CONVOLUTION_FP32_H_
|
||||
|
||||
#include <vector>
|
||||
#include "src/runtime/lite_kernel.h"
|
||||
|
@ -39,4 +39,4 @@ class ConvolutionCPUFp32 : public LiteKernel {
|
|||
};
|
||||
} // namespace mindspore::kernel
|
||||
|
||||
#endif // MINDSPORE_LITE_SRC_RUNTIME_KERNEL_ARM_FP32_CONVOLUTION_FP32_H_
|
||||
#endif // MINDSPORE_LITE_EXPERIMENTAL_KERNEL_CONVOLUTION_FP32_H_
|
||||
|
|
|
@ -14,8 +14,8 @@
|
|||
* limitations under the License.
|
||||
*/
|
||||
|
||||
#ifndef MINDSPORE_LITE_SRC_TENSOR_H_
|
||||
#define MINDSPORE_LITE_SRC_TENSOR_H_
|
||||
#ifndef MINDSPORE_LITE_EXPERIMENTAL_SRC_TENSOR_H_
|
||||
#define MINDSPORE_LITE_EXPERIMENTAL_SRC_TENSOR_H_
|
||||
|
||||
#include <math.h>
|
||||
#include <memory>
|
||||
|
@ -259,4 +259,4 @@ std::vector<tensor::MSTensor *> TensorVectorCast(const std::vector<Tensor *> &sr
|
|||
} // namespace mindspore
|
||||
|
||||
using TensorPtr = std::shared_ptr<mindspore::lite::Tensor>;
|
||||
#endif // MINDSPORE_LITE_SRC_TENSOR_H_
|
||||
#endif // MINDSPORE_LITE_EXPERIMENTAL_SRC_TENSOR_H_
|
||||
|
|
|
@ -14,8 +14,8 @@
|
|||
* limitations under the License.
|
||||
*/
|
||||
|
||||
#ifndef MINDSPORE_LITE_INCLUDE_REGISTRY_PARSER_CONTEXT_H_
|
||||
#define MINDSPORE_LITE_INCLUDE_REGISTRY_PARSER_CONTEXT_H_
|
||||
#ifndef MINDSPORE_LITE_INCLUDE_REGISTRY_CONVERTER_CONTEXT_H_
|
||||
#define MINDSPORE_LITE_INCLUDE_REGISTRY_CONVERTER_CONTEXT_H_
|
||||
|
||||
#include <map>
|
||||
#include <string>
|
||||
|
@ -83,4 +83,4 @@ class MS_API ConverterContext {
|
|||
} // namespace converter
|
||||
} // namespace mindspore
|
||||
|
||||
#endif // MINDSPORE_LITE_INCLUDE_REGISTRY_PARSER_CONTEXT_H_
|
||||
#endif // MINDSPORE_LITE_INCLUDE_REGISTRY_CONVERTER_CONTEXT_H_
|
||||
|
|
|
@ -14,8 +14,8 @@
|
|||
* limitations under the License.
|
||||
*/
|
||||
|
||||
#ifndef MINDSPORE_LITE_INCLUDE_REGISTRY_MODEL_PARSER_REGISTRY_H
|
||||
#define MINDSPORE_LITE_INCLUDE_REGISTRY_MODEL_PARSER_REGISTRY_H
|
||||
#ifndef MINDSPORE_LITE_INCLUDE_REGISTRY_MODEL_PARSER_REGISTRY_H_
|
||||
#define MINDSPORE_LITE_INCLUDE_REGISTRY_MODEL_PARSER_REGISTRY_H_
|
||||
|
||||
#include <memory>
|
||||
#include "include/registry/converter_context.h"
|
||||
|
@ -58,4 +58,4 @@ class MS_API ModelParserRegistry {
|
|||
} // namespace registry
|
||||
} // namespace mindspore
|
||||
|
||||
#endif // MINDSPORE_LITE_INCLUDE_REGISTRY_MODEL_PARSER_REGISTRY_H
|
||||
#endif // MINDSPORE_LITE_INCLUDE_REGISTRY_MODEL_PARSER_REGISTRY_H_
|
||||
|
|
|
@ -14,8 +14,8 @@
|
|||
* limitations under the License.
|
||||
*/
|
||||
|
||||
#ifndef MINDSPORE_LITE_INCLUDE_REGISTRY_OPENCL_RUNTIME_WRAPPER_H
|
||||
#define MINDSPORE_LITE_INCLUDE_REGISTRY_OPENCL_RUNTIME_WRAPPER_H
|
||||
#ifndef MINDSPORE_LITE_INCLUDE_REGISTRY_OPENCL_RUNTIME_WRAPPER_H_
|
||||
#define MINDSPORE_LITE_INCLUDE_REGISTRY_OPENCL_RUNTIME_WRAPPER_H_
|
||||
|
||||
#include <vector>
|
||||
#include <map>
|
||||
|
@ -135,4 +135,4 @@ Status OpenCLRuntimeWrapper::BuildKernel(cl::Kernel *kernel, const std::string &
|
|||
VectorStringToChar(build_options_ext));
|
||||
}
|
||||
} // namespace mindspore::registry::opencl
|
||||
#endif // MINDSPORE_LITE_INCLUDE_REGISTRY_OPENCL_RUNTIME_WRAPPER_H
|
||||
#endif // MINDSPORE_LITE_INCLUDE_REGISTRY_OPENCL_RUNTIME_WRAPPER_H_
|
||||
|
|
|
@ -14,8 +14,8 @@
|
|||
* limitations under the License.
|
||||
*/
|
||||
|
||||
#ifndef MINDSPORE_LITE_INCLUDE_REGISTRY_KERNEL_INTERFACE_H_
|
||||
#define MINDSPORE_LITE_INCLUDE_REGISTRY_KERNEL_INTERFACE_H_
|
||||
#ifndef MINDSPORE_LITE_INCLUDE_REGISTRY_REGISTER_KERNEL_INTERFACE_H_
|
||||
#define MINDSPORE_LITE_INCLUDE_REGISTRY_REGISTER_KERNEL_INTERFACE_H_
|
||||
|
||||
#include <set>
|
||||
#include <string>
|
||||
|
@ -136,4 +136,4 @@ std::shared_ptr<kernel::KernelInterface> RegisterKernelInterface::GetKernelInter
|
|||
} // namespace registry
|
||||
} // namespace mindspore
|
||||
|
||||
#endif // MINDSPORE_LITE_INCLUDE_REGISTRY_KERNEL_INTERFACE_H_
|
||||
#endif // MINDSPORE_LITE_INCLUDE_REGISTRY_REGISTER_KERNEL_INTERFACE_H_
|
||||
|
|
|
@ -14,10 +14,10 @@
|
|||
* limitations under the License.
|
||||
*/
|
||||
|
||||
#ifndef MINDSPORE_LITE_JAVA_SRC_COMMON_JNI_UTILS_H
|
||||
#define MINDSPORE_LITE_JAVA_SRC_COMMON_JNI_UTILS_H
|
||||
#ifndef MINDSPORE_LITE_JAVA_NATIVE_COMMON_JNI_UTILS_H_
|
||||
#define MINDSPORE_LITE_JAVA_NATIVE_COMMON_JNI_UTILS_H_
|
||||
#include <string>
|
||||
|
||||
std::string RealPath(const char *path);
|
||||
|
||||
#endif // MINDSPORE_LITE_JAVA_SRC_COMMON_JNI_UTILS_H
|
||||
#endif // MINDSPORE_LITE_JAVA_NATIVE_COMMON_JNI_UTILS_H_
|
||||
|
|
|
@ -14,8 +14,8 @@
|
|||
* limitations under the License.
|
||||
*/
|
||||
|
||||
#ifndef MINDSPORE_LITE_JAVA_SRC_COMMON_MS_LOG_H
|
||||
#define MINDSPORE_LITE_JAVA_SRC_COMMON_MS_LOG_H
|
||||
#ifndef MINDSPORE_LITE_JAVA_NATIVE_COMMON_MS_LOG_H_
|
||||
#define MINDSPORE_LITE_JAVA_NATIVE_COMMON_MS_LOG_H_
|
||||
|
||||
#define TAG "MS_LITE"
|
||||
|
||||
|
@ -42,4 +42,4 @@
|
|||
{ printf("[INFO] %s|%s|%s[%d]|: " #fmt "\r\n", TAG, __FILE__, __func__, __LINE__, ##args); }
|
||||
#endif
|
||||
|
||||
#endif // MINDSPORE_LITE_JAVA_SRC_COMMON_MS_LOG_H
|
||||
#endif // MINDSPORE_LITE_JAVA_NATIVE_COMMON_MS_LOG_H_
|
||||
|
|
|
@ -13,8 +13,8 @@
|
|||
* See the License for the specific language governing permissions and
|
||||
* limitations under the License.
|
||||
*/
|
||||
#ifndef DATASET_MDTODAPI_H_
|
||||
#define DATASET_MDTODAPI_H_
|
||||
#ifndef MINDSPORE_LITE_MINDDATA_WRAPPER_MDTODAPI_H_
|
||||
#define MINDSPORE_LITE_MINDDATA_WRAPPER_MDTODAPI_H_
|
||||
|
||||
#include <stdint.h>
|
||||
#include <sys/types.h>
|
||||
|
|
|
@ -14,8 +14,8 @@
|
|||
* limitations under the License.
|
||||
*/
|
||||
|
||||
#ifndef DPICO_COMMON_CHECK_BASE_H_
|
||||
#define DPICO_COMMON_CHECK_BASE_H_
|
||||
#ifndef MINDSPORE_LITE_PROVIDERS_DPICO_COMMON_CHECK_BASE_H_
|
||||
#define MINDSPORE_LITE_PROVIDERS_DPICO_COMMON_CHECK_BASE_H_
|
||||
|
||||
#include <limits.h>
|
||||
#include "common/log_util.h"
|
||||
|
@ -84,4 +84,4 @@
|
|||
} \
|
||||
} while (0)
|
||||
|
||||
#endif // DPICO_COMMON_CHECK_BASE_H_
|
||||
#endif // MINDSPORE_LITE_PROVIDERS_DPICO_COMMON_CHECK_BASE_H_
|
||||
|
|
|
@ -14,8 +14,8 @@
|
|||
* limitations under the License.
|
||||
*/
|
||||
|
||||
#ifndef DPICO_COMMON_CUSTOM_ENUM_H_
|
||||
#define DPICO_COMMON_CUSTOM_ENUM_H_
|
||||
#ifndef MINDSPORE_LITE_PROVIDERS_DPICO_COMMON_CUSTOM_ENUM_H_
|
||||
#define MINDSPORE_LITE_PROVIDERS_DPICO_COMMON_CUSTOM_ENUM_H_
|
||||
|
||||
namespace mindspore {
|
||||
namespace lite {
|
||||
|
@ -33,4 +33,4 @@ enum DetectBoxParam : size_t {
|
|||
};
|
||||
} // namespace lite
|
||||
} // namespace mindspore
|
||||
#endif // DPICO_COMMON_CUSTOM_ENUM_H_
|
||||
#endif // MINDSPORE_LITE_PROVIDERS_DPICO_COMMON_CUSTOM_ENUM_H_
|
||||
|
|
|
@ -14,8 +14,8 @@
|
|||
* limitations under the License.
|
||||
*/
|
||||
|
||||
#ifndef DPICO_COMMON_INFER_UTIL_H_
|
||||
#define DPICO_COMMON_INFER_UTIL_H_
|
||||
#ifndef MINDSPORE_LITE_PROVIDERS_DPICO_COMMON_INFER_UTIL_H_
|
||||
#define MINDSPORE_LITE_PROVIDERS_DPICO_COMMON_INFER_UTIL_H_
|
||||
|
||||
#include <string>
|
||||
#include <vector>
|
||||
|
@ -30,4 +30,4 @@ int CheckCustomParam(const schema::Custom *param, const std::string ¶m_name)
|
|||
} // namespace lite
|
||||
} // namespace mindspore
|
||||
|
||||
#endif // DPICO_COMMON_INFER_UTIL_H_
|
||||
#endif // MINDSPORE_LITE_PROVIDERS_DPICO_COMMON_INFER_UTIL_H_
|
||||
|
|
|
@ -14,8 +14,8 @@
|
|||
* limitations under the License.
|
||||
*/
|
||||
|
||||
#ifndef DPICO_COMMON_LOG_UTIL_H_
|
||||
#define DPICO_COMMON_LOG_UTIL_H_
|
||||
#ifndef MINDSPORE_LITE_PROVIDERS_DPICO_COMMON_LOG_UTIL_H_
|
||||
#define MINDSPORE_LITE_PROVIDERS_DPICO_COMMON_LOG_UTIL_H_
|
||||
|
||||
#include <memory>
|
||||
#include <sstream>
|
||||
|
@ -97,4 +97,4 @@ class DpicoLogWriter {
|
|||
#define MS_LOG_ERROR MSLOG_IF(mindspore::DpicoLogLevel::ERROR)
|
||||
} // namespace mindspore
|
||||
|
||||
#endif // DPICO_COMMON_LOG_UTIL_H_
|
||||
#endif // MINDSPORE_LITE_PROVIDERS_DPICO_COMMON_LOG_UTIL_H_
|
||||
|
|
|
@ -14,8 +14,8 @@
|
|||
* limitations under the License.
|
||||
*/
|
||||
|
||||
#ifndef DPICO_COMMON_OP_ATTR_H_
|
||||
#define DPICO_COMMON_OP_ATTR_H_
|
||||
#ifndef MINDSPORE_LITE_PROVIDERS_DPICO_COMMON_OP_ATTR_H_
|
||||
#define MINDSPORE_LITE_PROVIDERS_DPICO_COMMON_OP_ATTR_H_
|
||||
|
||||
namespace mindspore {
|
||||
constexpr auto kAclConfigPath = "AclConfigPath";
|
||||
|
@ -32,4 +32,4 @@ constexpr auto kOutputsFormat = "outputs_format";
|
|||
constexpr auto kOutputsShape = "outputs_shape";
|
||||
constexpr auto kScoreThreshold = "ScoreThreshold";
|
||||
} // namespace mindspore
|
||||
#endif // DPICO_COMMON_OP_ATTR_H_
|
||||
#endif // MINDSPORE_LITE_PROVIDERS_DPICO_COMMON_OP_ATTR_H_
|
||||
|
|
|
@ -14,8 +14,8 @@
|
|||
* limitations under the License.
|
||||
*/
|
||||
|
||||
#ifndef DPICO_COMMON_STRING_UTIL_H_
|
||||
#define DPICO_COMMON_STRING_UTIL_H_
|
||||
#ifndef MINDSPORE_LITE_PROVIDERS_DPICO_COMMON_STRING_UTIL_H_
|
||||
#define MINDSPORE_LITE_PROVIDERS_DPICO_COMMON_STRING_UTIL_H_
|
||||
|
||||
#include <string>
|
||||
#include <vector>
|
||||
|
@ -33,4 +33,4 @@ bool IsValidUnsignedNum(const std::string &num_str);
|
|||
bool IsValidDoubleNum(const std::string &num_str);
|
||||
} // namespace lite
|
||||
} // namespace mindspore
|
||||
#endif // DPICO_COMMON_STRING_UTIL_H_
|
||||
#endif // MINDSPORE_LITE_PROVIDERS_DPICO_COMMON_STRING_UTIL_H_
|
||||
|
|
|
@ -14,8 +14,8 @@
|
|||
* limitations under the License.
|
||||
*/
|
||||
|
||||
#ifndef DPICO_INFER_CUSTOM_INFER_H_
|
||||
#define DPICO_INFER_CUSTOM_INFER_H_
|
||||
#ifndef MINDSPORE_LITE_PROVIDERS_DPICO_INFER_CUSTOM_INFER_H_
|
||||
#define MINDSPORE_LITE_PROVIDERS_DPICO_INFER_CUSTOM_INFER_H_
|
||||
|
||||
#include <vector>
|
||||
#include "include/kernel_interface.h"
|
||||
|
@ -34,4 +34,4 @@ class CustomInterface : public kernel::KernelInterface {
|
|||
} // namespace dpico
|
||||
} // namespace mindspore
|
||||
|
||||
#endif // DPICO_INFER_CUSTOM_INFER_H_
|
||||
#endif // MINDSPORE_LITE_PROVIDERS_DPICO_INFER_CUSTOM_INFER_H_
|
||||
|
|
|
@ -14,8 +14,8 @@
|
|||
* limitations under the License.
|
||||
*/
|
||||
|
||||
#ifndef DPICO_MANAGER_ACL_CONTEXT_MANAGER_H_
|
||||
#define DPICO_MANAGER_ACL_CONTEXT_MANAGER_H_
|
||||
#ifndef MINDSPORE_LITE_PROVIDERS_DPICO_MANAGER_ACL_CONTEXT_MANAGER_H_
|
||||
#define MINDSPORE_LITE_PROVIDERS_DPICO_MANAGER_ACL_CONTEXT_MANAGER_H_
|
||||
|
||||
#include <memory>
|
||||
#include <map>
|
||||
|
@ -40,4 +40,4 @@ class AclContextManager {
|
|||
using AclContextManagerPtr = std::shared_ptr<AclContextManager>;
|
||||
} // namespace lite
|
||||
} // namespace mindspore
|
||||
#endif // DPICO_MANAGER_ACL_CONTEXT_MANAGER_H_
|
||||
#endif // MINDSPORE_LITE_PROVIDERS_DPICO_MANAGER_ACL_CONTEXT_MANAGER_H_
|
||||
|
|
|
@ -14,8 +14,8 @@
|
|||
* limitations under the License.
|
||||
*/
|
||||
|
||||
#ifndef DPICO_MANAGER_ACL_MODEL_HELPER_H_
|
||||
#define DPICO_MANAGER_ACL_MODEL_HELPER_H_
|
||||
#ifndef MINDSPORE_LITE_PROVIDERS_DPICO_MANAGER_ACL_MODEL_HELPER_H_
|
||||
#define MINDSPORE_LITE_PROVIDERS_DPICO_MANAGER_ACL_MODEL_HELPER_H_
|
||||
|
||||
#include <vector>
|
||||
#include <unordered_map>
|
||||
|
@ -48,4 +48,4 @@ int WriteDetBoxesToTensorData(const std::vector<std::vector<float>> &det_boxes,
|
|||
int AclMalloc(void **buf, size_t size);
|
||||
} // namespace lite
|
||||
} // namespace mindspore
|
||||
#endif // DPICO_MANAGER_ACL_MODEL_HELPER_H_
|
||||
#endif // MINDSPORE_LITE_PROVIDERS_DPICO_MANAGER_ACL_MODEL_HELPER_H_
|
||||
|
|
|
@ -14,8 +14,8 @@
|
|||
* limitations under the License.
|
||||
*/
|
||||
|
||||
#ifndef DPICO_MANAGER_ACL_MODEL_MANAGER_H_
|
||||
#define DPICO_MANAGER_ACL_MODEL_MANAGER_H_
|
||||
#ifndef MINDSPORE_LITE_PROVIDERS_DPICO_MANAGER_ACL_MODEL_MANAGER_H_
|
||||
#define MINDSPORE_LITE_PROVIDERS_DPICO_MANAGER_ACL_MODEL_MANAGER_H_
|
||||
|
||||
#include <memory>
|
||||
#include <vector>
|
||||
|
@ -83,4 +83,4 @@ class AclModelManager {
|
|||
using AclModelManagerPtr = std::shared_ptr<AclModelManager>;
|
||||
} // namespace lite
|
||||
} // namespace mindspore
|
||||
#endif // DPICO_MANAGER_ACL_MODEL_MANAGER_H_
|
||||
#endif // MINDSPORE_LITE_PROVIDERS_DPICO_MANAGER_ACL_MODEL_MANAGER_H_
|
||||
|
|
|
@ -14,8 +14,8 @@
|
|||
* limitations under the License.
|
||||
*/
|
||||
|
||||
#ifndef DPICO_MANAGER_CUSTOM_CONFIG_MANAGER_H_
|
||||
#define DPICO_MANAGER_CUSTOM_CONFIG_MANAGER_H_
|
||||
#ifndef MINDSPORE_LITE_PROVIDERS_DPICO_MANAGER_CUSTOM_CONFIG_MANAGER_H_
|
||||
#define MINDSPORE_LITE_PROVIDERS_DPICO_MANAGER_CUSTOM_CONFIG_MANAGER_H_
|
||||
|
||||
#include <map>
|
||||
#include <string>
|
||||
|
@ -54,4 +54,4 @@ using CustomConfigManagerPtr = std::shared_ptr<CustomConfigManager>;
|
|||
} // namespace lite
|
||||
} // namespace mindspore
|
||||
|
||||
#endif // DPICO_MANAGER_CUSTOM_CONFIG_MANAGER_H_
|
||||
#endif // MINDSPORE_LITE_PROVIDERS_DPICO_MANAGER_CUSTOM_CONFIG_MANAGER_H_
|
||||
|
|
|
@ -14,8 +14,8 @@
|
|||
* limitations under the License.
|
||||
*/
|
||||
|
||||
#ifndef DPICO_SRC_CUSTOM_ALLOCATOR_H_
|
||||
#define DPICO_SRC_CUSTOM_ALLOCATOR_H_
|
||||
#ifndef MINDSPORE_LITE_PROVIDERS_DPICO_SRC_CUSTOM_ALLOCATOR_H_
|
||||
#define MINDSPORE_LITE_PROVIDERS_DPICO_SRC_CUSTOM_ALLOCATOR_H_
|
||||
|
||||
#include <memory>
|
||||
#include <string>
|
||||
|
@ -73,4 +73,4 @@ constexpr int64_t MAX_MALLOC_SIZE = static_cast<size_t>(2000) * 1024 * 1024;
|
|||
constexpr int64_t MAX_THREAD_POOL_SIZE = static_cast<size_t>(3000) * 1024 * 1024;
|
||||
} // namespace lite
|
||||
} // namespace mindspore
|
||||
#endif // DPICO_SRC_CUSTOM_ALLOCATOR_H_
|
||||
#endif // MINDSPORE_LITE_PROVIDERS_DPICO_SRC_CUSTOM_ALLOCATOR_H_
|
||||
|
|
|
@ -14,8 +14,8 @@
|
|||
* limitations under the License.
|
||||
*/
|
||||
|
||||
#ifndef DPICO_SRC_CUSTOM_FP32_H_
|
||||
#define DPICO_SRC_CUSTOM_FP32_H_
|
||||
#ifndef MINDSPORE_LITE_PROVIDERS_DPICO_SRC_CUSTOM_FP32_H_
|
||||
#define MINDSPORE_LITE_PROVIDERS_DPICO_SRC_CUSTOM_FP32_H_
|
||||
|
||||
#include <vector>
|
||||
#include "include/api/kernel.h"
|
||||
|
@ -49,4 +49,4 @@ class CustomCPUKernel : public Kernel {
|
|||
} // namespace lite
|
||||
} // namespace mindspore
|
||||
|
||||
#endif // DPICO_SRC_CUSTOM_FP32_H_
|
||||
#endif // MINDSPORE_LITE_PROVIDERS_DPICO_SRC_CUSTOM_FP32_H_
|
||||
|
|
|
@ -14,8 +14,8 @@
|
|||
* limitations under the License.
|
||||
*/
|
||||
|
||||
#ifndef NNIE_SRC_CUSTOM_ALLOCATOR_H_
|
||||
#define NNIE_SRC_CUSTOM_ALLOCATOR_H_
|
||||
#ifndef MINDSPORE_LITE_PROVIDERS_NNIE_SRC_CUSTOM_ALLOCATOR_H_
|
||||
#define MINDSPORE_LITE_PROVIDERS_NNIE_SRC_CUSTOM_ALLOCATOR_H_
|
||||
|
||||
#include <memory>
|
||||
#include <string>
|
||||
|
@ -44,4 +44,4 @@ class CustomAllocator : public Allocator {
|
|||
} // namespace nnie
|
||||
} // namespace mindspore
|
||||
|
||||
#endif // NNIE_SRC_CUSTOM_ALLOCATOR_H_
|
||||
#endif // MINDSPORE_LITE_PROVIDERS_NNIE_SRC_CUSTOM_ALLOCATOR_H_
|
||||
|
|
|
@ -14,8 +14,8 @@
|
|||
* limitations under the License.
|
||||
*/
|
||||
|
||||
#ifndef MINDSPORE_LITE_SRC_RUNTIME_KERNEL_ARM_FP32_CUSTOM_H_
|
||||
#define MINDSPORE_LITE_SRC_RUNTIME_KERNEL_ARM_FP32_CUSTOM_H_
|
||||
#ifndef MINDSPORE_LITE_PROVIDERS_NNIE_SRC_CUSTOM_FP32_H_
|
||||
#define MINDSPORE_LITE_PROVIDERS_NNIE_SRC_CUSTOM_FP32_H_
|
||||
|
||||
#include <vector>
|
||||
#include <string>
|
||||
|
@ -69,4 +69,4 @@ class CustomCPUKernel : public Kernel {
|
|||
};
|
||||
} // namespace nnie
|
||||
} // namespace mindspore
|
||||
#endif // MINDSPORE_LITE_SRC_RUNTIME_KERNEL_ARM_FP32_CUSTOM_H_
|
||||
#endif // MINDSPORE_LITE_PROVIDERS_NNIE_SRC_CUSTOM_FP32_H_
|
||||
|
|
|
@ -13,8 +13,8 @@
|
|||
* See the License for the specific language governing permissions and
|
||||
* limitations under the License.
|
||||
*/
|
||||
#ifndef MINDSPORE_LITE_NNACL_CUSTOM_PARAMETER_H_
|
||||
#define MINDSPORE_LITE_NNACL_CUSTOM_PARAMETER_H_
|
||||
#ifndef MINDSPORE_LITE_PROVIDERS_NNIE_SRC_CUSTOM_INFER_H_
|
||||
#define MINDSPORE_LITE_PROVIDERS_NNIE_SRC_CUSTOM_INFER_H_
|
||||
#include <vector>
|
||||
#include <memory>
|
||||
#include "include/kernel_interface.h"
|
||||
|
@ -32,4 +32,4 @@ class CustomInterface : public mindspore::kernel::KernelInterface {
|
|||
};
|
||||
} // namespace nnie
|
||||
} // namespace mindspore
|
||||
#endif // MINDSPORE_LITE_NNACL_CUSTOM_PARAMETER_H_
|
||||
#endif // MINDSPORE_LITE_PROVIDERS_NNIE_SRC_CUSTOM_INFER_H_
|
||||
|
|
|
@ -13,8 +13,8 @@
|
|||
* See the License for the specific language governing permissions and
|
||||
* limitations under the License.
|
||||
*/
|
||||
#ifndef MINDSPORE_LITE_TOOLS_BENCHMARK_NNIE_NNIE_CFG_PARSER_H_
|
||||
#define MINDSPORE_LITE_TOOLS_BENCHMARK_NNIE_NNIE_CFG_PARSER_H_
|
||||
#ifndef MINDSPORE_LITE_PROVIDERS_NNIE_SRC_NNIE_CFG_PARSER_H_
|
||||
#define MINDSPORE_LITE_PROVIDERS_NNIE_SRC_NNIE_CFG_PARSER_H_
|
||||
#include <vector>
|
||||
#include <map>
|
||||
#include <string>
|
||||
|
|
|
@ -14,8 +14,8 @@
|
|||
* limitations under the License.
|
||||
*/
|
||||
|
||||
#ifndef MINDSPORE_LITE_SRC_RUNTIME_AGENT_NNIE_NNIE_COMMON_H_
|
||||
#define MINDSPORE_LITE_SRC_RUNTIME_AGENT_NNIE_NNIE_COMMON_H_
|
||||
#ifndef MINDSPORE_LITE_PROVIDERS_NNIE_SRC_NNIE_COMMON_H_
|
||||
#define MINDSPORE_LITE_PROVIDERS_NNIE_SRC_NNIE_COMMON_H_
|
||||
#include <iostream>
|
||||
#include <string>
|
||||
#include <vector>
|
||||
|
@ -119,4 +119,4 @@ int NnieCommGetOutputData(NnieRunCfg *nnie_run_cfg, float *data, int64_t *shape,
|
|||
HI_U32 GetBlobSize(const SVP_SRC_BLOB_S &blob);
|
||||
} // namespace nnie
|
||||
} // namespace mindspore
|
||||
#endif // MINDSPORE_LITE_SRC_RUNTIME_AGENT_NNIE_NNIE_COMMON_H_
|
||||
#endif // MINDSPORE_LITE_PROVIDERS_NNIE_SRC_NNIE_COMMON_H_
|
||||
|
|
|
@ -14,8 +14,8 @@
|
|||
* limitations under the License.
|
||||
*/
|
||||
|
||||
#ifndef MINDSPORE_LITE_SRC_RUNTIME_AGENT_NNIE_NNIE_MANAGER_H_
|
||||
#define MINDSPORE_LITE_SRC_RUNTIME_AGENT_NNIE_NNIE_MANAGER_H_
|
||||
#ifndef MINDSPORE_LITE_PROVIDERS_NNIE_SRC_NNIE_MANAGER_H_
|
||||
#define MINDSPORE_LITE_PROVIDERS_NNIE_SRC_NNIE_MANAGER_H_
|
||||
#include <vector>
|
||||
#include <string>
|
||||
#include <map>
|
||||
|
@ -114,4 +114,4 @@ class NNIEManager {
|
|||
};
|
||||
} // namespace nnie
|
||||
} // namespace mindspore
|
||||
#endif // MINDSPORE_LITE_SRC_RUNTIME_AGENT_NNIE_NNIE_MANAGER_H_
|
||||
#endif // MINDSPORE_LITE_PROVIDERS_NNIE_SRC_NNIE_MANAGER_H_
|
||||
|
|
|
@ -13,8 +13,8 @@
|
|||
* See the License for the specific language governing permissions and
|
||||
* limitations under the License.
|
||||
*/
|
||||
#ifndef MINDSPORE_LITE_SRC_RUNTIME_AGENT_NNIE_NNIE_MEMORY_H_
|
||||
#define MINDSPORE_LITE_SRC_RUNTIME_AGENT_NNIE_NNIE_MEMORY_H_
|
||||
#ifndef MINDSPORE_LITE_PROVIDERS_NNIE_SRC_NNIE_MEMORY_H_
|
||||
#define MINDSPORE_LITE_PROVIDERS_NNIE_SRC_NNIE_MEMORY_H_
|
||||
#include <fcntl.h>
|
||||
#include <sys/mman.h>
|
||||
#include <sys/stat.h>
|
||||
|
@ -47,4 +47,4 @@ HI_S32 NnieMemFlushCache(HI_U64 phy_addr, HI_VOID *pv_vir_addr, HI_U32 size);
|
|||
HI_S32 NnieGetVirMemInfo(HI_U64 pv_vir_addr, HI_U64 *phy_addr);
|
||||
} // namespace nnie
|
||||
} // namespace mindspore
|
||||
#endif // MINDSPORE_LITE_SRC_RUNTIME_AGENT_NNIE_NNIE_MEMORY_H_
|
||||
#endif // MINDSPORE_LITE_PROVIDERS_NNIE_SRC_NNIE_MEMORY_H_
|
||||
|
|
|
@ -13,8 +13,8 @@
|
|||
* See the License for the specific language governing permissions and
|
||||
* limitations under the License.
|
||||
*/
|
||||
#ifndef MINDSPORE_LITE_SRC_RUNTIME_AGENT_NNIE_NNIE_PRINT_H_
|
||||
#define MINDSPORE_LITE_SRC_RUNTIME_AGENT_NNIE_NNIE_PRINT_H_
|
||||
#ifndef MINDSPORE_LITE_PROVIDERS_NNIE_SRC_NNIE_PRINT_H_
|
||||
#define MINDSPORE_LITE_PROVIDERS_NNIE_SRC_NNIE_PRINT_H_
|
||||
#include "include/mpi_nnie.h"
|
||||
#include "include/hi_type.h"
|
||||
#include "src/nnie_common.h"
|
||||
|
@ -55,4 +55,4 @@ HI_S32 NniePrintReportResult(NnieParam *pst_nnie_param);
|
|||
HI_S32 NniePrintReportResultInputSeg(NnieParam *pst_nnie_param, int segnum);
|
||||
} // namespace nnie
|
||||
} // namespace mindspore
|
||||
#endif // MINDSPORE_LITE_SRC_RUNTIME_AGENT_NNIE_NNIE_PRINT_H_
|
||||
#endif // MINDSPORE_LITE_PROVIDERS_NNIE_SRC_NNIE_PRINT_H_
|
||||
|
|
|
@ -14,8 +14,8 @@
|
|||
* limitations under the License.
|
||||
*/
|
||||
|
||||
#ifndef MINDSPORE_LITE_TOOLS_BENCHMARK_NNIE_PROPOSAL_PROPOSAL_H_
|
||||
#define MINDSPORE_LITE_TOOLS_BENCHMARK_NNIE_PROPOSAL_PROPOSAL_H_
|
||||
#ifndef MINDSPORE_LITE_PROVIDERS_NNIE_PROPOSAL_SRC_PROPOSAL_H_
|
||||
#define MINDSPORE_LITE_PROVIDERS_NNIE_PROPOSAL_SRC_PROPOSAL_H_
|
||||
#include <vector>
|
||||
#include "include/api/types.h"
|
||||
|
||||
|
@ -92,4 +92,4 @@ int32_t ProposalRun(std::vector<mindspore::MSTensor> *inputs, std::vector<mindsp
|
|||
void ProposalDeInit(ProposalParam *param);
|
||||
} // namespace proposal
|
||||
} // namespace mindspore
|
||||
#endif // MINDSPORE_LITE_TOOLS_BENCHMARK_NNIE_PROPOSAL_PROPOSAL_H_
|
||||
#endif // MINDSPORE_LITE_PROVIDERS_NNIE_PROPOSAL_SRC_PROPOSAL_H_
|
||||
|
|
|
@ -14,8 +14,8 @@
|
|||
* limitations under the License.
|
||||
*/
|
||||
|
||||
#ifndef MINDSPORE_LITE_TOOLS_BENCHMARK_NNIE_PROPOSAL_PROPOSAL_FP32_H_
|
||||
#define MINDSPORE_LITE_TOOLS_BENCHMARK_NNIE_PROPOSAL_PROPOSAL_FP32_H_
|
||||
#ifndef MINDSPORE_LITE_PROVIDERS_NNIE_PROPOSAL_SRC_PROPOSAL_FP32_H_
|
||||
#define MINDSPORE_LITE_PROVIDERS_NNIE_PROPOSAL_SRC_PROPOSAL_FP32_H_
|
||||
|
||||
#include <vector>
|
||||
#include "schema/model_generated.h"
|
||||
|
@ -48,4 +48,4 @@ class ProposalCPUKernel : public Kernel {
|
|||
} // namespace proposal
|
||||
} // namespace mindspore
|
||||
|
||||
#endif // MINDSPORE_LITE_TOOLS_BENCHMARK_NNIE_PROPOSAL_PROPOSAL_FP32_H_
|
||||
#endif // MINDSPORE_LITE_PROVIDERS_NNIE_PROPOSAL_SRC_PROPOSAL_FP32_H_
|
||||
|
|
|
@ -14,8 +14,8 @@
|
|||
* limitations under the License.
|
||||
*/
|
||||
|
||||
#ifndef MINDSPORE_LITE_TOOLS_BENCHMARK_NNIE_PROPOSAL_PROPOSAL_INFER_H_
|
||||
#define MINDSPORE_LITE_TOOLS_BENCHMARK_NNIE_PROPOSAL_PROPOSAL_INFER_H_
|
||||
#ifndef MINDSPORE_LITE_PROVIDERS_NNIE_PROPOSAL_SRC_PROPOSAL_INFER_H_
|
||||
#define MINDSPORE_LITE_PROVIDERS_NNIE_PROPOSAL_SRC_PROPOSAL_INFER_H_
|
||||
#include <vector>
|
||||
#include "include/kernel_interface.h"
|
||||
|
||||
|
@ -32,4 +32,4 @@ class ProposalInterface : public mindspore::kernel::KernelInterface {
|
|||
};
|
||||
} // namespace proposal
|
||||
} // namespace mindspore
|
||||
#endif // MINDSPORE_LITE_TOOLS_BENCHMARK_NNIE_PROPOSAL_PROPOSAL_INFER_H_
|
||||
#endif // MINDSPORE_LITE_PROVIDERS_NNIE_PROPOSAL_SRC_PROPOSAL_INFER_H_
|
||||
|
|
|
@ -14,8 +14,8 @@
|
|||
* limitations under the License.
|
||||
*/
|
||||
|
||||
#ifndef MINDSPORE_CORE_UTILS_DECRYPT_H_
|
||||
#define MINDSPORE_CORE_UTILS_DECRYPT_H_
|
||||
#ifndef MINDSPORE_LITE_SRC_COMMON_DECRYPT_H_
|
||||
#define MINDSPORE_LITE_SRC_COMMON_DECRYPT_H_
|
||||
|
||||
#include <string>
|
||||
#include <memory>
|
||||
|
|
|
@ -13,8 +13,8 @@
|
|||
* See the License for the specific language governing permissions and
|
||||
* limitations under the License.
|
||||
*/
|
||||
#ifndef MINDSPORE_LITE_SRC_OPS_ANF_UTILS_H_
|
||||
#define MINDSPORE_LITE_SRC_OPS_ANF_UTILS_H_
|
||||
#ifndef MINDSPORE_LITE_SRC_COMMON_OPS_ANF_UTILS_H_
|
||||
#define MINDSPORE_LITE_SRC_COMMON_OPS_ANF_UTILS_H_
|
||||
|
||||
#include <memory>
|
||||
#include "src/common/ops/ops_utils.h"
|
||||
|
@ -26,4 +26,4 @@ std::unique_ptr<schema::PrimitiveT> GetPrimitiveT(const mindspore::AnfNodePtr &n
|
|||
}
|
||||
} // namespace mindspore
|
||||
#endif
|
||||
#endif // MINDSPORE_LITE_SRC_OPS_ANF_UTILS_H_
|
||||
#endif // MINDSPORE_LITE_SRC_COMMON_OPS_ANF_UTILS_H_
|
||||
|
|
|
@ -13,8 +13,8 @@
|
|||
* See the License for the specific language governing permissions and
|
||||
* limitations under the License.
|
||||
*/
|
||||
#ifndef MINDSPORE_LITE_SRC_OPS_OPS_DEF_H_
|
||||
#define MINDSPORE_LITE_SRC_OPS_OPS_DEF_H_
|
||||
#ifndef MINDSPORE_LITE_SRC_COMMON_OPS_OPS_DEF_H_
|
||||
#define MINDSPORE_LITE_SRC_COMMON_OPS_OPS_DEF_H_
|
||||
#include <string>
|
||||
#include <map>
|
||||
#include <memory>
|
||||
|
@ -181,4 +181,4 @@
|
|||
#define OP_SCHEMA_DEF_ONLY_END(OP)
|
||||
#endif
|
||||
|
||||
#endif // MINDSPORE_LITE_SRC_OPS_OPS_DEF_H_
|
||||
#endif // MINDSPORE_LITE_SRC_COMMON_OPS_OPS_DEF_H_
|
||||
|
|
|
@ -13,8 +13,8 @@
|
|||
* See the License for the specific language governing permissions and
|
||||
* limitations under the License.
|
||||
*/
|
||||
#ifndef MINDSPORE_LITE_SRC_OPS_OPS_FUNC_DECLARE_H_
|
||||
#define MINDSPORE_LITE_SRC_OPS_OPS_FUNC_DECLARE_H_
|
||||
#ifndef MINDSPORE_LITE_SRC_COMMON_OPS_OPS_FUNC_DECLARE_H_
|
||||
#define MINDSPORE_LITE_SRC_COMMON_OPS_OPS_FUNC_DECLARE_H_
|
||||
|
||||
#ifdef PRIMITIVE_WRITEABLE
|
||||
#include <memory>
|
||||
|
@ -490,4 +490,4 @@ FUNC_MSOP2SCHEMAOP_DECLARE(GroupNormFusion)
|
|||
#else
|
||||
#define FUNC_MSOP2SCHEMAOP_DECLARE(OP)
|
||||
#endif
|
||||
#endif // MINDSPORE_LITE_SRC_OPS_OPS_FUNC_DECLARE_H_
|
||||
#endif // MINDSPORE_LITE_SRC_COMMON_OPS_OPS_FUNC_DECLARE_H_
|
||||
|
|
|
@ -14,8 +14,8 @@
|
|||
* limitations under the License.
|
||||
*/
|
||||
|
||||
#ifndef MINDSPORE_LITE_SRC_OPS_MS_OPS_UTILS_H_
|
||||
#define MINDSPORE_LITE_SRC_OPS_MS_OPS_UTILS_H_
|
||||
#ifndef MINDSPORE_LITE_SRC_COMMON_OPS_OPS_UTILS_H_
|
||||
#define MINDSPORE_LITE_SRC_COMMON_OPS_OPS_UTILS_H_
|
||||
|
||||
#include <map>
|
||||
#include <string>
|
||||
|
@ -70,4 +70,4 @@ class RegistryMSOps {
|
|||
} // namespace mindspore
|
||||
#endif
|
||||
|
||||
#endif // MINDSPORE_LITE_SRC_OPS_MS_OPS_UTILS_H_
|
||||
#endif // MINDSPORE_LITE_SRC_COMMON_OPS_OPS_UTILS_H_
|
||||
|
|
|
@ -13,8 +13,8 @@
|
|||
* See the License for the specific language governing permissions and
|
||||
* limitations under the License.
|
||||
*/
|
||||
#ifndef MINDSPORE_LITE_SRC_OPS_POPULATE_ARITHMETIC_POPULATE_H_
|
||||
#define MINDSPORE_LITE_SRC_OPS_POPULATE_ARITHMETIC_POPULATE_H_
|
||||
#ifndef MINDSPORE_LITE_SRC_COMMON_OPS_POPULATE_ARITHMETIC_POPULATE_H_
|
||||
#define MINDSPORE_LITE_SRC_COMMON_OPS_POPULATE_ARITHMETIC_POPULATE_H_
|
||||
|
||||
#include "nnacl/arithmetic.h"
|
||||
|
||||
|
@ -24,4 +24,4 @@ ArithmeticParameter *PopulateArithmeticCommonPara(const void *primitive);
|
|||
OpParameter *PopulateArithmetic(const void *primitive);
|
||||
} // namespace lite
|
||||
} // namespace mindspore
|
||||
#endif // MINDSPORE_LITE_SRC_OPS_POPULATE_ARITHMETIC_POPULATE_H_
|
||||
#endif // MINDSPORE_LITE_SRC_COMMON_OPS_POPULATE_ARITHMETIC_POPULATE_H_
|
||||
|
|
|
@ -13,8 +13,8 @@
|
|||
* See the License for the specific language governing permissions and
|
||||
* limitations under the License.
|
||||
*/
|
||||
#ifndef MINDSPORE_LITE_SRC_OPS_POPULATE_DEFAULT_POPULATE_H_
|
||||
#define MINDSPORE_LITE_SRC_OPS_POPULATE_DEFAULT_POPULATE_H_
|
||||
#ifndef MINDSPORE_LITE_SRC_COMMON_OPS_POPULATE_DEFAULT_POPULATE_H_
|
||||
#define MINDSPORE_LITE_SRC_COMMON_OPS_POPULATE_DEFAULT_POPULATE_H_
|
||||
|
||||
#include "nnacl/op_base.h"
|
||||
|
||||
|
@ -23,4 +23,4 @@ namespace lite {
|
|||
OpParameter *DefaultPopulateParameter(const void *prim);
|
||||
} // namespace lite
|
||||
} // namespace mindspore
|
||||
#endif // MINDSPORE_LITE_SRC_OPS_POPULATE_DEFAULT_POPULATE_H_
|
||||
#endif // MINDSPORE_LITE_SRC_COMMON_OPS_POPULATE_DEFAULT_POPULATE_H_
|
||||
|
|
|
@ -14,8 +14,8 @@
|
|||
* limitations under the License.
|
||||
*/
|
||||
|
||||
#ifndef MINDSPORE_LITE_SRC_OPS_POPULATE_POPULATE_REGISTER_H_
|
||||
#define MINDSPORE_LITE_SRC_OPS_POPULATE_POPULATE_REGISTER_H_
|
||||
#ifndef MINDSPORE_LITE_SRC_COMMON_OPS_POPULATE_POPULATE_REGISTER_H_
|
||||
#define MINDSPORE_LITE_SRC_COMMON_OPS_POPULATE_POPULATE_REGISTER_H_
|
||||
|
||||
#include <map>
|
||||
#include <vector>
|
||||
|
@ -85,4 +85,4 @@ class Registry {
|
|||
|
||||
} // namespace lite
|
||||
} // namespace mindspore
|
||||
#endif // MINDSPORE_LITE_SRC_OPS_POPULATE_POPULATE_REGISTER_H_
|
||||
#endif // MINDSPORE_LITE_SRC_COMMON_OPS_POPULATE_POPULATE_REGISTER_H_
|
||||
|
|
|
@ -14,8 +14,8 @@
|
|||
* limitations under the License.
|
||||
*/
|
||||
|
||||
#ifndef MINDSPORE_LITE_SRC_OPS_POPULATE_STRIDED_SLICE_POPULATE_H_
|
||||
#define MINDSPORE_LITE_SRC_OPS_POPULATE_STRIDED_SLICE_POPULATE_H_
|
||||
#ifndef MINDSPORE_LITE_SRC_COMMON_OPS_POPULATE_STRIDED_SLICE_POPULATE_H_
|
||||
#define MINDSPORE_LITE_SRC_COMMON_OPS_POPULATE_STRIDED_SLICE_POPULATE_H_
|
||||
|
||||
#include "src/common/ops/populate/populate_register.h"
|
||||
#include "nnacl/strided_slice_parameter.h"
|
||||
|
@ -25,4 +25,4 @@ namespace lite {
|
|||
OpParameter *PopulateStridedSliceParameter(const void *prim);
|
||||
} // namespace lite
|
||||
} // namespace mindspore
|
||||
#endif // MINDSPORE_LITE_SRC_OPS_POPULATE_STRIDED_SLICE_POPULATE_H_
|
||||
#endif // MINDSPORE_LITE_SRC_COMMON_OPS_POPULATE_STRIDED_SLICE_POPULATE_H_
|
||||
|
|
|
@ -13,8 +13,8 @@
|
|||
* See the License for the specific language governing permissions and
|
||||
* limitations under the License.
|
||||
*/
|
||||
#ifndef MINDSPORE_LITE_SRC_OPS_SCHEMA_REGISTER_H_
|
||||
#define MINDSPORE_LITE_SRC_OPS_SCHEMA_REGISTER_H_
|
||||
#ifndef MINDSPORE_LITE_SRC_COMMON_OPS_SCHEMA_REGISTER_H_
|
||||
#define MINDSPORE_LITE_SRC_COMMON_OPS_SCHEMA_REGISTER_H_
|
||||
#include <string>
|
||||
#include <vector>
|
||||
#include <functional>
|
||||
|
@ -57,4 +57,4 @@ class PrimitiveTypeRegister {
|
|||
};
|
||||
} // namespace mindspore::lite::ops
|
||||
|
||||
#endif // MINDSPORE_LITE_SRC_OPS_SCHEMA_REGISTER_H_
|
||||
#endif // MINDSPORE_LITE_SRC_COMMON_OPS_SCHEMA_REGISTER_H_
|
||||
|
|
|
@ -14,8 +14,8 @@
|
|||
* limitations under the License.
|
||||
*/
|
||||
|
||||
#ifndef MINDSPORE_LITE_SRC_COMMON_STORAGE_H
|
||||
#define MINDSPORE_LITE_SRC_COMMON_STORAGE_H
|
||||
#ifndef MINDSPORE_LITE_SRC_COMMON_STORAGE_H_
|
||||
#define MINDSPORE_LITE_SRC_COMMON_STORAGE_H_
|
||||
|
||||
#include <fstream>
|
||||
#include <string>
|
||||
|
|
|
@ -14,8 +14,8 @@
|
|||
* limitations under the License.
|
||||
*/
|
||||
|
||||
#ifndef MINDSPORE_LITE_SRC_COMMON_STRING_UTIL_H_
|
||||
#define MINDSPORE_LITE_SRC_COMMON_STRING_UTIL_H_
|
||||
#ifndef MINDSPORE_LITE_SRC_COMMON_STRING_UTILS_H_
|
||||
#define MINDSPORE_LITE_SRC_COMMON_STRING_UTILS_H_
|
||||
#include <vector>
|
||||
#include <string>
|
||||
#include <utility>
|
||||
|
@ -49,4 +49,4 @@ int GetStringCount(Tensor *tensor);
|
|||
uint64_t StringHash64(const char *s, size_t len);
|
||||
} // namespace lite
|
||||
} // namespace mindspore
|
||||
#endif // MINDSPORE_LITE_SRC_COMMON_STRING_UTIL_H_
|
||||
#endif // MINDSPORE_LITE_SRC_COMMON_STRING_UTILS_H_
|
||||
|
|
|
@ -14,8 +14,8 @@
|
|||
* limitations under the License.
|
||||
*/
|
||||
|
||||
#ifndef MINDSPORE_LITE_SRC_CONTROLFLOW_ACTOR_ENTRANCE_ACTOR_H_
|
||||
#define MINDSPORE_LITE_SRC_CONTROLFLOW_ACTOR_ENTRANCE_ACTOR_H_
|
||||
#ifndef MINDSPORE_LITE_SRC_CONTROL_FLOW_ACTOR_ENTRANCE_ACTOR_H_
|
||||
#define MINDSPORE_LITE_SRC_CONTROL_FLOW_ACTOR_ENTRANCE_ACTOR_H_
|
||||
#include <vector>
|
||||
#include <memory>
|
||||
#include <string>
|
||||
|
@ -42,4 +42,4 @@ class LiteEntranceOpActor : public LiteOpActor {
|
|||
AID entrance_input_aid_;
|
||||
};
|
||||
} // namespace mindspore::lite
|
||||
#endif // MINDSPORE_LITE_SRC_CONTROLFLOW_ACTOR_ENTRANCE_ACTOR_H_
|
||||
#endif // MINDSPORE_LITE_SRC_CONTROL_FLOW_ACTOR_ENTRANCE_ACTOR_H_
|
||||
|
|
|
@ -14,8 +14,8 @@
|
|||
* limitations under the License.
|
||||
*/
|
||||
|
||||
#ifndef MINDSPORE_LITE_SRC_CONTROLFLOW_ACTOR_EXIT_ACTOR_H_
|
||||
#define MINDSPORE_LITE_SRC_CONTROLFLOW_ACTOR_EXIT_ACTOR_H_
|
||||
#ifndef MINDSPORE_LITE_SRC_CONTROL_FLOW_ACTOR_EXIT_ACTOR_H_
|
||||
#define MINDSPORE_LITE_SRC_CONTROL_FLOW_ACTOR_EXIT_ACTOR_H_
|
||||
#include <vector>
|
||||
#include <memory>
|
||||
#include <string>
|
||||
|
@ -57,4 +57,4 @@ class LiteExitOpActor : public LiteOpActor {
|
|||
AID entrance_input_aid_;
|
||||
};
|
||||
} // namespace mindspore::lite
|
||||
#endif // MINDSPORE_LITE_SRC_CONTROLFLOW_ACTOR_EXIT_ACTOR_H_
|
||||
#endif // MINDSPORE_LITE_SRC_CONTROL_FLOW_ACTOR_EXIT_ACTOR_H_
|
||||
|
|
|
@ -14,8 +14,8 @@
|
|||
* limitations under the License.
|
||||
*/
|
||||
|
||||
#ifndef MINDSPORE_LITE_SRC_CONTROLFLOW_ACTOR_SWITCH_ACTOR_H_
|
||||
#define MINDSPORE_LITE_SRC_CONTROLFLOW_ACTOR_SWITCH_ACTOR_H_
|
||||
#ifndef MINDSPORE_LITE_SRC_CONTROL_FLOW_ACTOR_SWITCH_ACTOR_H_
|
||||
#define MINDSPORE_LITE_SRC_CONTROL_FLOW_ACTOR_SWITCH_ACTOR_H_
|
||||
#include <vector>
|
||||
#include <memory>
|
||||
#include <string>
|
||||
|
@ -73,4 +73,4 @@ class LiteSwitchOpActor : public LiteOpActor {
|
|||
std::vector<std::vector<OpDataPtr<Tensor>>> all_branchs_output_data_;
|
||||
};
|
||||
} // namespace mindspore::lite
|
||||
#endif // MINDSPORE_LITE_SRC_CONTROLFLOW_ACTOR_SWITCH_ACTOR_H_
|
||||
#endif // MINDSPORE_LITE_SRC_CONTROL_FLOW_ACTOR_SWITCH_ACTOR_H_
|
||||
|
|
|
@ -14,8 +14,8 @@
|
|||
* limitations under the License.
|
||||
*/
|
||||
|
||||
#ifndef MINDSPORE_LITE_SRC_CONTROLFLOW_CONTROL_ACTOR_CREATOR_H_
|
||||
#define MINDSPORE_LITE_SRC_CONTROLFLOW_CONTROL_ACTOR_CREATOR_H_
|
||||
#ifndef MINDSPORE_LITE_SRC_CONTROL_FLOW_CONTROL_ACTOR_CREATOR_H_
|
||||
#define MINDSPORE_LITE_SRC_CONTROL_FLOW_CONTROL_ACTOR_CREATOR_H_
|
||||
|
||||
#include <memory>
|
||||
#include "src/runtime/lite_kernel.h"
|
||||
|
@ -24,4 +24,4 @@
|
|||
namespace mindspore::lite {
|
||||
std::shared_ptr<LiteOpActor> CreateActor(kernel::KernelExec *kernel, lite::InnerContext *ctx);
|
||||
} // namespace mindspore::lite
|
||||
#endif // MINDSPORE_LITE_SRC_CONTROLFLOW_CONTROL_ACTOR_CREATOR_H_
|
||||
#endif // MINDSPORE_LITE_SRC_CONTROL_FLOW_CONTROL_ACTOR_CREATOR_H_
|
||||
|
|
|
@ -14,8 +14,8 @@
|
|||
* limitations under the License.
|
||||
*/
|
||||
|
||||
#ifndef MINDSPORE_LITE_SRC_CONTROL_FLOW_SCHEDULER_H_
|
||||
#define MINDSPORE_LITE_SRC_CONTROL_FLOW_SCHEDULER_H_
|
||||
#ifndef MINDSPORE_LITE_SRC_CONTROL_FLOW_CONTROL_FLOW_SCHEDULER_H_
|
||||
#define MINDSPORE_LITE_SRC_CONTROL_FLOW_CONTROL_FLOW_SCHEDULER_H_
|
||||
#include <string>
|
||||
#include <vector>
|
||||
#include <memory>
|
||||
|
@ -127,4 +127,4 @@ class ControlFlowScheduler {
|
|||
|
||||
using ControlFlowSchedulerPtr = std::shared_ptr<ControlFlowScheduler>;
|
||||
} // namespace mindspore::lite
|
||||
#endif // MINDSPORE_LITE_SRC_CONTROL_FLOW_SCHEDULER_H_
|
||||
#endif // MINDSPORE_LITE_SRC_CONTROL_FLOW_CONTROL_FLOW_SCHEDULER_H_
|
||||
|
|
|
@ -14,12 +14,12 @@
|
|||
* limitations under the License.
|
||||
*/
|
||||
|
||||
#ifndef MINDSPORE_LITE_SRC_CONTROLFLOW_CONTROL_SUBGRAPH_CREATOR_H_
|
||||
#define MINDSPORE_LITE_SRC_CONTROLFLOW_CONTROL_SUBGRAPH_CREATOR_H_
|
||||
#ifndef MINDSPORE_LITE_SRC_CONTROL_FLOW_CONTROL_SUBGRAPH_CREATOR_H_
|
||||
#define MINDSPORE_LITE_SRC_CONTROL_FLOW_CONTROL_SUBGRAPH_CREATOR_H_
|
||||
#include "src/runtime/lite_kernel.h"
|
||||
#include "src/runtime/sub_graph_kernel.h"
|
||||
|
||||
namespace mindspore::lite {
|
||||
kernel::SubGraphKernel *CreateControlSubgraph(const kernel::SubGraphType &type, kernel::LiteKernel *kernel);
|
||||
} // namespace mindspore::lite
|
||||
#endif // MINDSPORE_LITE_SRC_CONTROLFLOW_CONTROL_SUBGRAPH_CREATOR_H_
|
||||
#endif // MINDSPORE_LITE_SRC_CONTROL_FLOW_CONTROL_SUBGRAPH_CREATOR_H_
|
||||
|
|
|
@ -14,8 +14,8 @@
|
|||
* limitations under the License.
|
||||
*/
|
||||
|
||||
#ifndef MINDSPORE_LITE_SRC_CONTROLFLOW_KERNEL_ENTRANCE_SUBGRAPH_KERNEL_H_
|
||||
#define MINDSPORE_LITE_SRC_CONTROLFLOW_KERNEL_ENTRANCE_SUBGRAPH_KERNEL_H_
|
||||
#ifndef MINDSPORE_LITE_SRC_CONTROL_FLOW_KERNEL_ENTRANCE_SUBGRAPH_KERNEL_H_
|
||||
#define MINDSPORE_LITE_SRC_CONTROL_FLOW_KERNEL_ENTRANCE_SUBGRAPH_KERNEL_H_
|
||||
#include <atomic>
|
||||
#include <utility>
|
||||
#include <string>
|
||||
|
@ -52,4 +52,4 @@ class EntranceSubGraphKernel : public SubGraphKernel {
|
|||
int schema_version_ = lite::SCHEMA_VERSION::SCHEMA_CUR;
|
||||
};
|
||||
} // namespace mindspore::kernel
|
||||
#endif // MINDSPORE_LITE_SRC_CONTROLFLOW_KERNEL_ENTRANCE_SUBGRAPH_KERNEL_H_
|
||||
#endif // MINDSPORE_LITE_SRC_CONTROL_FLOW_KERNEL_ENTRANCE_SUBGRAPH_KERNEL_H_
|
||||
|
|
|
@ -14,8 +14,8 @@
|
|||
* limitations under the License.
|
||||
*/
|
||||
|
||||
#ifndef MINDSPORE_LITE_SRC_CONTROLFLOW_KERNEL_EXIT_SUBGRAPH_KERNEL_H_
|
||||
#define MINDSPORE_LITE_SRC_CONTROLFLOW_KERNEL_EXIT_SUBGRAPH_KERNEL_H_
|
||||
#ifndef MINDSPORE_LITE_SRC_CONTROL_FLOW_KERNEL_EXIT_SUBGRAPH_KERNEL_H_
|
||||
#define MINDSPORE_LITE_SRC_CONTROL_FLOW_KERNEL_EXIT_SUBGRAPH_KERNEL_H_
|
||||
#include <atomic>
|
||||
#include <utility>
|
||||
#include <string>
|
||||
|
@ -56,4 +56,4 @@ class ExitSubGraphKernel : public SubGraphKernel {
|
|||
std::set<kernel::KernelExec *> partials_;
|
||||
};
|
||||
} // namespace mindspore::kernel
|
||||
#endif // MINDSPORE_LITE_SRC_CONTROLFLOW_KERNEL_EXIT_SUBGRAPH_KERNEL_H_
|
||||
#endif // MINDSPORE_LITE_SRC_CONTROL_FLOW_KERNEL_EXIT_SUBGRAPH_KERNEL_H_
|
||||
|
|
|
@ -14,8 +14,8 @@
|
|||
* limitations under the License.
|
||||
*/
|
||||
|
||||
#ifndef MINDSPORE_LITE_SRC_CONTROLFLOW_KERNEL_IDENTITY_KERNEL_H_
|
||||
#define MINDSPORE_LITE_SRC_CONTROLFLOW_KERNEL_IDENTITY_KERNEL_H_
|
||||
#ifndef MINDSPORE_LITE_SRC_CONTROL_FLOW_KERNEL_IDENTITY_KERNEL_H_
|
||||
#define MINDSPORE_LITE_SRC_CONTROL_FLOW_KERNEL_IDENTITY_KERNEL_H_
|
||||
#include <atomic>
|
||||
#include <utility>
|
||||
#include <string>
|
||||
|
@ -55,4 +55,4 @@ class IdentityKernel : public LiteKernel {
|
|||
bool support_fp16_ = false;
|
||||
};
|
||||
} // namespace mindspore::kernel
|
||||
#endif // MINDSPORE_LITE_SRC_CONTROLFLOW_KERNEL_IDENTITY_KERNEL_H_
|
||||
#endif // MINDSPORE_LITE_SRC_CONTROL_FLOW_KERNEL_IDENTITY_KERNEL_H_
|
||||
|
|
|
@ -14,8 +14,8 @@
|
|||
* limitations under the License.
|
||||
*/
|
||||
|
||||
#ifndef MINDSPORE_CCSRC_CXX_API_ACL_UTILS_H
|
||||
#define MINDSPORE_CCSRC_CXX_API_ACL_UTILS_H
|
||||
#ifndef MINDSPORE_LITE_SRC_EXTENDRT_CXX_API_ACL_UTILS_H_
|
||||
#define MINDSPORE_LITE_SRC_EXTENDRT_CXX_API_ACL_UTILS_H_
|
||||
|
||||
#include <string>
|
||||
#include "acl/acl_base.h"
|
||||
|
@ -45,4 +45,4 @@ static inline bool IsAscendNo910Soc() {
|
|||
}
|
||||
} // namespace mindspore
|
||||
|
||||
#endif // MINDSPORE_CCSRC_CXX_API_ACL_UTILS_H
|
||||
#endif // MINDSPORE_LITE_SRC_EXTENDRT_CXX_API_ACL_UTILS_H_
|
||||
|
|
|
@ -13,10 +13,10 @@
|
|||
* See the License for the specific language governing permissions and
|
||||
* limitations under the License.
|
||||
*/
|
||||
#ifndef MINDSPORE_CCSRC_CXXAPI_AKG_KERNEL_REGISTER_H_
|
||||
#define MINDSPORE_CCSRC_CXXAPI_AKG_KERNEL_REGISTER_H_
|
||||
#ifndef MINDSPORE_LITE_SRC_EXTENDRT_CXX_API_AKG_KERNEL_REGISTER_H_
|
||||
#define MINDSPORE_LITE_SRC_EXTENDRT_CXX_API_AKG_KERNEL_REGISTER_H_
|
||||
|
||||
namespace mindspore {
|
||||
void RegAllOp();
|
||||
} // namespace mindspore
|
||||
#endif // MINDSPORE_CCSRC_CXXAPI_AKG_KERNEL_REGISTER_H_
|
||||
#endif // MINDSPORE_LITE_SRC_EXTENDRT_CXX_API_AKG_KERNEL_REGISTER_H_
|
||||
|
|
|
@ -13,8 +13,8 @@
|
|||
* See the License for the specific language governing permissions and
|
||||
* limitations under the License.
|
||||
*/
|
||||
#ifndef MINDSPORE_CCSRC_CXX_API_DLUTILS_H_
|
||||
#define MINDSPORE_CCSRC_CXX_API_DLUTILS_H_
|
||||
#ifndef MINDSPORE_LITE_SRC_EXTENDRT_CXX_API_DLUTILS_H_
|
||||
#define MINDSPORE_LITE_SRC_EXTENDRT_CXX_API_DLUTILS_H_
|
||||
#if !defined(_WIN32) && !defined(_WIN64)
|
||||
#include <dlfcn.h>
|
||||
#include <dirent.h>
|
||||
|
@ -98,4 +98,4 @@ inline void DLSoClose(void *handle) {
|
|||
} while (false)
|
||||
} // namespace mindspore
|
||||
#endif
|
||||
#endif // MINDSPORE_CCSRC_CXX_API_DLUTILS_H_
|
||||
#endif // MINDSPORE_LITE_SRC_EXTENDRT_CXX_API_DLUTILS_H_
|
||||
|
|
|
@ -13,8 +13,8 @@
|
|||
* See the License for the specific language governing permissions and
|
||||
* limitations under the License.
|
||||
*/
|
||||
#ifndef MINDSPORE_CCSRC_CXX_API_FACTORY_H
|
||||
#define MINDSPORE_CCSRC_CXX_API_FACTORY_H
|
||||
#ifndef MINDSPORE_LITE_SRC_EXTENDRT_CXX_API_FACTORY_H_
|
||||
#define MINDSPORE_LITE_SRC_EXTENDRT_CXX_API_FACTORY_H_
|
||||
#include <functional>
|
||||
#include <map>
|
||||
#include <string>
|
||||
|
@ -84,4 +84,4 @@ class Registrar {
|
|||
|
||||
#define API_FACTORY_REG(BASE, DERIVE) static const Registrar<BASE> g_api_##DERIVE##_reg(API_FACTORY_CREATOR(DERIVE));
|
||||
} // namespace mindspore
|
||||
#endif // MINDSPORE_CCSRC_CXX_API_FACTORY_H
|
||||
#endif // MINDSPORE_LITE_SRC_EXTENDRT_CXX_API_FACTORY_H_
|
||||
|
|
|
@ -13,8 +13,8 @@
|
|||
* See the License for the specific language governing permissions and
|
||||
* limitations under the License.
|
||||
*/
|
||||
#ifndef MINDSPORE_CCSRC_CXX_API_GRAPH_ACL_ACL_ENV_GUARD_H
|
||||
#define MINDSPORE_CCSRC_CXX_API_GRAPH_ACL_ACL_ENV_GUARD_H
|
||||
#ifndef MINDSPORE_LITE_SRC_EXTENDRT_CXX_API_GRAPH_ACL_ACL_ENV_GUARD_H_
|
||||
#define MINDSPORE_LITE_SRC_EXTENDRT_CXX_API_GRAPH_ACL_ACL_ENV_GUARD_H_
|
||||
|
||||
#include <memory>
|
||||
#include <mutex>
|
||||
|
@ -35,4 +35,4 @@ class __attribute__((visibility("default"))) AclEnvGuard {
|
|||
aclError errno_;
|
||||
};
|
||||
} // namespace mindspore
|
||||
#endif // MINDSPORE_CCSRC_CXX_API_GRAPH_ACL_ACL_ENV_GUARD_H
|
||||
#endif // MINDSPORE_LITE_SRC_EXTENDRT_CXX_API_GRAPH_ACL_ACL_ENV_GUARD_H_
|
||||
|
|
|
@ -13,8 +13,8 @@
|
|||
* See the License for the specific language governing permissions and
|
||||
* limitations under the License.
|
||||
*/
|
||||
#ifndef MINDSPORE_CCSRC_CXX_API_GRAPH_ACL_ACL_GRAPH_IMPL_H
|
||||
#define MINDSPORE_CCSRC_CXX_API_GRAPH_ACL_ACL_GRAPH_IMPL_H
|
||||
#ifndef MINDSPORE_LITE_SRC_EXTENDRT_CXX_API_GRAPH_ACL_ACL_GRAPH_IMPL_H_
|
||||
#define MINDSPORE_LITE_SRC_EXTENDRT_CXX_API_GRAPH_ACL_ACL_GRAPH_IMPL_H_
|
||||
#include <functional>
|
||||
#include <map>
|
||||
#include <string>
|
||||
|
@ -56,4 +56,4 @@ class AclGraphImpl : public GraphCell::GraphImpl {
|
|||
ModelProcess model_process_;
|
||||
};
|
||||
} // namespace mindspore
|
||||
#endif // MINDSPORE_CCSRC_CXX_API_GRAPH_ACL_ACL_GRAPH_IMPL_H
|
||||
#endif // MINDSPORE_LITE_SRC_EXTENDRT_CXX_API_GRAPH_ACL_ACL_GRAPH_IMPL_H_
|
||||
|
|
|
@ -14,8 +14,8 @@
|
|||
* limitations under the License.
|
||||
*/
|
||||
|
||||
#ifndef MINDSPORE_CCSRC_CXXAPI_GRAPH_ACL_MODEL_PROCESS_H
|
||||
#define MINDSPORE_CCSRC_CXXAPI_GRAPH_ACL_MODEL_PROCESS_H
|
||||
#ifndef MINDSPORE_LITE_SRC_EXTENDRT_CXX_API_GRAPH_ACL_MODEL_PROCESS_H_
|
||||
#define MINDSPORE_LITE_SRC_EXTENDRT_CXX_API_GRAPH_ACL_MODEL_PROCESS_H_
|
||||
#include <vector>
|
||||
#include <string>
|
||||
#include <map>
|
||||
|
@ -88,4 +88,4 @@ class ModelProcess {
|
|||
};
|
||||
} // namespace mindspore
|
||||
|
||||
#endif // MINDSPORE_CCSRC_CXXAPI_GRAPH_ACL_MODEL_PROCESS_H
|
||||
#endif // MINDSPORE_LITE_SRC_EXTENDRT_CXX_API_GRAPH_ACL_MODEL_PROCESS_H_
|
||||
|
|
|
@ -13,8 +13,8 @@
|
|||
* See the License for the specific language governing permissions and
|
||||
* limitations under the License.
|
||||
*/
|
||||
#ifndef MINDSPORE_CCSRC_CXX_API_GRAPH_MS_ASCEND_GRAPH_IMPL_H
|
||||
#define MINDSPORE_CCSRC_CXX_API_GRAPH_MS_ASCEND_GRAPH_IMPL_H
|
||||
#ifndef MINDSPORE_LITE_SRC_EXTENDRT_CXX_API_GRAPH_ASCEND_ASCEND_GRAPH_IMPL_H_
|
||||
#define MINDSPORE_LITE_SRC_EXTENDRT_CXX_API_GRAPH_ASCEND_ASCEND_GRAPH_IMPL_H_
|
||||
#include <functional>
|
||||
#include <map>
|
||||
#include <string>
|
||||
|
@ -93,4 +93,4 @@ class PythonEnvGuard {
|
|||
bool origin_init_status_;
|
||||
};
|
||||
} // namespace mindspore
|
||||
#endif // MINDSPORE_CCSRC_CXX_API_GRAPH_MS_ASCEND_GRAPH_IMPL_H
|
||||
#endif // MINDSPORE_LITE_SRC_EXTENDRT_CXX_API_GRAPH_ASCEND_ASCEND_GRAPH_IMPL_H_
|
||||
|
|
|
@ -13,8 +13,8 @@
|
|||
* See the License for the specific language governing permissions and
|
||||
* limitations under the License.
|
||||
*/
|
||||
#ifndef MINDSPORE_CCSRC_CXX_API_GRAPH_MS_GPU_GRAPH_IMPL_H
|
||||
#define MINDSPORE_CCSRC_CXX_API_GRAPH_MS_GPU_GRAPH_IMPL_H
|
||||
#ifndef MINDSPORE_LITE_SRC_EXTENDRT_CXX_API_GRAPH_GPU_GPU_GRAPH_IMPL_H_
|
||||
#define MINDSPORE_LITE_SRC_EXTENDRT_CXX_API_GRAPH_GPU_GPU_GRAPH_IMPL_H_
|
||||
#include <string>
|
||||
#include <vector>
|
||||
#include <utility>
|
||||
|
@ -66,4 +66,4 @@ class GPUGraphImpl : public GraphCell::GraphImpl {
|
|||
uint32_t workspace_size_;
|
||||
};
|
||||
} // namespace mindspore
|
||||
#endif // MINDSPORE_CCSRC_CXX_API_GRAPH_MS_GPU_GRAPH_IMPL_H
|
||||
#endif // MINDSPORE_LITE_SRC_EXTENDRT_CXX_API_GRAPH_GPU_GPU_GRAPH_IMPL_H_
|
||||
|
|
|
@ -13,8 +13,8 @@
|
|||
* See the License for the specific language governing permissions and
|
||||
* limitations under the License.
|
||||
*/
|
||||
#ifndef MINDSPORE_CCSRC_CXX_API_GRAPH_GRAPH_DATA_H
|
||||
#define MINDSPORE_CCSRC_CXX_API_GRAPH_GRAPH_DATA_H
|
||||
#ifndef MINDSPORE_LITE_SRC_EXTENDRT_CXX_API_GRAPH_GRAPH_DATA_H_
|
||||
#define MINDSPORE_LITE_SRC_EXTENDRT_CXX_API_GRAPH_GRAPH_DATA_H_
|
||||
|
||||
#include <string>
|
||||
#include <vector>
|
||||
|
@ -53,4 +53,4 @@ class Graph::GraphData {
|
|||
std::vector<std::shared_ptr<dataset::Execute>> data_graph_;
|
||||
};
|
||||
} // namespace mindspore
|
||||
#endif // MINDSPORE_CCSRC_CXX_API_GRAPH_GRAPH_DATA_H
|
||||
#endif // MINDSPORE_LITE_SRC_EXTENDRT_CXX_API_GRAPH_GRAPH_DATA_H_
|
||||
|
|
|
@ -13,8 +13,8 @@
|
|||
* See the License for the specific language governing permissions and
|
||||
* limitations under the License.
|
||||
*/
|
||||
#ifndef MINDSPORE_CCSRC_CXX_API_GRAPH_GRAPH_IMPL_H
|
||||
#define MINDSPORE_CCSRC_CXX_API_GRAPH_GRAPH_IMPL_H
|
||||
#ifndef MINDSPORE_LITE_SRC_EXTENDRT_CXX_API_GRAPH_GRAPH_IMPL_H_
|
||||
#define MINDSPORE_LITE_SRC_EXTENDRT_CXX_API_GRAPH_GRAPH_IMPL_H_
|
||||
#include <functional>
|
||||
#include <map>
|
||||
#include <string>
|
||||
|
@ -49,4 +49,4 @@ class GraphCell::GraphImpl {
|
|||
std::shared_ptr<Context> graph_context_;
|
||||
};
|
||||
} // namespace mindspore
|
||||
#endif // MINDSPORE_CCSRC_CXX_API_GRAPH_GRAPH_IMPL_H
|
||||
#endif // MINDSPORE_LITE_SRC_EXTENDRT_CXX_API_GRAPH_GRAPH_IMPL_H_
|
||||
|
|
|
@ -13,8 +13,8 @@
|
|||
* See the License for the specific language governing permissions and
|
||||
* limitations under the License.
|
||||
*/
|
||||
#ifndef MINDSPORE_CCSRC_CXX_API_ACL_MODEL_H
|
||||
#define MINDSPORE_CCSRC_CXX_API_ACL_MODEL_H
|
||||
#ifndef MINDSPORE_LITE_SRC_EXTENDRT_CXX_API_MODEL_ACL_ACL_MODEL_H_
|
||||
#define MINDSPORE_LITE_SRC_EXTENDRT_CXX_API_MODEL_ACL_ACL_MODEL_H_
|
||||
|
||||
#include <vector>
|
||||
#include <string>
|
||||
|
@ -52,4 +52,4 @@ class AclModel : public ModelImpl {
|
|||
std::map<std::string, std::shared_ptr<Graph>> dynamic_size_graph_map_;
|
||||
};
|
||||
} // namespace mindspore
|
||||
#endif // MINDSPORE_CCSRC_CXX_API_ACL_MODEL_H
|
||||
#endif // MINDSPORE_LITE_SRC_EXTENDRT_CXX_API_MODEL_ACL_ACL_MODEL_H_
|
||||
|
|
|
@ -13,8 +13,8 @@
|
|||
* See the License for the specific language governing permissions and
|
||||
* limitations under the License.
|
||||
*/
|
||||
#ifndef MINDSPORE_CCSRC_CXX_API_ACL_MODEL_MULTI_H
|
||||
#define MINDSPORE_CCSRC_CXX_API_ACL_MODEL_MULTI_H
|
||||
#ifndef MINDSPORE_LITE_SRC_EXTENDRT_CXX_API_MODEL_ACL_ACL_MODEL_MULTI_H_
|
||||
#define MINDSPORE_LITE_SRC_EXTENDRT_CXX_API_MODEL_ACL_ACL_MODEL_MULTI_H_
|
||||
|
||||
#include "cxx_api/model/acl/acl_model.h"
|
||||
#include <memory>
|
||||
|
@ -51,4 +51,4 @@ class AclModelMulti : public AclModel {
|
|||
std::vector<MSTensor> outputs_ = {};
|
||||
};
|
||||
} // namespace mindspore
|
||||
#endif // MINDSPORE_CCSRC_CXX_API_ACL_MODEL_MULTI_H
|
||||
#endif // MINDSPORE_LITE_SRC_EXTENDRT_CXX_API_MODEL_ACL_ACL_MODEL_MULTI_H_
|
||||
|
|
|
@ -14,8 +14,8 @@
|
|||
* limitations under the License.
|
||||
*/
|
||||
|
||||
#ifndef MINDSPORE_CCSRC_CXXAPI_SESSION_ACL_OPTION_PARSER_H
|
||||
#define MINDSPORE_CCSRC_CXXAPI_SESSION_ACL_OPTION_PARSER_H
|
||||
#ifndef MINDSPORE_LITE_SRC_EXTENDRT_CXX_API_MODEL_ACL_ACL_MODEL_OPTIONS_H_
|
||||
#define MINDSPORE_LITE_SRC_EXTENDRT_CXX_API_MODEL_ACL_ACL_MODEL_OPTIONS_H_
|
||||
#include <vector>
|
||||
#include <string>
|
||||
#include <map>
|
||||
|
@ -63,4 +63,4 @@ class MS_API AclModelOptions {
|
|||
};
|
||||
} // namespace mindspore
|
||||
|
||||
#endif // MINDSPORE_CCSRC_CXXAPI_SESSION_ACL_OPTION_PARSER_H
|
||||
#endif // MINDSPORE_LITE_SRC_EXTENDRT_CXX_API_MODEL_ACL_ACL_MODEL_OPTIONS_H_
|
||||
|
|
|
@ -13,8 +13,8 @@
|
|||
* See the License for the specific language governing permissions and
|
||||
* limitations under the License.
|
||||
*/
|
||||
#ifndef MINDSPORE_CCSRC_CXX_API_ACL_VM_ACL_MULTI_GRAPH_SESSION_H
|
||||
#define MINDSPORE_CCSRC_CXX_API_ACL_VM_ACL_MULTI_GRAPH_SESSION_H
|
||||
#ifndef MINDSPORE_LITE_SRC_EXTENDRT_CXX_API_MODEL_ACL_ACL_VM_ACL_MULTI_GRAPH_SESSION_H_
|
||||
#define MINDSPORE_LITE_SRC_EXTENDRT_CXX_API_MODEL_ACL_ACL_VM_ACL_MULTI_GRAPH_SESSION_H_
|
||||
|
||||
#include <deque>
|
||||
#include <vector>
|
||||
|
@ -46,4 +46,4 @@ class MultiGraphAclSession : public session::SessionBasic {
|
|||
};
|
||||
} // namespace session
|
||||
} // namespace mindspore
|
||||
#endif // MINDSPORE_CCSRC_CXX_API_ACL_VM_ACL_MULTI_GRAPH_SESSION_H
|
||||
#endif // MINDSPORE_LITE_SRC_EXTENDRT_CXX_API_MODEL_ACL_ACL_VM_ACL_MULTI_GRAPH_SESSION_H_
|
||||
|
|
|
@ -13,8 +13,8 @@
|
|||
* See the License for the specific language governing permissions and
|
||||
* limitations under the License.
|
||||
*/
|
||||
#ifndef MINDSPORE_CCSRC_CXX_API_ACL_VM_ACL_VM_H
|
||||
#define MINDSPORE_CCSRC_CXX_API_ACL_VM_ACL_VM_H
|
||||
#ifndef MINDSPORE_LITE_SRC_EXTENDRT_CXX_API_MODEL_ACL_ACL_VM_ACL_VM_H_
|
||||
#define MINDSPORE_LITE_SRC_EXTENDRT_CXX_API_MODEL_ACL_ACL_VM_ACL_VM_H_
|
||||
|
||||
#include <vector>
|
||||
#include <memory>
|
||||
|
@ -59,4 +59,4 @@ class AclCompileGraphs : public compile::CompileGraphs {
|
|||
~AclCompileGraphs() override = default;
|
||||
};
|
||||
} // namespace mindspore
|
||||
#endif // MINDSPORE_CCSRC_CXX_API_ACL_VM_ACL_VM_H
|
||||
#endif // MINDSPORE_LITE_SRC_EXTENDRT_CXX_API_MODEL_ACL_ACL_VM_ACL_VM_H_
|
||||
|
|
|
@ -13,8 +13,8 @@
|
|||
* See the License for the specific language governing permissions and
|
||||
* limitations under the License.
|
||||
*/
|
||||
#ifndef MINDSPORE_CCSRC_CXX_API_ACL_VM_MS_TENSOR_REF_H
|
||||
#define MINDSPORE_CCSRC_CXX_API_ACL_VM_MS_TENSOR_REF_H
|
||||
#ifndef MINDSPORE_LITE_SRC_EXTENDRT_CXX_API_MODEL_ACL_ACL_VM_MS_TENSOR_REF_H_
|
||||
#define MINDSPORE_LITE_SRC_EXTENDRT_CXX_API_MODEL_ACL_ACL_VM_MS_TENSOR_REF_H_
|
||||
|
||||
#include <memory>
|
||||
#include <string>
|
||||
|
@ -46,4 +46,4 @@ class MSTensorRef : public BaseRef {
|
|||
MSTensor ms_tensor_;
|
||||
};
|
||||
} // namespace mindspore
|
||||
#endif // MINDSPORE_CCSRC_CXX_API_ACL_VM_MS_TENSOR_REF_H
|
||||
#endif // MINDSPORE_LITE_SRC_EXTENDRT_CXX_API_MODEL_ACL_ACL_VM_MS_TENSOR_REF_H_
|
||||
|
|
|
@ -14,8 +14,8 @@
|
|||
* limitations under the License.
|
||||
*/
|
||||
|
||||
#ifndef MINDSPORE_CCSRC_CXXAPI_SESSION_ACL_MODEL_CONVERTER_H
|
||||
#define MINDSPORE_CCSRC_CXXAPI_SESSION_ACL_MODEL_CONVERTER_H
|
||||
#ifndef MINDSPORE_LITE_SRC_EXTENDRT_CXX_API_MODEL_ACL_MODEL_CONVERTER_H_
|
||||
#define MINDSPORE_LITE_SRC_EXTENDRT_CXX_API_MODEL_ACL_MODEL_CONVERTER_H_
|
||||
#include <vector>
|
||||
#include <string>
|
||||
#include <map>
|
||||
|
@ -48,4 +48,4 @@ class MS_API ModelConverter {
|
|||
std::weak_ptr<AclModelOptions> options_;
|
||||
};
|
||||
} // namespace mindspore
|
||||
#endif // MINDSPORE_CCSRC_CXXAPI_SESSION_ACL_MODEL_CONVERTER_H
|
||||
#endif // MINDSPORE_LITE_SRC_EXTENDRT_CXX_API_MODEL_ACL_MODEL_CONVERTER_H_
|
||||
|
|
|
@ -14,8 +14,8 @@
|
|||
* limitations under the License.
|
||||
*/
|
||||
|
||||
#ifndef MINDSPORE_CCSRC_CXXAPI_MULTI_PROCESS_H
|
||||
#define MINDSPORE_CCSRC_CXXAPI_MULTI_PROCESS_H
|
||||
#ifndef MINDSPORE_LITE_SRC_EXTENDRT_CXX_API_MODEL_MODEL_CONVERTER_UTILS_MULTI_PROCESS_H_
|
||||
#define MINDSPORE_LITE_SRC_EXTENDRT_CXX_API_MODEL_MODEL_CONVERTER_UTILS_MULTI_PROCESS_H_
|
||||
#include <iostream>
|
||||
#include <functional>
|
||||
#include "include/api/status.h"
|
||||
|
@ -60,4 +60,4 @@ class MultiProcess {
|
|||
void ChildProcess(const ProcessFuncCall &child_process);
|
||||
};
|
||||
} // namespace mindspore
|
||||
#endif // MINDSPORE_CCSRC_CXXAPI_MULTI_PROCESS_H
|
||||
#endif // MINDSPORE_LITE_SRC_EXTENDRT_CXX_API_MODEL_MODEL_CONVERTER_UTILS_MULTI_PROCESS_H_
|
||||
|
|
|
@ -14,8 +14,8 @@
|
|||
* limitations under the License.
|
||||
*/
|
||||
|
||||
#ifndef MINDSPORE_CCSRC_CXXAPI_SHARED_MEMORY_H
|
||||
#define MINDSPORE_CCSRC_CXXAPI_SHARED_MEMORY_H
|
||||
#ifndef MINDSPORE_LITE_SRC_EXTENDRT_CXX_API_MODEL_MODEL_CONVERTER_UTILS_SHARED_MEMORY_H_
|
||||
#define MINDSPORE_LITE_SRC_EXTENDRT_CXX_API_MODEL_MODEL_CONVERTER_UTILS_SHARED_MEMORY_H_
|
||||
#include <iostream>
|
||||
#include "include/api/status.h"
|
||||
|
||||
|
@ -35,4 +35,4 @@ class SharedMemory {
|
|||
uint8_t *shmat_addr_ = nullptr;
|
||||
};
|
||||
} // namespace mindspore
|
||||
#endif // MINDSPORE_CCSRC_CXXAPI_SHARED_MEMORY_H
|
||||
#endif // MINDSPORE_LITE_SRC_EXTENDRT_CXX_API_MODEL_MODEL_CONVERTER_UTILS_SHARED_MEMORY_H_
|
||||
|
|
|
@ -13,8 +13,8 @@
|
|||
* See the License for the specific language governing permissions and
|
||||
* limitations under the License.
|
||||
*/
|
||||
#ifndef MINDSPORE_CCSRC_CXX_API_MODEL_MODEL_IMPL_H
|
||||
#define MINDSPORE_CCSRC_CXX_API_MODEL_MODEL_IMPL_H
|
||||
#ifndef MINDSPORE_LITE_SRC_EXTENDRT_CXX_API_MODEL_MODEL_IMPL_H_
|
||||
#define MINDSPORE_LITE_SRC_EXTENDRT_CXX_API_MODEL_MODEL_IMPL_H_
|
||||
#include <functional>
|
||||
#include <map>
|
||||
#include <string>
|
||||
|
@ -77,4 +77,4 @@ class ModelImpl {
|
|||
}
|
||||
};
|
||||
} // namespace mindspore
|
||||
#endif // MINDSPORE_CCSRC_CXX_API_MODEL_MODEL_IMPL_H
|
||||
#endif // MINDSPORE_LITE_SRC_EXTENDRT_CXX_API_MODEL_MODEL_IMPL_H_
|
||||
|
|
|
@ -13,8 +13,8 @@
|
|||
* See the License for the specific language governing permissions and
|
||||
* limitations under the License.
|
||||
*/
|
||||
#ifndef MINDSPORE_CCSRC_SESSION_SESSION_H
|
||||
#define MINDSPORE_CCSRC_SESSION_SESSION_H
|
||||
#ifndef MINDSPORE_LITE_SRC_EXTENDRT_CXX_API_MODEL_MS_MS_MODEL_H_
|
||||
#define MINDSPORE_LITE_SRC_EXTENDRT_CXX_API_MODEL_MS_MS_MODEL_H_
|
||||
|
||||
#include <vector>
|
||||
#include <string>
|
||||
|
|
|
@ -14,8 +14,8 @@
|
|||
* limitations under the License.
|
||||
*/
|
||||
|
||||
#ifndef MINDSPORE_LITE_LRU_CACHE_ALGORITHM_H_
|
||||
#define MINDSPORE_LITE_LRU_CACHE_ALGORITHM_H_
|
||||
#ifndef MINDSPORE_LITE_SRC_EXTENDRT_DELEGATE_PARAMETER_CACHE_CACHE_ALGORITHM_H_
|
||||
#define MINDSPORE_LITE_SRC_EXTENDRT_DELEGATE_PARAMETER_CACHE_CACHE_ALGORITHM_H_
|
||||
|
||||
#include <vector>
|
||||
#include "include/api/status.h"
|
||||
|
@ -40,4 +40,4 @@ class CacheAlgorithm {
|
|||
};
|
||||
} // namespace cache
|
||||
} // namespace mindspore
|
||||
#endif // MINDSPORE_LITE_LRU_CACHE_ALGORITHM_H_
|
||||
#endif // MINDSPORE_LITE_SRC_EXTENDRT_DELEGATE_PARAMETER_CACHE_CACHE_ALGORITHM_H_
|
||||
|
|
|
@ -14,8 +14,8 @@
|
|||
* limitations under the License.
|
||||
*/
|
||||
|
||||
#ifndef MINDSPORE_LITE_PARAMETER_CACHE_MEM_BASE_H_
|
||||
#define MINDSPORE_LITE_PARAMETER_CACHE_MEM_BASE_H_
|
||||
#ifndef MINDSPORE_LITE_SRC_EXTENDRT_DELEGATE_PARAMETER_CACHE_CACHE_MEM_BASE_H_
|
||||
#define MINDSPORE_LITE_SRC_EXTENDRT_DELEGATE_PARAMETER_CACHE_CACHE_MEM_BASE_H_
|
||||
#include <utility>
|
||||
#include <memory>
|
||||
|
||||
|
@ -38,4 +38,4 @@ class CacheMemBase {
|
|||
};
|
||||
} // namespace cache
|
||||
} // namespace mindspore
|
||||
#endif // MINDSPORE_LITE_PARAMETER_CACHE_MEM_BASE_H_
|
||||
#endif // MINDSPORE_LITE_SRC_EXTENDRT_DELEGATE_PARAMETER_CACHE_CACHE_MEM_BASE_H_
|
||||
|
|
|
@ -14,8 +14,8 @@
|
|||
* limitations under the License.
|
||||
*/
|
||||
|
||||
#ifndef MINDSPORE_LITE_EMBEDDING_CACHE_H_
|
||||
#define MINDSPORE_LITE_EMBEDDING_CACHE_H_
|
||||
#ifndef MINDSPORE_LITE_SRC_EXTENDRT_DELEGATE_PARAMETER_CACHE_EMBEDDING_CACHE_H_
|
||||
#define MINDSPORE_LITE_SRC_EXTENDRT_DELEGATE_PARAMETER_CACHE_EMBEDDING_CACHE_H_
|
||||
#include <cmath>
|
||||
#include <algorithm>
|
||||
#include <memory>
|
||||
|
@ -86,4 +86,4 @@ class EmbeddingCache {
|
|||
};
|
||||
} // namespace cache
|
||||
} // namespace mindspore
|
||||
#endif // MINDSPORE_LITE_EMBEDDING_CACHE_H_
|
||||
#endif // MINDSPORE_LITE_SRC_EXTENDRT_DELEGATE_PARAMETER_CACHE_EMBEDDING_CACHE_H_
|
||||
|
|
|
@ -14,8 +14,8 @@
|
|||
* limitations under the License.
|
||||
*/
|
||||
|
||||
#ifndef MINDSPORE_LITE_EMBEDDING_CACHE_MANAGER_H_
|
||||
#define MINDSPORE_LITE_EMBEDDING_CACHE_MANAGER_H_
|
||||
#ifndef MINDSPORE_LITE_SRC_EXTENDRT_DELEGATE_PARAMETER_CACHE_EMBEDDING_CACHE_MANAGER_H_
|
||||
#define MINDSPORE_LITE_SRC_EXTENDRT_DELEGATE_PARAMETER_CACHE_EMBEDDING_CACHE_MANAGER_H_
|
||||
#include <memory>
|
||||
#include <map>
|
||||
#include <string>
|
||||
|
@ -57,4 +57,4 @@ class EmbeddingCacheManager {
|
|||
};
|
||||
} // namespace cache
|
||||
} // namespace mindspore
|
||||
#endif // MINDSPORE_LITE_EMBEDDING_CACHE_MANAGER_H_
|
||||
#endif // MINDSPORE_LITE_SRC_EXTENDRT_DELEGATE_PARAMETER_CACHE_EMBEDDING_CACHE_MANAGER_H_
|
||||
|
|
|
@ -14,8 +14,8 @@
|
|||
* limitations under the License.
|
||||
*/
|
||||
|
||||
#ifndef MINDSPORE_LITE_FACTORY_MGR_BASE_H_
|
||||
#define MINDSPORE_LITE_FACTORY_MGR_BASE_H_
|
||||
#ifndef MINDSPORE_LITE_SRC_EXTENDRT_DELEGATE_PARAMETER_CACHE_FACTORY_MGR_BASE_H_
|
||||
#define MINDSPORE_LITE_SRC_EXTENDRT_DELEGATE_PARAMETER_CACHE_FACTORY_MGR_BASE_H_
|
||||
#include <map>
|
||||
#include <memory>
|
||||
#include "include/api/status.h"
|
||||
|
@ -78,4 +78,4 @@ class CommonProcductRegistrar : public ProcductRegistrar<KEY, PRODUCT> {
|
|||
static mindspore::lite::CommonProcductRegistrar<KEY, PRODUCT, PRODUCT_IMPL> g_commonProcductRegistrar##name(key);
|
||||
} // namespace lite
|
||||
} // namespace mindspore
|
||||
#endif // MINDSPORE_LITE_FACTORY_MGR_BASE_H_
|
||||
#endif // MINDSPORE_LITE_SRC_EXTENDRT_DELEGATE_PARAMETER_CACHE_FACTORY_MGR_BASE_H_
|
||||
|
|
|
@ -14,8 +14,8 @@
|
|||
* limitations under the License.
|
||||
*/
|
||||
|
||||
#ifndef MINDSPORE_LITE_PARAMETER_GPU_CACHE_MEM_H_
|
||||
#define MINDSPORE_LITE_PARAMETER_GPU_CACHE_MEM_H_
|
||||
#ifndef MINDSPORE_LITE_SRC_EXTENDRT_DELEGATE_PARAMETER_CACHE_GPU_GPU_CACHE_MEM_H_
|
||||
#define MINDSPORE_LITE_SRC_EXTENDRT_DELEGATE_PARAMETER_CACHE_GPU_GPU_CACHE_MEM_H_
|
||||
|
||||
#include <cuda_runtime_api.h>
|
||||
#include <memory>
|
||||
|
@ -45,4 +45,4 @@ class GPUCacheMem : public cache::CacheMemBase {
|
|||
} // namespace gpu
|
||||
} // namespace cache
|
||||
} // namespace mindspore
|
||||
#endif // MINDSPORE_LITE_PARAMETER_GPU_CACHE_MEM_H_
|
||||
#endif // MINDSPORE_LITE_SRC_EXTENDRT_DELEGATE_PARAMETER_CACHE_GPU_GPU_CACHE_MEM_H_
|
||||
|
|
|
@ -14,8 +14,8 @@
|
|||
* limitations under the License.
|
||||
*/
|
||||
|
||||
#ifndef MINDSPORE_LITE_LRU_CACHE_H_
|
||||
#define MINDSPORE_LITE_LRU_CACHE_H_
|
||||
#ifndef MINDSPORE_LITE_SRC_EXTENDRT_DELEGATE_PARAMETER_CACHE_LFU_CACHE_H_
|
||||
#define MINDSPORE_LITE_SRC_EXTENDRT_DELEGATE_PARAMETER_CACHE_LFU_CACHE_H_
|
||||
|
||||
#include <map>
|
||||
#include <unordered_map>
|
||||
|
@ -52,4 +52,4 @@ class LFUCacheAlgorithm : public CacheAlgorithm {
|
|||
};
|
||||
} // namespace cache
|
||||
} // namespace mindspore
|
||||
#endif // MINDSPORE_LITE_LRU_CACHE_H_
|
||||
#endif // MINDSPORE_LITE_SRC_EXTENDRT_DELEGATE_PARAMETER_CACHE_LFU_CACHE_H_
|
||||
|
|
|
@ -14,8 +14,8 @@
|
|||
* limitations under the License.
|
||||
*/
|
||||
|
||||
#ifndef MINDSPORE_LITE_LOAD_HOST_CACHE_DATA_H_
|
||||
#define MINDSPORE_LITE_LOAD_HOST_CACHE_DATA_H_
|
||||
#ifndef MINDSPORE_LITE_SRC_EXTENDRT_DELEGATE_PARAMETER_CACHE_LOAD_HOST_CACHE_MODEL_H_
|
||||
#define MINDSPORE_LITE_SRC_EXTENDRT_DELEGATE_PARAMETER_CACHE_LOAD_HOST_CACHE_MODEL_H_
|
||||
|
||||
#include <map>
|
||||
#include <string>
|
||||
|
|
|
@ -14,8 +14,8 @@
|
|||
* limitations under the License.
|
||||
*/
|
||||
|
||||
#ifndef MINDSPORE_LITE_SRC_DELEGATE_TENSORRT_CDUA_IMPL_CUBLAS_UTILS_H_
|
||||
#define MINDSPORE_LITE_SRC_DELEGATE_TENSORRT_CDUA_IMPL_CUBLAS_UTILS_H_
|
||||
#ifndef MINDSPORE_LITE_SRC_EXTENDRT_DELEGATE_TENSORRT_CUDA_IMPL_CUBLAS_UTILS_H_
|
||||
#define MINDSPORE_LITE_SRC_EXTENDRT_DELEGATE_TENSORRT_CUDA_IMPL_CUBLAS_UTILS_H_
|
||||
|
||||
#include <cublas_v2.h>
|
||||
#include "src/extendrt/delegate/tensorrt/cuda_impl/cuda_helper.h"
|
||||
|
@ -59,4 +59,4 @@ void CublasMM1Batch(const void *a_addr, const void *b_addr, void *c_addr, const
|
|||
void CublasMMBatched(void **a_addrs, void **b_addrs, void **c_addrs, const int *params,
|
||||
const cublasOperation_t *operations, const cudaDataType *data_types, cublasHandle_t cublas_handle);
|
||||
} // namespace mindspore::lite
|
||||
#endif // MINDSPORE_LITE_SRC_DELEGATE_TENSORRT_CDUA_IMPL_CUBLAS_UTILS_H_
|
||||
#endif // MINDSPORE_LITE_SRC_EXTENDRT_DELEGATE_TENSORRT_CUDA_IMPL_CUBLAS_UTILS_H_
|
||||
|
|
|
@ -14,8 +14,8 @@
|
|||
* limitations under the License.
|
||||
*/
|
||||
|
||||
#ifndef MINDSPORE_LITE_SRC_DELEGATE_TENSORRT_CDUA_IMPL_CUDA_HELPER_H_
|
||||
#define MINDSPORE_LITE_SRC_DELEGATE_TENSORRT_CDUA_IMPL_CUDA_HELPER_H_
|
||||
#ifndef MINDSPORE_LITE_SRC_EXTENDRT_DELEGATE_TENSORRT_CUDA_IMPL_CUDA_HELPER_H_
|
||||
#define MINDSPORE_LITE_SRC_EXTENDRT_DELEGATE_TENSORRT_CUDA_IMPL_CUDA_HELPER_H_
|
||||
|
||||
#include <cuda_runtime.h>
|
||||
#include <algorithm>
|
||||
|
@ -60,4 +60,4 @@ class CudaHelper {
|
|||
} \
|
||||
} while (0)
|
||||
|
||||
#endif // MINDSPORE_LITE_SRC_DELEGATE_TENSORRT_CDUA_IMPL_CUDA_HELPER_H_
|
||||
#endif // MINDSPORE_LITE_SRC_EXTENDRT_DELEGATE_TENSORRT_CUDA_IMPL_CUDA_HELPER_H_
|
||||
|
|
|
@ -14,8 +14,8 @@
|
|||
* limitations under the License.
|
||||
*/
|
||||
|
||||
#ifndef MINDSPORE_LITE_SRC_DELEGATE_TENSORRT_CDUA_IMPL_CUDNN_UTILS_H_
|
||||
#define MINDSPORE_LITE_SRC_DELEGATE_TENSORRT_CDUA_IMPL_CUDNN_UTILS_H_
|
||||
#ifndef MINDSPORE_LITE_SRC_EXTENDRT_DELEGATE_TENSORRT_CUDA_IMPL_CUDNN_UTILS_H_
|
||||
#define MINDSPORE_LITE_SRC_EXTENDRT_DELEGATE_TENSORRT_CUDA_IMPL_CUDNN_UTILS_H_
|
||||
|
||||
#include <cudnn.h>
|
||||
#include <NvInfer.h>
|
||||
|
@ -45,4 +45,4 @@ cudnnDataType_t ConvertCudnnDataType(nvinfer1::DataType trt_datatype);
|
|||
int CudnnActivation(cudnnHandle_t handle, cudnnActivationDescriptor_t activation_desc,
|
||||
const cudnnTensorDescriptor_t x_esc, const void *x, const cudnnTensorDescriptor_t y_dsc, void *y);
|
||||
} // namespace mindspore::lite
|
||||
#endif // MINDSPORE_LITE_SRC_DELEGATE_TENSORRT_CDUA_IMPL_CUDNN_UTILS_H_
|
||||
#endif // MINDSPORE_LITE_SRC_EXTENDRT_DELEGATE_TENSORRT_CUDA_IMPL_CUDNN_UTILS_H_
|
||||
|
|
|
@ -14,8 +14,8 @@
|
|||
* limitations under the License.
|
||||
*/
|
||||
|
||||
#ifndef MINDSPORE_LITE_SRC_DELEGATE_TENSORRT_DISTRIBUTION_DISTRIBUTION_BASE_H_
|
||||
#define MINDSPORE_LITE_SRC_DELEGATE_TENSORRT_DISTRIBUTION_DISTRIBUTION_BASE_H_
|
||||
#ifndef MINDSPORE_LITE_SRC_EXTENDRT_DELEGATE_TENSORRT_DISTRIBUTION_DISTRIBUTION_BASE_H_
|
||||
#define MINDSPORE_LITE_SRC_EXTENDRT_DELEGATE_TENSORRT_DISTRIBUTION_DISTRIBUTION_BASE_H_
|
||||
|
||||
#include <string>
|
||||
#include "src/common/log_adapter.h"
|
||||
|
@ -28,4 +28,4 @@ int GetGPUGroupSize();
|
|||
|
||||
int GetRankID();
|
||||
} // namespace mindspore::lite
|
||||
#endif // MINDSPORE_LITE_SRC_DELEGATE_TENSORRT_DISTRIBUTION_DISTRIBUTION_BASE_H_
|
||||
#endif // MINDSPORE_LITE_SRC_EXTENDRT_DELEGATE_TENSORRT_DISTRIBUTION_DISTRIBUTION_BASE_H_
|
||||
|
|
|
@ -13,8 +13,8 @@
|
|||
* See the License for the specific language governing permissions and
|
||||
* limitations under the License.
|
||||
*/
|
||||
#ifndef MINDSPORE_LITE_SRC_DELEGATE_TENSORRT_DISTRIBUTION_DISTRIBUTION_COLLECTIVE_H_
|
||||
#define MINDSPORE_LITE_SRC_DELEGATE_TENSORRT_DISTRIBUTION_DISTRIBUTION_COLLECTIVE_H_
|
||||
#ifndef MINDSPORE_LITE_SRC_EXTENDRT_DELEGATE_TENSORRT_DISTRIBUTION_DISTRIBUTION_COLLECTIVE_H_
|
||||
#define MINDSPORE_LITE_SRC_EXTENDRT_DELEGATE_TENSORRT_DISTRIBUTION_DISTRIBUTION_COLLECTIVE_H_
|
||||
|
||||
#include <string>
|
||||
#include "NvInfer.h"
|
||||
|
@ -42,4 +42,4 @@ class DistributionCollective {
|
|||
~DistributionCollective() = default;
|
||||
};
|
||||
} // namespace mindspore::lite
|
||||
#endif // MINDSPORE_LITE_SRC_DELEGATE_TENSORRT_DISTRIBUTION_DISTRIBUTION_COLLECTIVE_H_
|
||||
#endif // MINDSPORE_LITE_SRC_EXTENDRT_DELEGATE_TENSORRT_DISTRIBUTION_DISTRIBUTION_COLLECTIVE_H_
|
||||
|
|
|
@ -13,8 +13,8 @@
|
|||
* See the License for the specific language governing permissions and
|
||||
* limitations under the License.
|
||||
*/
|
||||
#ifndef MINDSPORE_LITE_SRC_DELEGATE_TENSORRT_DISTRIBUTION_DISTRIBUTION_UTILS_H_
|
||||
#define MINDSPORE_LITE_SRC_DELEGATE_TENSORRT_DISTRIBUTION_DISTRIBUTION_UTILS_H_
|
||||
#ifndef MINDSPORE_LITE_SRC_EXTENDRT_DELEGATE_TENSORRT_DISTRIBUTION_DISTRIBUTION_UTILS_H_
|
||||
#define MINDSPORE_LITE_SRC_EXTENDRT_DELEGATE_TENSORRT_DISTRIBUTION_DISTRIBUTION_UTILS_H_
|
||||
|
||||
#include <nccl.h>
|
||||
#include "include/errorcode.h"
|
||||
|
@ -29,4 +29,4 @@ ncclDataType_t ConvertNCCLDataType(nvinfer1::DataType type_id);
|
|||
|
||||
ncclRedOp_t ConvertNCCLReduceMode(schema::ReduceMode mode);
|
||||
} // namespace mindspore::lite
|
||||
#endif // MINDSPORE_LITE_SRC_DELEGATE_TENSORRT_DISTRIBUTION_DISTRIBUTION_UTILS_H_
|
||||
#endif // MINDSPORE_LITE_SRC_EXTENDRT_DELEGATE_TENSORRT_DISTRIBUTION_DISTRIBUTION_UTILS_H_
|
||||
|
|
|
@ -14,8 +14,8 @@
|
|||
* limitations under the License.
|
||||
*/
|
||||
|
||||
#ifndef MINDSPORE_LITE_SRC_DELEGATE_TENSORRT_OP_ACTIVATION_OPT_PLUGIN_H_
|
||||
#define MINDSPORE_LITE_SRC_DELEGATE_TENSORRT_OP_ACTIVATION_OPT_PLUGIN_H_
|
||||
#ifndef MINDSPORE_LITE_SRC_EXTENDRT_DELEGATE_TENSORRT_OP_ACTIVATION_OPT_PLUGIN_H_
|
||||
#define MINDSPORE_LITE_SRC_EXTENDRT_DELEGATE_TENSORRT_OP_ACTIVATION_OPT_PLUGIN_H_
|
||||
|
||||
#include <string>
|
||||
#include "src/extendrt/delegate/tensorrt/op/tensorrt_op.h"
|
||||
|
@ -69,4 +69,4 @@ class ActivationOptPluginCreater : public TensorRTPluginCreater<ActivationOptPlu
|
|||
ActivationOptPluginCreater() : TensorRTPluginCreater(std::string(ACTIVATION_OPT_PLUGIN_NAME)) {}
|
||||
};
|
||||
} // namespace mindspore::lite
|
||||
#endif // MINDSPORE_LITE_SRC_DELEGATE_TENSORRT_OP_ACTIVATION_OPT_PLUGIN_H_
|
||||
#endif // MINDSPORE_LITE_SRC_EXTENDRT_DELEGATE_TENSORRT_OP_ACTIVATION_OPT_PLUGIN_H_
|
||||
|
|
|
@ -13,8 +13,8 @@
|
|||
* See the License for the specific language governing permissions and
|
||||
* limitations under the License.
|
||||
*/
|
||||
#ifndef MINDSPORE_LITE_SRC_DELEGATE_TENSORRT_OP_ACTIVATION_TENSORRT_H_
|
||||
#define MINDSPORE_LITE_SRC_DELEGATE_TENSORRT_OP_ACTIVATION_TENSORRT_H_
|
||||
#ifndef MINDSPORE_LITE_SRC_EXTENDRT_DELEGATE_TENSORRT_OP_ACTIVATION_TENSORRT_H_
|
||||
#define MINDSPORE_LITE_SRC_EXTENDRT_DELEGATE_TENSORRT_OP_ACTIVATION_TENSORRT_H_
|
||||
#include <string>
|
||||
#include <vector>
|
||||
#include "src/extendrt/delegate/tensorrt/op/tensorrt_op.h"
|
||||
|
@ -40,4 +40,4 @@ class ActivationTensorRT : public TensorRTOp {
|
|||
schema::QuantType quant_type = schema::QuantType_QUANT_NONE);
|
||||
};
|
||||
} // namespace mindspore::lite
|
||||
#endif // MINDSPORE_LITE_SRC_DELEGATE_TENSORRT_OP_ACTIVATION_TENSORRT_H_
|
||||
#endif // MINDSPORE_LITE_SRC_EXTENDRT_DELEGATE_TENSORRT_OP_ACTIVATION_TENSORRT_H_
|
||||
|
|
|
@ -13,8 +13,8 @@
|
|||
* See the License for the specific language governing permissions and
|
||||
* limitations under the License.
|
||||
*/
|
||||
#ifndef MINDSPORE_LITE_SRC_DELEGATE_TENSORRT_OP_ALLGATHER_TENSORRT_H_
|
||||
#define MINDSPORE_LITE_SRC_DELEGATE_TENSORRT_OP_ALLGATHER_TENSORRT_H_
|
||||
#ifndef MINDSPORE_LITE_SRC_EXTENDRT_DELEGATE_TENSORRT_OP_ALLGATHER_TENSORRT_H_
|
||||
#define MINDSPORE_LITE_SRC_EXTENDRT_DELEGATE_TENSORRT_OP_ALLGATHER_TENSORRT_H_
|
||||
#include <string>
|
||||
#include <vector>
|
||||
#include <memory>
|
||||
|
@ -72,4 +72,4 @@ class AllGatherPluginCreater : public TensorRTPluginCreater<AllGatherPlugin> {
|
|||
AllGatherPluginCreater() : TensorRTPluginCreater(std::string(ALLGATHER_PLUGIN_NAME)) {}
|
||||
};
|
||||
} // namespace mindspore::lite
|
||||
#endif // MINDSPORE_LITE_SRC_DELEGATE_TENSORRT_OP_ALLGATHER_TENSORRT_H_
|
||||
#endif // MINDSPORE_LITE_SRC_EXTENDRT_DELEGATE_TENSORRT_OP_ALLGATHER_TENSORRT_H_
|
||||
|
|
|
@ -13,8 +13,8 @@
|
|||
* See the License for the specific language governing permissions and
|
||||
* limitations under the License.
|
||||
*/
|
||||
#ifndef MINDSPORE_LITE_SRC_DELEGATE_TENSORRT_OP_CAST_TENSORRT_H_
|
||||
#define MINDSPORE_LITE_SRC_DELEGATE_TENSORRT_OP_CAST_TENSORRT_H_
|
||||
#ifndef MINDSPORE_LITE_SRC_EXTENDRT_DELEGATE_TENSORRT_OP_CAST_TENSORRT_H_
|
||||
#define MINDSPORE_LITE_SRC_EXTENDRT_DELEGATE_TENSORRT_OP_CAST_TENSORRT_H_
|
||||
#include <string>
|
||||
#include <vector>
|
||||
#include "src/extendrt/delegate/tensorrt/op/tensorrt_op.h"
|
||||
|
@ -83,4 +83,4 @@ class CastPluginCreater : public TensorRTPluginCreater<CastPlugin> {
|
|||
CastPluginCreater() : TensorRTPluginCreater(std::string(CAST_PLUGIN_NAME)) {}
|
||||
};
|
||||
} // namespace mindspore::lite
|
||||
#endif // MINDSPORE_LITE_SRC_DELEGATE_TENSORRT_OP_CAST_TENSORRT_H_
|
||||
#endif // MINDSPORE_LITE_SRC_EXTENDRT_DELEGATE_TENSORRT_OP_CAST_TENSORRT_H_
|
||||
|
|
|
@ -13,8 +13,8 @@
|
|||
* See the License for the specific language governing permissions and
|
||||
* limitations under the License.
|
||||
*/
|
||||
#ifndef MINDSPORE_LITE_SRC_DELEGATE_TENSORRT_OP_CONCATE_TENSORRT_H_
|
||||
#define MINDSPORE_LITE_SRC_DELEGATE_TENSORRT_OP_CONCATE_TENSORRT_H_
|
||||
#ifndef MINDSPORE_LITE_SRC_EXTENDRT_DELEGATE_TENSORRT_OP_CONCATE_TENSORRT_H_
|
||||
#define MINDSPORE_LITE_SRC_EXTENDRT_DELEGATE_TENSORRT_OP_CONCATE_TENSORRT_H_
|
||||
#include <string>
|
||||
#include <vector>
|
||||
#include "src/extendrt/delegate/tensorrt/op/tensorrt_op.h"
|
||||
|
@ -47,4 +47,4 @@ class ConcateTensorRT : public TensorRTOp {
|
|||
int axis_;
|
||||
};
|
||||
} // namespace mindspore::lite
|
||||
#endif // MINDSPORE_LITE_SRC_DELEGATE_TENSORRT_OP_CONCATE_TENSORRT_H_
|
||||
#endif // MINDSPORE_LITE_SRC_EXTENDRT_DELEGATE_TENSORRT_OP_CONCATE_TENSORRT_H_
|
||||
|
|
|
@ -13,8 +13,8 @@
|
|||
* See the License for the specific language governing permissions and
|
||||
* limitations under the License.
|
||||
*/
|
||||
#ifndef MINDSPORE_LITE_SRC_TENSORRT_OP_CONVOLUTION_TENSORRT_H_
|
||||
#define MINDSPORE_LITE_SRC_TENSORRT_OP_CONVOLUTION_TENSORRT_H_
|
||||
#ifndef MINDSPORE_LITE_SRC_EXTENDRT_DELEGATE_TENSORRT_OP_CONVOLUTION_TENSORRT_H_
|
||||
#define MINDSPORE_LITE_SRC_EXTENDRT_DELEGATE_TENSORRT_OP_CONVOLUTION_TENSORRT_H_
|
||||
#include <string>
|
||||
#include <vector>
|
||||
#include "src/extendrt/delegate/tensorrt/op/tensorrt_op.h"
|
||||
|
@ -40,4 +40,4 @@ class ConvolutionTensorRT : public TensorRTOp {
|
|||
void *pack_weight_{nullptr};
|
||||
};
|
||||
} // namespace mindspore::lite
|
||||
#endif // MINDSPORE_LITE_SRC_TENSORRT_OP_CONVOLUTION_TENSORRT_H_
|
||||
#endif // MINDSPORE_LITE_SRC_EXTENDRT_DELEGATE_TENSORRT_OP_CONVOLUTION_TENSORRT_H_
|
||||
|
|
|
@ -13,8 +13,8 @@
|
|||
* See the License for the specific language governing permissions and
|
||||
* limitations under the License.
|
||||
*/
|
||||
#ifndef MINDSPORE_LITE_SRC_TENSORRT_OP_DECONVOLUTION_TENSORRT_H_
|
||||
#define MINDSPORE_LITE_SRC_TENSORRT_OP_DECONVOLUTION_TENSORRT_H_
|
||||
#ifndef MINDSPORE_LITE_SRC_EXTENDRT_DELEGATE_TENSORRT_OP_DECONVOLUTION_TENSORRT_H_
|
||||
#define MINDSPORE_LITE_SRC_EXTENDRT_DELEGATE_TENSORRT_OP_DECONVOLUTION_TENSORRT_H_
|
||||
#include <string>
|
||||
#include <vector>
|
||||
#include "src/extendrt/delegate/tensorrt/op/tensorrt_op.h"
|
||||
|
@ -40,4 +40,4 @@ class DeconvolutionTensorRT : public TensorRTOp {
|
|||
void *pack_weight_{nullptr};
|
||||
};
|
||||
} // namespace mindspore::lite
|
||||
#endif // MINDSPORE_LITE_SRC_TENSORRT_OP_DECONVOLUTION_TENSORRT_H_
|
||||
#endif // MINDSPORE_LITE_SRC_EXTENDRT_DELEGATE_TENSORRT_OP_DECONVOLUTION_TENSORRT_H_
|
||||
|
|
|
@ -13,8 +13,8 @@
|
|||
* See the License for the specific language governing permissions and
|
||||
* limitations under the License.
|
||||
*/
|
||||
#ifndef MINDSPORE_LITE_SRC_DELEGATE_TENSORRT_OP_ELEMENTWISE_TENSORRT_H_
|
||||
#define MINDSPORE_LITE_SRC_DELEGATE_TENSORRT_OP_ELEMENTWISE_TENSORRT_H_
|
||||
#ifndef MINDSPORE_LITE_SRC_EXTENDRT_DELEGATE_TENSORRT_OP_ELEMENTWISE_TENSORRT_H_
|
||||
#define MINDSPORE_LITE_SRC_EXTENDRT_DELEGATE_TENSORRT_OP_ELEMENTWISE_TENSORRT_H_
|
||||
#include <string>
|
||||
#include <vector>
|
||||
#include <map>
|
||||
|
@ -47,4 +47,4 @@ class ElementWiseTensorRT : public TensorRTOp {
|
|||
nvinfer1::ElementWiseOperation element_wise_op_;
|
||||
};
|
||||
} // namespace mindspore::lite
|
||||
#endif // MINDSPORE_LITE_SRC_DELEGATE_TENSORRT_OP_ELEMENTWISE_TENSORRT_H_
|
||||
#endif // MINDSPORE_LITE_SRC_EXTENDRT_DELEGATE_TENSORRT_OP_ELEMENTWISE_TENSORRT_H_
|
||||
|
|
|
@ -13,8 +13,8 @@
|
|||
* See the License for the specific language governing permissions and
|
||||
* limitations under the License.
|
||||
*/
|
||||
#ifndef MINDSPORE_LITE_SRC_DELEGATE_TENSORRT_OP_EQUAL_TENSORRT_H_
|
||||
#define MINDSPORE_LITE_SRC_DELEGATE_TENSORRT_OP_EQUAL_TENSORRT_H_
|
||||
#ifndef MINDSPORE_LITE_SRC_EXTENDRT_DELEGATE_TENSORRT_OP_EQUAL_TENSORRT_H_
|
||||
#define MINDSPORE_LITE_SRC_EXTENDRT_DELEGATE_TENSORRT_OP_EQUAL_TENSORRT_H_
|
||||
#include <string>
|
||||
#include <vector>
|
||||
#include "src/extendrt/delegate/tensorrt/op/tensorrt_op.h"
|
||||
|
@ -59,4 +59,4 @@ class EqualPluginCreater : public TensorRTPluginCreater<EqualPlugin> {
|
|||
EqualPluginCreater() : TensorRTPluginCreater(std::string(EQUAL_PLUGIN_NAME)) {}
|
||||
};
|
||||
} // namespace mindspore::lite
|
||||
#endif // MINDSPORE_LITE_SRC_DELEGATE_TENSORRT_OP_EQUAL_TENSORRT_H_
|
||||
#endif // MINDSPORE_LITE_SRC_EXTENDRT_DELEGATE_TENSORRT_OP_EQUAL_TENSORRT_H_
|
||||
|
|
|
@ -13,8 +13,8 @@
|
|||
* See the License for the specific language governing permissions and
|
||||
* limitations under the License.
|
||||
*/
|
||||
#ifndef MINDSPORE_LITE_SRC_DELEGATE_TENSORRT_OP_FULLYCONNECTED_TENSORRT_H_
|
||||
#define MINDSPORE_LITE_SRC_DELEGATE_TENSORRT_OP_FULLYCONNECTED_TENSORRT_H_
|
||||
#ifndef MINDSPORE_LITE_SRC_EXTENDRT_DELEGATE_TENSORRT_OP_FULLYCONNECTED_TENSORRT_H_
|
||||
#define MINDSPORE_LITE_SRC_EXTENDRT_DELEGATE_TENSORRT_OP_FULLYCONNECTED_TENSORRT_H_
|
||||
|
||||
#include <string>
|
||||
#include <vector>
|
||||
|
@ -42,4 +42,4 @@ class FullyConnectedTensorRT : public TensorRTOp {
|
|||
schema::ActivationType activation_{schema::ActivationType::ActivationType_NO_ACTIVATION};
|
||||
};
|
||||
} // namespace mindspore::lite
|
||||
#endif // MINDSPORE_LITE_SRC_DELEGATE_TENSORRT_OP_FULLYCONNECTED_TENSORRT_H_
|
||||
#endif // MINDSPORE_LITE_SRC_EXTENDRT_DELEGATE_TENSORRT_OP_FULLYCONNECTED_TENSORRT_H_
|
||||
|
|
Some files were not shown because too many files have changed in this diff Show More
Loading…
Reference in New Issue