Skip to content

MegEngine
MegEngine

MegEngine 天元 / MegEngine
1 年多 前同步成功

通知 404
Star 4705
Fork 582
MegEngine
MegEngine
提交 f9719828 编写于 作者: M Megvii Engine Team
浏览文件
操作

feat(lite): add header licence 

GitOrigin-RevId: bcd6860e8c8e8cdaaaeae5340284650547fb6a4b
上级 71230e9a
显示空白变更内容
内联 并排
Showing with 414 addition and 322 deletion
ACKNOWLEDGMENTS
浏览文件 @ f9719828
......@@ -50,6 +50,9 @@ Copyright 2016-2020 Advanced Micro Devices, Inc.
14. rocRAND
Copyright (c) 2017 Advanced Micro Devices, Inc. All rights reserved.
15. json
Copyright (c) 2013-2021 Niels Lohmann
Terms of the MIT License:
--------------------------------------------------------------------
Permission is hereby granted, free of charge, to any person obtaining a copy of this software and associated documentation files (the "Software"), to deal in the Software without restriction, including without limitation the rights to use, copy, modify, merge, publish, distribute, sublicense, and/or sell copies of the Software, and to permit persons to whom the Software is furnished to do so, subject to the following conditions:
......
CMakeLists.txt
浏览文件 @ f9719828
......@@ -43,6 +43,7 @@ option(MGE_ARMV8_2_FEATURE_FP16 "Enable armv8.2-a+fp16 support" OFF)
option(MGE_DISABLE_FLOAT16 "Disable MegEngine float16 support." OFF)
option(MGE_WITH_CUDA "Enable MegEngine CUDA support." ON)
option(MGE_CUDA_USE_STATIC "Enable MegEngine CUDA static linking." ON)
option(MGE_WITH_LITE "Build MGE with lite" ON)
option(MGE_WITH_TRT "Build MegEngine with TensorRT." ON)
option(MGE_WITH_CUDA_STUB "Build MegEngine with CUDA stub." ON)
option(MGE_WITH_NVRTC_STUB "Build MegEngine with NVRTC stub." OFF)
......@@ -1179,3 +1180,7 @@ if(MGE_WITH_CUDA AND MGE_CUDA_USE_STATIC AND("${CUDNN_VERSION}" VERSION_GREATER
message(WARNING "Static link CUDNN8 with many sm is unworkable, please use -DMGE_WITH_CUDNN_SHARED=ON or -DMGE_WITH_LARGE_ARCHIVE=ON -DMGE_CUDA_GENCODE=\"-gencode arch=compute_70,code=sm_70 arch=compute_75,code=sm_75\" ")
message(WARNING "Static link CUDNN8 with many sm is unworkable, please use -DMGE_WITH_CUDNN_SHARED=ON or -DMGE_WITH_LARGE_ARCHIVE=ON -DMGE_CUDA_GENCODE=\"-gencode arch=compute_70,code=sm_70 arch=compute_75,code=sm_75\" ")
endif()
if(MGE_WITH_LITE)
add_subdirectory(lite)
endif()
lite/README.md
浏览文件 @ f9719828
......@@ -3,76 +3,7 @@
It is a lite warper of MegEngine, to enable MegEngine easy to be integrated in
user's SDK
## bazel build
目前支持内部 bazel 和 CMake 编译，支持 C++/C, Python 接口，
下面是 bazel 中 lite_shared 目标的编译，可以作为其他目标的编译的参考，
该编译依赖内部 bazel 编译以及 megvii3。
### 配置编译环境
需要使用 megvii3 workspace 来完成 bazel 的编译
#### Clone megvii3 安装 bazel
```bash
git clone git@git-core.megvii-inc.com:brain-sdk/megvii3.git
./utils/bazel/get_bazel.sh
```
#### Clone megbrain
```
git submodule update brain/megbrain brain/midout
```
### 编译 x86 CUDA 版本
```bash
./bazel build //brain/megbrain/lite:lite_shared --cpu="k8" \
--compiler="gcc7_cuda10" -c opt
```
### 编译 x86 CPU 版本
```bash
./bazel build //brain/megbrain/lite:lite_shared --cpu="k8" \
--compiler="gcc9" -c opt
```
### 编译 arm OpenCL 版本
```bash
./bazel build //brain/megbrain/lite:lite_shared_shared --cpu=android_aarch64 \
-c opt --define enable_opencl=1 --define enable_opencl_search=1
```
### 编译 arm opencl lite_examples
bazel-3.0.0-megvii2 build //brain/megbrain/lite:lite_shared_examples \
--cpu=android_aarch64 --define enable_opencl=1 --define enable_opencl_search=1
####如何运行snpe_loder 的lite_exampes 请查看下面的wiki
https://wiki.megvii-inc.com/pages/viewpage.action?pageId=268786906
### 编译 armv7 CPU 版本
```bash
./bazel build //brain/megbrain/lite:lite_shared --cpu=android_armv7 \
-c opt
```
### 编译 arm64 CPU 版本
```bash
./bazel build //brain/megbrain/lite:lite_shared --cpu=android_aarch64 \
-c opt
```
### 编译 arm64 CPU v8.2 版本
```bash
./bazel build //brain/megbrain/lite:lite_shared --cpu=android_aarch64 \
--copt -march=armv8.2-a+fp16+dotprod -c opt
```
## 同时支持cmake构建
## 支持cmake构建
cmake构建参考scripts/cmake-build/BUILD_README.md,下面example表示同时支持编译megengine
和RKNPU后端且打开OpenCL的release模式
```bash
......@@ -235,17 +166,3 @@ Lite 中提供自定义的方式，用户可以自定义自己 Info 部分的类
aes_encrypt.sh xxx.mdl xxx_encrypted.mdl \
000102030405060708090A0B0C0D0E0F101112131415161718191A1B1C1D1E1F
```
* rc4_encypt.cpp 可以被编译成为一个 rc4 加密的工具，这个工具可以通过
制定的 key 或者默认的 key 加密制定的文件，支持 rc4 方法和
simple_fast_rc4 两种方法，支持自定义 key。
* bazel 编译 x86 命令为：
```bash
bazel build //brain/megbrain/lite:rc4_encryptor \
--cpu='k8' --compiler='gcc9'
```
* 加密文件，具体用法见 help
```bash
rc4_encryptor encrypt_predefined_rc4 \
to_be_encrypt.file encrypted.file
```
lite/example/example.h
浏览文件 @ f9719828
/**
* \file example/example.cpp
* MegEngine is Licensed under the Apache License, Version 2.0 (the "License")
*
* This file is part of MegEngine, a deep learning framework developed by
* Megvii.
* Copyright (c) 2014-2021 Megvii Inc. All rights reserved.
*
* \copyright Copyright (c) 2020-2021 Megvii Inc. All rights reserved.
* Unless required by applicable law or agreed to in writing,
* software distributed under the License is distributed on an
* "AS IS" BASIS, WITHOUT ARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
*/
#pragma once
......
lite/example/main.cpp
浏览文件 @ f9719828
/**
* \file example/example.cpp
* MegEngine is Licensed under the Apache License, Version 2.0 (the "License")
*
* This file is part of MegEngine, a deep learning framework developed by
* Megvii.
* Copyright (c) 2014-2021 Megvii Inc. All rights reserved.
*
* \copyright Copyright (c) 2020-2021 Megvii Inc. All rights reserved.
* Unless required by applicable law or agreed to in writing,
* software distributed under the License is distributed on an
* "AS IS" BASIS, WITHOUT ARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
*/
#include "lite/global.h"
......
lite/example/mge/basic.cpp
浏览文件 @ f9719828
/**
* \file example/basic.cpp
* MegEngine is Licensed under the Apache License, Version 2.0 (the "License")
*
* This file is part of MegEngine, a deep learning framework developed by
* Megvii.
* Copyright (c) 2014-2021 Megvii Inc. All rights reserved.
*
* \copyright Copyright (c) 2020-2021 Megvii Inc. All rights reserved.
* Unless required by applicable law or agreed to in writing,
* software distributed under the License is distributed on an
* "AS IS" BASIS, WITHOUT ARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
*/
#include <thread>
......
lite/example/mge/cpu_affinity.cpp
浏览文件 @ f9719828
/**
* \file example/cpu_affinity.cpp
* MegEngine is Licensed under the Apache License, Version 2.0 (the "License")
*
* This file is part of MegEngine, a deep learning framework developed by
* Megvii.
* Copyright (c) 2014-2021 Megvii Inc. All rights reserved.
*
* \copyright Copyright (c) 2020-2021 Megvii Inc. All rights reserved.
* Unless required by applicable law or agreed to in writing,
* software distributed under the License is distributed on an
* "AS IS" BASIS, WITHOUT ARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
*/
#include "../example.h"
......
lite/example/mge/device_io.cpp
浏览文件 @ f9719828
/**
* \file example/device_io.cpp
* MegEngine is Licensed under the Apache License, Version 2.0 (the "License")
*
* This file is part of MegEngine, a deep learning framework developed by
* Megvii.
* Copyright (c) 2014-2021 Megvii Inc. All rights reserved.
*
* \copyright Copyright (c) 2020-2021 Megvii Inc. All rights reserved.
* Unless required by applicable law or agreed to in writing,
* software distributed under the License is distributed on an
* "AS IS" BASIS, WITHOUT ARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
*/
#include <thread>
......
lite/example/mge/lite_c_interface.cpp
浏览文件 @ f9719828
/**
* \file example/basic_c_interface.cpp
* MegEngine is Licensed under the Apache License, Version 2.0 (the "License")
*
* This file is part of MegEngine, a deep learning framework developed by
* Megvii.
* Copyright (c) 2014-2021 Megvii Inc. All rights reserved.
*
* \copyright Copyright (c) 2020-2021 Megvii Inc. All rights reserved.
* Unless required by applicable law or agreed to in writing,
* software distributed under the License is distributed on an
* "AS IS" BASIS, WITHOUT ARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
*/
#include "../example.h"
......
lite/example/mge/network_share_weights.cpp
浏览文件 @ f9719828
/**
* \file example/network_share_weights.cpp
* MegEngine is Licensed under the Apache License, Version 2.0 (the "License")
*
* This file is part of MegEngine, a deep learning framework developed by
* Megvii.
* Copyright (c) 2014-2021 Megvii Inc. All rights reserved.
*
* \copyright Copyright (c) 2020-2021 Megvii Inc. All rights reserved.
* Unless required by applicable law or agreed to in writing,
* software distributed under the License is distributed on an
* "AS IS" BASIS, WITHOUT ARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
*/
#include "../example.h"
......
lite/example/mge/reset_io.cpp
浏览文件 @ f9719828
/**
* \file example/reset_io.cpp
* MegEngine is Licensed under the Apache License, Version 2.0 (the "License")
*
* This file is part of MegEngine, a deep learning framework developed by
* Megvii.
* Copyright (c) 2014-2021 Megvii Inc. All rights reserved.
*
* \copyright Copyright (c) 2020-2021 Megvii Inc. All rights reserved.
* Unless required by applicable law or agreed to in writing,
* software distributed under the License is distributed on an
* "AS IS" BASIS, WITHOUT ARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
*/
#include "../example.h"
......
lite/example/mge/user_allocator.cpp
浏览文件 @ f9719828
/**
* \file example/user_allocator.cpp
* MegEngine is Licensed under the Apache License, Version 2.0 (the "License")
*
* This file is part of MegEngine, a deep learning framework developed by
* Megvii.
* Copyright (c) 2014-2021 Megvii Inc. All rights reserved.
*
* \copyright Copyright (c) 2020-2021 Megvii Inc. All rights reserved.
* Unless required by applicable law or agreed to in writing,
* software distributed under the License is distributed on an
* "AS IS" BASIS, WITHOUT ARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
*/
#include "../example.h"
......
lite/example/mge/user_cryption.cpp
浏览文件 @ f9719828
/**
* \file example/user_cryption.cpp
* MegEngine is Licensed under the Apache License, Version 2.0 (the "License")
*
* This file is part of MegEngine, a deep learning framework developed by
* Megvii.
* Copyright (c) 2014-2021 Megvii Inc. All rights reserved.
*
* \copyright Copyright (c) 2020-2021 Megvii Inc. All rights reserved.
* Unless required by applicable law or agreed to in writing,
* software distributed under the License is distributed on an
* "AS IS" BASIS, WITHOUT ARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
*/
#include "../example.h"
......
lite/include/lite/common_enum_c.h
浏览文件 @ f9719828
/**
* \file inlude/lite/common_enum_c.h
* MegEngine is Licensed under the Apache License, Version 2.0 (the "License")
*
* This file is part of MegEngine, a deep learning framework developed by
* Megvii.
* Copyright (c) 2014-2021 Megvii Inc. All rights reserved.
*
* \copyright Copyright (c) 2020-2021 Megvii Inc. All rights reserved.
* Unless required by applicable law or agreed to in writing,
* software distributed under the License is distributed on an
* "AS IS" BASIS, WITHOUT ARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
*/
#ifndef LITE_COMMON_ENUM_C_H_
......
lite/include/lite/global.h
浏览文件 @ f9719828
/**
* \file inlude/lite/global.h
* MegEngine is Licensed under the Apache License, Version 2.0 (the "License")
*
* This file is part of MegEngine, a deep learning framework developed by
* Megvii.
* Copyright (c) 2014-2021 Megvii Inc. All rights reserved.
*
* \copyright Copyright (c) 2020-2021 Megvii Inc. All rights reserved.
* Unless required by applicable law or agreed to in writing,
* software distributed under the License is distributed on an
* "AS IS" BASIS, WITHOUT ARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
*/
#pragma once
......
lite/include/lite/macro.h
浏览文件 @ f9719828
/**
* \file include/lite/macro.h
* MegEngine is Licensed under the Apache License, Version 2.0 (the "License")
*
* This file is part of MegEngine, a deep learning framework developed by
* Megvii.
* Copyright (c) 2014-2021 Megvii Inc. All rights reserved.
*
* \copyright Copyright (c) 2020-2021 Megvii Inc. All rights reserved.
* Unless required by applicable law or agreed to in writing,
* software distributed under the License is distributed on an
* "AS IS" BASIS, WITHOUT ARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
*/
#ifndef LITE_MACRO_H_
......
lite/include/lite/network.h
浏览文件 @ f9719828
/**
* \file inlude/lite/network.h
* MegEngine is Licensed under the Apache License, Version 2.0 (the "License")
*
* This file is part of MegEngine, a deep learning framework developed by
* Megvii.
* Copyright (c) 2014-2021 Megvii Inc. All rights reserved.
*
* \copyright Copyright (c) 2020-2021 Megvii Inc. All rights reserved.
* Unless required by applicable law or agreed to in writing,
* software distributed under the License is distributed on an
* "AS IS" BASIS, WITHOUT ARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
*/
#pragma once
......
lite/include/lite/tensor.h
浏览文件 @ f9719828
/**
* \file inlude/lite/tensor.h
* MegEngine is Licensed under the Apache License, Version 2.0 (the "License")
*
* This file is part of MegEngine, a deep learning framework developed by
* Megvii.
* Copyright (c) 2014-2021 Megvii Inc. All rights reserved.
*
* \copyright Copyright (c) 2020-2021 Megvii Inc. All rights reserved.
* Unless required by applicable law or agreed to in writing,
* software distributed under the License is distributed on an
* "AS IS" BASIS, WITHOUT ARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
*/
#pragma once
......
lite/lite-c/include/lite-c/global_c.h
浏览文件 @ f9719828
/**
* \file lite-c/include/lite-c/global-c.h
* MegEngine is Licensed under the Apache License, Version 2.0 (the "License")
*
* This file is part of MegEngine, a deep learning framework developed by
* Megvii.
* Copyright (c) 2014-2021 Megvii Inc. All rights reserved.
*
* \copyright Copyright (c) 2020-2021 Megvii Inc. All rights reserved.
* Unless required by applicable law or agreed to in writing,
* software distributed under the License is distributed on an
* "AS IS" BASIS, WITHOUT ARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
*/
#ifndef LITE_C_GLOBAL_H_
......
lite/lite-c/include/lite-c/network_c.h
浏览文件 @ f9719828
/**
* \file lite-c/include/lite-c/network_c.h
* MegEngine is Licensed under the Apache License, Version 2.0 (the "License")
*
* This file is part of MegEngine, a deep learning framework developed by
* Megvii.
* Copyright (c) 2014-2021 Megvii Inc. All rights reserved.
*
* \copyright Copyright (c) 2020-2021 Megvii Inc. All rights reserved.
* Unless required by applicable law or agreed to in writing,
* software distributed under the License is distributed on an
* "AS IS" BASIS, WITHOUT ARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
*/
#ifndef LITE_C_NETWORK_H_
......
lite/lite-c/include/lite-c/tensor_c.h
浏览文件 @ f9719828
/**
* \file lite-c/include/lite-c/tensor_c.h
* MegEngine is Licensed under the Apache License, Version 2.0 (the "License")
*
* This file is part of MegEngine, a deep learning framework developed by
* Megvii.
* Copyright (c) 2014-2021 Megvii Inc. All rights reserved.
*
* \copyright Copyright (c) 2020-2021 Megvii Inc. All rights reserved.
* Unless required by applicable law or agreed to in writing,
* software distributed under the License is distributed on an
* "AS IS" BASIS, WITHOUT ARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
*/
#ifndef LITE_TENSOR_C_H_
......
lite/lite-c/src/common.h
浏览文件 @ f9719828
/**
* \file lite-c/src/common.h
* MegEngine is Licensed under the Apache License, Version 2.0 (the "License")
*
* This file is part of MegEngine, a deep learning framework developed by
* Megvii.
* Copyright (c) 2014-2021 Megvii Inc. All rights reserved.
*
* \copyright Copyright (c) 2020-2021 Megvii Inc. All rights reserved.
* Unless required by applicable law or agreed to in writing,
* software distributed under the License is distributed on an
* "AS IS" BASIS, WITHOUT ARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
*/
#ifndef LITE_C_COMMON_H_
......
lite/lite-c/src/global.cpp
浏览文件 @ f9719828
/**
* \file lite-c/src/tensor.cpp
* MegEngine is Licensed under the Apache License, Version 2.0 (the "License")
*
* This file is part of MegEngine, a deep learning framework developed by
* Megvii.
* Copyright (c) 2014-2021 Megvii Inc. All rights reserved.
*
* \copyright Copyright (c) 2020-2021 Megvii Inc. All rights reserved.
* Unless required by applicable law or agreed to in writing,
* software distributed under the License is distributed on an
* "AS IS" BASIS, WITHOUT ARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
*/
#include "lite/global.h"
......
lite/lite-c/src/network.cpp
浏览文件 @ f9719828
/**
* \file lite-c/src/network.cpp
* MegEngine is Licensed under the Apache License, Version 2.0 (the "License")
*
* This file is part of MegEngine, a deep learning framework developed by
* Megvii.
* Copyright (c) 2014-2021 Megvii Inc. All rights reserved.
*
* \copyright Copyright (c) 2020-2021 Megvii Inc. All rights reserved.
* Unless required by applicable law or agreed to in writing,
* software distributed under the License is distributed on an
* "AS IS" BASIS, WITHOUT ARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
*/
#include "lite/network.h"
......
lite/lite-c/src/tensor.cpp
浏览文件 @ f9719828
/**
* \file lite-c/src/tensor.cpp
* MegEngine is Licensed under the Apache License, Version 2.0 (the "License")
*
* This file is part of MegEngine, a deep learning framework developed by
* Megvii.
* Copyright (c) 2014-2021 Megvii Inc. All rights reserved.
*
* \copyright Copyright (c) 2020-2021 Megvii Inc. All rights reserved.
* Unless required by applicable law or agreed to in writing,
* software distributed under the License is distributed on an
* "AS IS" BASIS, WITHOUT ARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
*/
#include "lite/tensor.h"
......
lite/pylite/megenginelite/__init__.py
浏览文件 @ f9719828
# -*- coding: utf-8 -*-
# This file is part of MegEngine, a deep learning framework developed by
# Megvii.
# MegEngine is Licensed under the Apache License, Version 2.0 (the "License")
#
# Copyright (c) Copyright (c) 2020-2021 Megvii Inc. All rights reserved.
# Copyright (c) 2014-2021 Megvii Inc. All rights reserved.
#
# Unless required by applicable law or agreed to in writing,
# software distributed under the License is distributed on an
# "AS IS" BASIS, WITHOUT ARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
from .base import *
from .global_setting import *
......
lite/pylite/megenginelite/base.py
浏览文件 @ f9719828
# -*- coding: utf-8 -*-
# This file is part of MegEngine, a deep learning framework developed by
# Megvii.
# MegEngine is Licensed under the Apache License, Version 2.0 (the "License")
#
# Copyright (c) Copyright (c) 2020-2021 Megvii Inc. All rights reserved.
# Copyright (c) 2014-2021 Megvii Inc. All rights reserved.
#
# Unless required by applicable law or agreed to in writing,
# software distributed under the License is distributed on an
# "AS IS" BASIS, WITHOUT ARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
import ctypes
import glob
......
lite/pylite/megenginelite/global_setting.py
浏览文件 @ f9719828
# -*- coding: utf-8 -*-
# This file is part of MegEngine, a deep learning framework developed by
# Megvii.
# MegEngine is Licensed under the Apache License, Version 2.0 (the "License")
#
# Copyright (c) Copyright (c) 2020-2021 Megvii Inc. All rights reserved.
# Copyright (c) 2014-2021 Megvii Inc. All rights reserved.
#
# Unless required by applicable law or agreed to in writing,
# software distributed under the License is distributed on an
# "AS IS" BASIS, WITHOUT ARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
from ctypes import *
......
lite/pylite/megenginelite/network.py
浏览文件 @ f9719828
# -*- coding: utf-8 -*-
# This file is part of MegEngine, a deep learning framework developed by
# Megvii.
# MegEngine is Licensed under the Apache License, Version 2.0 (the "License")
#
# Copyright (c) Copyright (c) 2020-2021 Megvii Inc. All rights reserved.
# Copyright (c) 2014-2021 Megvii Inc. All rights reserved.
#
# Unless required by applicable law or agreed to in writing,
# software distributed under the License is distributed on an
# "AS IS" BASIS, WITHOUT ARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
from ctypes import *
......
lite/pylite/megenginelite/struct.py
浏览文件 @ f9719828
# -*- coding: utf-8 -*-
# This file is part of MegEngine, a deep learning framework developed by
# Megvii.
# MegEngine is Licensed under the Apache License, Version 2.0 (the "License")
#
# Copyright (c) Copyright (c) 2020-2021 Megvii Inc. All rights reserved.
# Copyright (c) 2014-2021 Megvii Inc. All rights reserved.
#
# Unless required by applicable law or agreed to in writing,
# software distributed under the License is distributed on an
# "AS IS" BASIS, WITHOUT ARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
import logging
from ctypes import *
......
lite/pylite/megenginelite/tensor.py
浏览文件 @ f9719828
# -*- coding: utf-8 -*-
# This file is part of MegEngine, a deep learning framework developed by
# Megvii.
# MegEngine is Licensed under the Apache License, Version 2.0 (the "License")
#
# Copyright (c) Copyright (c) 2020-2021 Megvii Inc. All rights reserved.
# Copyright (c) 2014-2021 Megvii Inc. All rights reserved.
#
# Unless required by applicable law or agreed to in writing,
# software distributed under the License is distributed on an
# "AS IS" BASIS, WITHOUT ARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
from ctypes import *
......
lite/pylite/megenginelite/utils.py
浏览文件 @ f9719828
# -*- coding: utf-8 -*-
# This file is part of MegEngine, a deep learning framework developed by
# Megvii.
# MegEngine is Licensed under the Apache License, Version 2.0 (the "License")
#
# Copyright (c) Copyright (c) 2020-2021 Megvii Inc. All rights reserved.
# Copyright (c) 2014-2021 Megvii Inc. All rights reserved.
#
# Unless required by applicable law or agreed to in writing,
# software distributed under the License is distributed on an
# "AS IS" BASIS, WITHOUT ARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
import threading
import numpy as np
......
lite/pylite/setup.py
浏览文件 @ f9719828
# -*- coding: utf-8 -*-
# This file is part of MegEngine, a deep learning framework developed by
# Megvii.
# MegEngine is Licensed under the Apache License, Version 2.0 (the "License")
#
# Copyright (c) Copyright (c) 2020-2021 Megvii Inc. All rights reserved.
# Copyright (c) 2014-2021 Megvii Inc. All rights reserved.
#
# Unless required by applicable law or agreed to in writing,
# software distributed under the License is distributed on an
# "AS IS" BASIS, WITHOUT ARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
import os
import re
......
lite/pylite/test/test_global.py
浏览文件 @ f9719828
# -*- coding: utf-8 -*-
# This file is part of MegEngine, a deep learning framework developed by
# Megvii.
# MegEngine is Licensed under the Apache License, Version 2.0 (the "License")
#
# Copyright (c) Copyright (c) 2020-2021 Megvii Inc. All rights reserved.
# Copyright (c) 2014-2021 Megvii Inc. All rights reserved.
#
# Unless required by applicable law or agreed to in writing,
# software distributed under the License is distributed on an
# "AS IS" BASIS, WITHOUT ARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
import os
import unittest
......
lite/pylite/test/test_network.py
浏览文件 @ f9719828
# -*- coding: utf-8 -*-
# This file is part of MegEngine, a deep learning framework developed by
# Megvii.
# MegEngine is Licensed under the Apache License, Version 2.0 (the "License")
#
# Copyright (c) Copyright (c) 2020-2021 Megvii Inc. All rights reserved.
# Copyright (c) 2014-2021 Megvii Inc. All rights reserved.
#
# Unless required by applicable law or agreed to in writing,
# software distributed under the License is distributed on an
# "AS IS" BASIS, WITHOUT ARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
import os
import unittest
......
lite/pylite/test/test_network_cuda.py
浏览文件 @ f9719828
# -*- coding: utf-8 -*-
# This file is part of MegEngine, a deep learning framework developed by
# Megvii.
# MegEngine is Licensed under the Apache License, Version 2.0 (the "License")
#
# Copyright (c) Copyright (c) 2020-2021 Megvii Inc. All rights reserved.
# Copyright (c) 2014-2021 Megvii Inc. All rights reserved.
#
# Unless required by applicable law or agreed to in writing,
# software distributed under the License is distributed on an
# "AS IS" BASIS, WITHOUT ARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
import functools
import os
......
lite/pylite/test/test_tensor.py
浏览文件 @ f9719828
# -*- coding: utf-8 -*-
# This file is part of MegEngine, a deep learning framework developed by
# Megvii.
# MegEngine is Licensed under the Apache License, Version 2.0 (the "License")
#
# Copyright (c) Copyright (c) 2020-2021 Megvii Inc. All rights reserved.
# Copyright (c) 2014-2021 Megvii Inc. All rights reserved.
#
# Unless required by applicable law or agreed to in writing,
# software distributed under the License is distributed on an
# "AS IS" BASIS, WITHOUT ARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
import functools
......
lite/pylite/test/test_utils.py
浏览文件 @ f9719828
# -*- coding: utf-8 -*-
# This file is part of MegEngine, a deep learning framework developed by
# Megvii.
# MegEngine is Licensed under the Apache License, Version 2.0 (the "License")
#
# Copyright (c) Copyright (c) 2020-2021 Megvii Inc. All rights reserved.
# Copyright (c) 2014-2021 Megvii Inc. All rights reserved.
#
# Unless required by applicable law or agreed to in writing,
# software distributed under the License is distributed on an
# "AS IS" BASIS, WITHOUT ARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
import functools
......
lite/src/decryption/aes_decrypt.h
浏览文件 @ f9719828
/**
* \file src/decryption/aes_decrypt.h
* MegEngine is Licensed under the Apache License, Version 2.0 (the "License")
*
* This file is part of MegEngine, a deep learning framework developed by
* Megvii.
* Copyright (c) 2014-2021 Megvii Inc. All rights reserved.
*
* \copyright Copyright (c) 2020-2021 Megvii Inc. All rights reserved.
* Unless required by applicable law or agreed to in writing,
* software distributed under the License is distributed on an
* "AS IS" BASIS, WITHOUT ARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
*/
#include "./mbedtls/aes.h"
......
lite/src/decryption/decrypt_base.h
浏览文件 @ f9719828
......@@ -2,10 +2,11 @@
* \file src/decryption/decrypt_base.h
* MegEngine is Licensed under the Apache License, Version 2.0 (the "License")
*
* This file is part of MegEngine, a deep learning framework developed by
* Megvii.
* Copyright (c) 2014-2021 Megvii Inc. All rights reserved.
*
* \copyright Copyright (c) 2020-2021 Megvii Inc. All rights reserved.
* Unless required by applicable law or agreed to in writing,
* software distributed under the License is distributed on an
* "AS IS" BASIS, WITHOUT ARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
*/
#pragma once
......
lite/src/decryption/rc4/rc4_cryption_base.h
浏览文件 @ f9719828
/**
* \file src/decryption/rc4/rc4_cryption_base.h
* MegEngine is Licensed under the Apache License, Version 2.0 (the "License")
*
* This file is part of MegEngine, a deep learning framework developed by
* Megvii.
* Copyright (c) 2014-2021 Megvii Inc. All rights reserved.
*
* \copyright Copyright (c) 2020-2021 Megvii Inc. All rights reserved.
* Unless required by applicable law or agreed to in writing,
* software distributed under the License is distributed on an
* "AS IS" BASIS, WITHOUT ARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
*/
#pragma once
......
lite/src/decryption/rc4/rc4_cryption_impl.cpp
浏览文件 @ f9719828
/**
* \file src/decryption/rc4/rc4_cryption_impl.cpp
* MegEngine is Licensed under the Apache License, Version 2.0 (the "License")
*
* This file is part of MegEngine, a deep learning framework developed by
* Megvii.
* Copyright (c) 2014-2021 Megvii Inc. All rights reserved.
*
* \copyright Copyright (c) 2020-2021 Megvii Inc. All rights reserved.
* Unless required by applicable law or agreed to in writing,
* software distributed under the License is distributed on an
* "AS IS" BASIS, WITHOUT ARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
*/
#include "rc4_cryption_impl.h"
......
lite/src/decryption/rc4/rc4_cryption_impl.h
浏览文件 @ f9719828
/**
* \file src/decryption/rc4/rc4_cryption_impl.h
* MegEngine is Licensed under the Apache License, Version 2.0 (the "License")
*
* This file is part of MegEngine, a deep learning framework developed by
* Megvii.
* Copyright (c) 2014-2021 Megvii Inc. All rights reserved.
*
* \copyright Copyright (c) 2020-2021 Megvii Inc. All rights reserved.
* Unless required by applicable law or agreed to in writing,
* software distributed under the License is distributed on an
* "AS IS" BASIS, WITHOUT ARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
*/
#pragma once
#include "rc4_cryption_base.h"
......
lite/src/decryption/rc4_cryption.cpp
浏览文件 @ f9719828
/**
* \file src/decryption/rc4_cryption.cpp
* MegEngine is Licensed under the Apache License, Version 2.0 (the "License")
*
* This file is part of MegEngine, a deep learning framework developed by
* Megvii.
* Copyright (c) 2014-2021 Megvii Inc. All rights reserved.
*
* \copyright Copyright (c) 2020-2021 Megvii Inc. All rights reserved.
* Unless required by applicable law or agreed to in writing,
* software distributed under the License is distributed on an
* "AS IS" BASIS, WITHOUT ARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
*/
#include "rc4_cryption.h"
......
lite/src/decryption/rc4_cryption.h
浏览文件 @ f9719828
/**
* \file src/decryption/rc4_cryption.h
* MegEngine is Licensed under the Apache License, Version 2.0 (the "License")
*
* This file is part of MegEngine, a deep learning framework developed by
* Megvii.
* Copyright (c) 2014-2021 Megvii Inc. All rights reserved.
*
* \copyright Copyright (c) 2020-2021 Megvii Inc. All rights reserved.
* Unless required by applicable law or agreed to in writing,
* software distributed under the License is distributed on an
* "AS IS" BASIS, WITHOUT ARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
*/
#pragma once
......
lite/src/function_base.h
浏览文件 @ f9719828
/**
* \file src/function_base.h
* MegEngine is Licensed under the Apache License, Version 2.0 (the "License")
*
* This file is part of MegEngine, a deep learning framework developed by
* Megvii.
* Copyright (c) 2014-2021 Megvii Inc. All rights reserved.
*
* \copyright Copyright (c) 2020-2021 Megvii Inc. All rights reserved.
* Unless required by applicable law or agreed to in writing,
* software distributed under the License is distributed on an
* "AS IS" BASIS, WITHOUT ARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
*/
#pragma once
......
lite/src/global.cpp
浏览文件 @ f9719828
/**
* \file src/global.cpp
* MegEngine is Licensed under the Apache License, Version 2.0 (the "License")
*
* This file is part of MegEngine, a deep learning framework developed by
* Megvii.
* Copyright (c) 2014-2021 Megvii Inc. All rights reserved.
*
* \copyright Copyright (c) 2020-2021 Megvii Inc. All rights reserved.
* Unless required by applicable law or agreed to in writing,
* software distributed under the License is distributed on an
* "AS IS" BASIS, WITHOUT ARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
*/
#include <lite_build_config.h>
......
lite/src/lite_build_config.h.in
浏览文件 @ f9719828
/**
* \file lite/src/lite_build_config.h.in
* MegEngine is Licensed under the Apache License, Version 2.0 (the "License")
*
* This file is part of MegEngine, a deep learning framework developed by
* Megvii.
* Copyright (c) 2014-2021 Megvii Inc. All rights reserved.
*
* \copyright Copyright (c) 2020-2021 Megvii Inc. All rights reserved.
* Unless required by applicable law or agreed to in writing,
* software distributed under the License is distributed on an
* "AS IS" BASIS, WITHOUT ARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
*/
#ifndef _HEADER_LITE_BUILD_CONFIG
#define _HEADER_LITE_BUILD_CONFIG
......
lite/src/mge/algo_cache/file_cache.cpp
浏览文件 @ f9719828
/**
* \file lite/src/mge/algo_cache/file_cache.cpp
* MegEngine is Licensed under the Apache License, Version 2.0 (the "License")
*
* This file is part of MegEngine, a deep learning framework developed by
* Megvii.
* Copyright (c) 2014-2021 Megvii Inc. All rights reserved.
*
* \copyright Copyright (c) 2020-2020 Megvii Inc. All rights reserved.
* Unless required by applicable law or agreed to in writing,
* software distributed under the License is distributed on an
* "AS IS" BASIS, WITHOUT ARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
*/
#include "lite_build_config.h"
......
lite/src/mge/algo_cache/file_cache.h
浏览文件 @ f9719828
/**
* \file lite/src/mge/algo_cache/file_cache.h
* MegEngine is Licensed under the Apache License, Version 2.0 (the "License")
*
* This file is part of MegEngine, a deep learning framework developed by
* Megvii.
* Copyright (c) 2014-2021 Megvii Inc. All rights reserved.
*
* \copyright Copyright (c) 2020-2020 Megvii Inc. All rights reserved.
* Unless required by applicable law or agreed to in writing,
* software distributed under the License is distributed on an
* "AS IS" BASIS, WITHOUT ARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
*/
#pragma once
......
lite/src/mge/algo_cache/redis_cache.cpp
浏览文件 @ f9719828
/**
* \file lite/src/mge/algo_cache/redis_cache.cpp
* MegEngine is Licensed under the Apache License, Version 2.0 (the "License")
*
* This file is part of MegEngine, a deep learning framework developed by
* Megvii.
* Copyright (c) 2014-2021 Megvii Inc. All rights reserved.
*
* \copyright Copyright (c) 2020-2020 Megvii Inc. All rights reserved.
* Unless required by applicable law or agreed to in writing,
* software distributed under the License is distributed on an
* "AS IS" BASIS, WITHOUT ARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
*/
#include "lite_build_config.h"
......
lite/src/mge/algo_cache/redis_cache.h
浏览文件 @ f9719828
/**
* \file lite/src/mge/algo_cache/redis_cache.h
* MegEngine is Licensed under the Apache License, Version 2.0 (the "License")
*
* This file is part of MegEngine, a deep learning framework developed by
* Megvii.
* Copyright (c) 2014-2021 Megvii Inc. All rights reserved.
*
* \copyright Copyright (c) 2020-2020 Megvii Inc. All rights reserved.
* Unless required by applicable law or agreed to in writing,
* software distributed under the License is distributed on an
* "AS IS" BASIS, WITHOUT ARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
*/
#pragma once
......
lite/src/mge/common.cpp
浏览文件 @ f9719828
/**
* \file src/mge/common.cpp
* MegEngine is Licensed under the Apache License, Version 2.0 (the "License")
*
* This file is part of MegEngine, a deep learning framework developed by
* Megvii.
* Copyright (c) 2014-2021 Megvii Inc. All rights reserved.
*
* \copyright Copyright (c) 2020-2021 Megvii Inc. All rights reserved.
* Unless required by applicable law or agreed to in writing,
* software distributed under the License is distributed on an
* "AS IS" BASIS, WITHOUT ARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
*/
#include "lite_build_config.h"
......
lite/src/mge/common.h
浏览文件 @ f9719828
/**
* \file src/mge/common.h
* MegEngine is Licensed under the Apache License, Version 2.0 (the "License")
*
* This file is part of MegEngine, a deep learning framework developed by
* Megvii.
* Copyright (c) 2014-2021 Megvii Inc. All rights reserved.
*
* \copyright Copyright (c) 2020-2021 Megvii Inc. All rights reserved.
* Unless required by applicable law or agreed to in writing,
* software distributed under the License is distributed on an
* "AS IS" BASIS, WITHOUT ARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
*/
#pragma once
......
lite/src/mge/function_dft.h
浏览文件 @ f9719828
/**
* \file src/mge/function_dft.h
* MegEngine is Licensed under the Apache License, Version 2.0 (the "License")
*
* This file is part of MegEngine, a deep learning framework developed by
* Megvii.
* Copyright (c) 2014-2021 Megvii Inc. All rights reserved.
*
* \copyright Copyright (c) 2020-2021 Megvii Inc. All rights reserved.
* Unless required by applicable law or agreed to in writing,
* software distributed under the License is distributed on an
* "AS IS" BASIS, WITHOUT ARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
*/
#pragma once
......
lite/src/mge/memory_allocator.h
浏览文件 @ f9719828
/**
* \file src/mge/memory_alloctor.h
* MegEngine is Licensed under the Apache License, Version 2.0 (the "License")
*
* This file is part of MegEngine, a deep learning framework developed by
* Megvii.
* Copyright (c) 2014-2021 Megvii Inc. All rights reserved.
*
* \copyright Copyright (c) 2020-2021 Megvii Inc. All rights reserved.
* Unless required by applicable law or agreed to in writing,
* software distributed under the License is distributed on an
* "AS IS" BASIS, WITHOUT ARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
*/
#pragma once
......
lite/src/mge/network_impl.cpp
浏览文件 @ f9719828
/**
* \file src/mge/network_impl.cpp
* MegEngine is Licensed under the Apache License, Version 2.0 (the "License")
*
* This file is part of MegEngine, a deep learning framework developed by
* Megvii.
* Copyright (c) 2014-2021 Megvii Inc. All rights reserved.
*
* \copyright Copyright (c) 2020-2021 Megvii Inc. All rights reserved.
* Unless required by applicable law or agreed to in writing,
* software distributed under the License is distributed on an
* "AS IS" BASIS, WITHOUT ARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
*/
#include "lite_build_config.h"
......
lite/src/mge/network_impl.h
浏览文件 @ f9719828
/**
* \file src/mge/network_impl.h
* MegEngine is Licensed under the Apache License, Version 2.0 (the "License")
*
* This file is part of MegEngine, a deep learning framework developed by
* Megvii.
* Copyright (c) 2014-2021 Megvii Inc. All rights reserved.
*
* \copyright Copyright (c) 2020-2021 Megvii Inc. All rights reserved.
* Unless required by applicable law or agreed to in writing,
* software distributed under the License is distributed on an
* "AS IS" BASIS, WITHOUT ARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
*/
#pragma once
......
lite/src/mge/tensor_impl.cpp
浏览文件 @ f9719828
/**
* \file inlude/mge/tensor.cpp
* MegEngine is Licensed under the Apache License, Version 2.0 (the "License")
*
* This file is part of MegEngine, a deep learning framework developed by
* Megvii.
* Copyright (c) 2014-2021 Megvii Inc. All rights reserved.
*
* \copyright Copyright (c) 2020-2021 Megvii Inc. All rights reserved.
* Unless required by applicable law or agreed to in writing,
* software distributed under the License is distributed on an
* "AS IS" BASIS, WITHOUT ARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
*/
#include "lite_build_config.h"
......
lite/src/mge/tensor_impl.h
浏览文件 @ f9719828
/**
* \file src/mge/tensor_impl.h
* MegEngine is Licensed under the Apache License, Version 2.0 (the "License")
*
* This file is part of MegEngine, a deep learning framework developed by
* Megvii.
* Copyright (c) 2014-2021 Megvii Inc. All rights reserved.
*
* \copyright Copyright (c) 2020-2021 Megvii Inc. All rights reserved.
* Unless required by applicable law or agreed to in writing,
* software distributed under the License is distributed on an
* "AS IS" BASIS, WITHOUT ARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
*/
#pragma once
......
lite/src/misc.cpp
浏览文件 @ f9719828
/**
* \file inlude/misc.cpp
* MegEngine is Licensed under the Apache License, Version 2.0 (the "License")
*
* This file is part of MegEngine, a deep learning framework developed by
* Megvii.
* Copyright (c) 2014-2021 Megvii Inc. All rights reserved.
*
* \copyright Copyright (c) 2020-2021 Megvii Inc. All rights reserved.
* Unless required by applicable law or agreed to in writing,
* software distributed under the License is distributed on an
* "AS IS" BASIS, WITHOUT ARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
*/
#include "./misc.h"
......
lite/src/misc.h
浏览文件 @ f9719828
/**
* \file include/misc.h
* MegEngine is Licensed under the Apache License, Version 2.0 (the "License")
*
* This file is part of MegEngine, a deep learning framework developed by
* Megvii.
* Copyright (c) 2014-2021 Megvii Inc. All rights reserved.
*
* \copyright Copyright (c) 2020-2021 Megvii Inc. All rights reserved.
* Unless required by applicable law or agreed to in writing,
* software distributed under the License is distributed on an
* "AS IS" BASIS, WITHOUT ARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
*/
#pragma once
......
lite/src/network.cpp
浏览文件 @ f9719828
/**
* \file src/network.cpp
* MegEngine is Licensed under the Apache License, Version 2.0 (the "License")
*
* This file is part of MegEngine, a deep learning framework developed by
* Megvii.
* Copyright (c) 2014-2021 Megvii Inc. All rights reserved.
*
* \copyright Copyright (c) 2020-2021 Megvii Inc. All rights reserved.
* Unless required by applicable law or agreed to in writing,
* software distributed under the License is distributed on an
* "AS IS" BASIS, WITHOUT ARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
*/
#include "lite/network.h"
......
lite/src/network_impl_base.h
浏览文件 @ f9719828
/**
* \file src/network_impl_base.h
* MegEngine is Licensed under the Apache License, Version 2.0 (the "License")
*
* This file is part of MegEngine, a deep learning framework developed by
* Megvii.
* Copyright (c) 2014-2021 Megvii Inc. All rights reserved.
*
* \copyright Copyright (c) 2020-2021 Megvii Inc. All rights reserved.
* Unless required by applicable law or agreed to in writing,
* software distributed under the License is distributed on an
* "AS IS" BASIS, WITHOUT ARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
*/
#pragma once
......
lite/src/parse_info/default_parse.h
浏览文件 @ f9719828
......@@ -2,10 +2,11 @@
* \file src/parse_info/default_parse.h
* MegEngine is Licensed under the Apache License, Version 2.0 (the "License")
*
* This file is part of MegEngine, a deep learning framework developed by
* Megvii.
* Copyright (c) 2014-2021 Megvii Inc. All rights reserved.
*
* \copyright Copyright (c) 2020-2021 Megvii Inc. All rights reserved.
* Unless required by applicable law or agreed to in writing,
* software distributed under the License is distributed on an
* "AS IS" BASIS, WITHOUT ARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
*/
#pragma once
......
lite/src/parse_info/parse_info_base.h
浏览文件 @ f9719828
......@@ -2,10 +2,11 @@
* \file src/parse_info/parse_info_base.h
* MegEngine is Licensed under the Apache License, Version 2.0 (the "License")
*
* This file is part of MegEngine, a deep learning framework developed by
* Megvii.
* Copyright (c) 2014-2021 Megvii Inc. All rights reserved.
*
* \copyright Copyright (c) 2020-2021 Megvii Inc. All rights reserved.
* Unless required by applicable law or agreed to in writing,
* software distributed under the License is distributed on an
* "AS IS" BASIS, WITHOUT ARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
*/
#pragma once
......
lite/src/parse_model/model_parser.cpp
浏览文件 @ f9719828
/**
* \file src/model_parser.cpp
* MegEngine is Licensed under the Apache License, Version 2.0 (the "License")
*
* This file is part of MegEngine, a deep learning framework developed by
* Megvii.
* Copyright (c) 2014-2021 Megvii Inc. All rights reserved.
*
* \copyright Copyright (c) 2020-2021 Megvii Inc. All rights reserved.
* Unless required by applicable law or agreed to in writing,
* software distributed under the License is distributed on an
* "AS IS" BASIS, WITHOUT ARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
*/
#include "model_parser.h"
......
lite/src/parse_model/model_parser.h
浏览文件 @ f9719828
/**
* \file src/model_parser.h
* MegEngine is Licensed under the Apache License, Version 2.0 (the "License")
*
* This file is part of MegEngine, a deep learning framework developed by
* Megvii.
* Copyright (c) 2014-2021 Megvii Inc. All rights reserved.
*
* \copyright Copyright (c) 2020-2021 Megvii Inc. All rights reserved.
* Unless required by applicable law or agreed to in writing,
* software distributed under the License is distributed on an
* "AS IS" BASIS, WITHOUT ARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
*/
#pragma once
......
lite/src/tensor.cpp
浏览文件 @ f9719828
/**
* \file src/tensor.cpp
* MegEngine is Licensed under the Apache License, Version 2.0 (the "License")
*
* This file is part of MegEngine, a deep learning framework developed by
* Megvii.
* Copyright (c) 2014-2021 Megvii Inc. All rights reserved.
*
* \copyright Copyright (c) 2020-2021 Megvii Inc. All rights reserved.
* Unless required by applicable law or agreed to in writing,
* software distributed under the License is distributed on an
* "AS IS" BASIS, WITHOUT ARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
*/
#include "lite/tensor.h"
......
lite/src/tensor_impl_base.h
浏览文件 @ f9719828
/**
* \file src/tensor_impl_base.h
* MegEngine is Licensed under the Apache License, Version 2.0 (the "License")
*
* This file is part of MegEngine, a deep learning framework developed by
* Megvii.
* Copyright (c) 2014-2021 Megvii Inc. All rights reserved.
*
* \copyright Copyright (c) 2020-2021 Megvii Inc. All rights reserved.
* Unless required by applicable law or agreed to in writing,
* software distributed under the License is distributed on an
* "AS IS" BASIS, WITHOUT ARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
*/
#pragma once
......
lite/src/type_info.h
浏览文件 @ f9719828
/**
* \file src/type_info.h
* MegEngine is Licensed under the Apache License, Version 2.0 (the "License")
*
* This file is part of MegEngine, a deep learning framework developed by
* Megvii.
* Copyright (c) 2014-2021 Megvii Inc. All rights reserved.
*
* \copyright Copyright (c) 2020-2021 Megvii Inc. All rights reserved.
* Unless required by applicable law or agreed to in writing,
* software distributed under the License is distributed on an
* "AS IS" BASIS, WITHOUT ARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
*/
#pragma once
......
lite/test/CMakeLists.txt
浏览文件 @ f9719828
......@@ -7,6 +7,10 @@ if (MGE_WITH_TEST)
if(LITE_BUILD_WITH_MGE)
# lite_test will depends megbrain interface
target_link_libraries(lite_test megbrain)
if (MGE_WITH_ROCM)
# FIXME: hip obj can not find cpp obj only through lite_static
target_link_libraries(lite_test megdnn)
endif ()
endif()
if(UNIX)
......
lite/test/main.cpp
浏览文件 @ f9719828
/**
* \file test/main.cpp
* MegEngine is Licensed under the Apache License, Version 2.0 (the "License")
*
* This file is part of MegEngine, a deep learning framework developed by
* Megvii.
* Copyright (c) 2014-2021 Megvii Inc. All rights reserved.
*
* \copyright Copyright (c) 2020-2021 Megvii Inc. All rights reserved.
* Unless required by applicable law or agreed to in writing,
* software distributed under the License is distributed on an
* "AS IS" BASIS, WITHOUT ARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
*/
#include <gtest/gtest.h>
......
lite/test/test_common.h
浏览文件 @ f9719828
/**
* \file test/test_common.h
* MegEngine is Licensed under the Apache License, Version 2.0 (the "License")
*
* This file is part of MegEngine, a deep learning framework developed by
* Megvii.
* Copyright (c) 2014-2021 Megvii Inc. All rights reserved.
*
* \copyright Copyright (c) 2020-2021 Megvii Inc. All rights reserved.
* Unless required by applicable law or agreed to in writing,
* software distributed under the License is distributed on an
* "AS IS" BASIS, WITHOUT ARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
*/
#pragma once
......
lite/test/test_misc.cpp
浏览文件 @ f9719828
/**
* \file test/test_misc.cpp
* MegEngine is Licensed under the Apache License, Version 2.0 (the "License")
*
* This file is part of MegEngine, a deep learning framework developed by
* Megvii.
* Copyright (c) 2014-2021 Megvii Inc. All rights reserved.
*
* \copyright Copyright (c) 2020-2021 Megvii Inc. All rights reserved.
* Unless required by applicable law or agreed to in writing,
* software distributed under the License is distributed on an
* "AS IS" BASIS, WITHOUT ARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
*/
#include "lite_build_config.h"
......
lite/test/test_network.cpp
浏览文件 @ f9719828
/**
* \file test/test_network.cpp
* MegEngine is Licensed under the Apache License, Version 2.0 (the "License")
*
* This file is part of MegEngine, a deep learning framework developed by
* Megvii.
* Copyright (c) 2014-2021 Megvii Inc. All rights reserved.
*
* \copyright Copyright (c) 2020-2021 Megvii Inc. All rights reserved.
* Unless required by applicable law or agreed to in writing,
* software distributed under the License is distributed on an
* "AS IS" BASIS, WITHOUT ARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
*/
#include "lite_build_config.h"
......
lite/test/test_network_c.cpp
浏览文件 @ f9719828
/**
* \file test/test_network_c.cpp
* MegEngine is Licensed under the Apache License, Version 2.0 (the "License")
*
* This file is part of MegEngine, a deep learning framework developed by
* Megvii.
* Copyright (c) 2014-2021 Megvii Inc. All rights reserved.
*
* \copyright Copyright (c) 2020-2021 Megvii Inc. All rights reserved.
* Unless required by applicable law or agreed to in writing,
* software distributed under the License is distributed on an
* "AS IS" BASIS, WITHOUT ARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
*/
#include "../src/misc.h"
......
lite/test/test_network_options.cpp
浏览文件 @ f9719828
/**
* \file test/test_network_options.cpp
* MegEngine is Licensed under the Apache License, Version 2.0 (the "License")
*
* This file is part of MegEngine, a deep learning framework developed by
* Megvii.
* Copyright (c) 2014-2021 Megvii Inc. All rights reserved.
*
* \copyright Copyright (c) 2020-2021 Megvii Inc. All rights reserved.
* Unless required by applicable law or agreed to in writing,
* software distributed under the License is distributed on an
* "AS IS" BASIS, WITHOUT ARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
*/
#include "lite_build_config.h"
......
lite/test/test_tensor.cpp
浏览文件 @ f9719828
/**
* \file test/test_tensor.cpp
* MegEngine is Licensed under the Apache License, Version 2.0 (the "License")
*
* This file is part of MegEngine, a deep learning framework developed by
* Megvii.
* Copyright (c) 2014-2021 Megvii Inc. All rights reserved.
*
* \copyright Copyright (c) 2020-2021 Megvii Inc. All rights reserved.
* Unless required by applicable law or agreed to in writing,
* software distributed under the License is distributed on an
* "AS IS" BASIS, WITHOUT ARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
*/
#include "lite_build_config.h"
......
lite/test/test_tensor_c.cpp
浏览文件 @ f9719828
/**
* \file test/test_tensor_c.cpp
* MegEngine is Licensed under the Apache License, Version 2.0 (the "License")
*
* This file is part of MegEngine, a deep learning framework developed by
* Megvii.
* Copyright (c) 2014-2021 Megvii Inc. All rights reserved.
*
* \copyright Copyright (c) 2020-2021 Megvii Inc. All rights reserved.
* Unless required by applicable law or agreed to in writing,
* software distributed under the License is distributed on an
* "AS IS" BASIS, WITHOUT ARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
*/
#include "lite_build_config.h"
......
lite/tools/dump_model_mgb.py
浏览文件 @ f9719828
#!/usr/bin/env mdl
# -*- coding: utf-8 -*-
# MegEngine is Licensed under the Apache License, Version 2.0 (the "License")
#
# Copyright (c) 2020-2021 Megvii Inc. All rights reserved.
# Copyright (c) 2014-2021 Megvii Inc. All rights reserved.
#
# Unless required by applicable law or agreed to in writing,
# software distributed under the License is distributed on an
......
lite/tools/pack_model/pack_model_and_info.py
浏览文件 @ f9719828
#!/usr/bin/python3
# -*- coding: utf-8 -*-
# MegEngine is Licensed under the Apache License, Version 2.0 (the "License")
#
# This file is part of MegEngine, a deep learning framework developed by
# Megvii.
# Copyright (c) 2014-2021 Megvii Inc. All rights reserved.
#
# copyright Copyright (c) 2020-2021 Megvii Inc. All rights reserved.
# Unless required by applicable law or agreed to in writing,
# software distributed under the License is distributed on an
# "AS IS" BASIS, WITHOUT ARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
import argparse
import struct
......
lite/tools/rc4_encrypt.cpp
浏览文件 @ f9719828
/** \file tools/rc4_encrypt.cpp
* MegEngine is Licensed under the Apache License, Version 2.0 (the "License")
*
* This file is part of MegEngine, a deep learning framework developed by
* Megvii.
* Copyright (c) 2014-2021 Megvii Inc. All rights reserved.
*
* \copyright Copyright (c) 2020-2021 Megvii Inc. All rights reserved.
* Unless required by applicable law or agreed to in writing,
* software distributed under the License is distributed on an
* "AS IS" BASIS, WITHOUT ARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
*/
#include <stdio.h>
......
Markdown is supported
0% 或 .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑！
想要评论请 注册 或