p01753428
/
InfiniTensor
forked from jiuyuan/InfiniTensor
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity
106 Commits 72 Branches 1 Tag 141 MiB
C++ 76.3%
Cuda 18%
Python 4.3%
CMake 1.1%
Shell 0.1%
Go to file
Haojie Wang dd5d091dbc
wip: onnx 导出 (#65) 
| Notice | Work in progress
|-|-

> based on #63 

## Progress

1. [x] 对节点拓扑排序
2. [x] 遍历节点，命名并导出其输出张量（`make_tensor_value_info`）
3. [x] 识别图的输入张量，命名并导出（`make_tensor_value_info`）
4. [x] 根据节点类型，识别权重及属性，导出节点（`make_node`）
5. [x] `make_graph` -> `check_graph` -> `make_model` -> `check_model`
 		2023-03-15 15:22:09 +08:00
.github/workflows fix: revert ci yml 2023-02-15 07:45:21 +08:00
3rd-party test: enhance ci (#62) 2023-02-12 00:01:36 +08:00
include feat: 导出 ReduceMean 到 onnx 2023-03-15 15:09:12 +08:00
proto Tensor serialization (#25) 2022-09-13 11:27:41 +08:00
pyinfinitensor fix: 移除中文注释 2023-03-15 15:16:16 +08:00
python Add search engine (#64) 2023-02-12 18:27:52 +08:00
src fix: 移除中文注释 2023-03-15 15:16:16 +08:00
test feat: GraphObj 增加一个拓扑排序方法及其测试 2023-03-15 15:09:12 +08:00
.clang-format Add: graph, tensor, and operator 2022-07-31 21:44:03 +08:00
.cmake-format.json Add: graph, tensor, and operator 2022-07-31 21:44:03 +08:00
.gitignore feat: 创建 pyinfinitensor 前端 2023-02-13 09:19:05 +08:00
.gitmodules add code for backtrace (#21) 2022-09-01 20:30:12 +08:00
CMakeLists.txt feat: 创建 pyinfinitensor 前端 2023-02-13 09:19:05 +08:00
LICENSE Initial commit 2022-07-27 22:40:23 +08:00
Makefile feat: 创建 pyinfinitensor 前端 2023-02-13 09:19:05 +08:00
README.md test: enhance ci (#62) 2023-02-12 00:01:36 +08:00

README.md

InfiniTensor

Compilation on Lotus

# Enter the root of InfiniTensor
source test/script/env_lotus.sh
mkdir build && cd build
cmake -DUSE_CUDA=ON .. && make -j 12

CMake Options

There are several configurable CMake options, see the CMakeLists.txt file.

  • If USE_BACKTRACE is ON, libdw-dev have to be installed. See the README of backward-cpp for details.
  • If USE_PROTOBUF is ON, protobuf have to be installed. See the README of protobuf for details.

Contributor Guide

InfiniTensor development is based on the pull request on Github. Before requesting for merging, a PR should satisfy the following requirements

  1. Pass all tests.
    1. Now CI on Github will test everything that can be tested in the ci environment, including code format. So, script test/script/clang_format_inplace.sh is for formatting all code.
    2. Contributors should run ctest manually and copy its output to the PR. Use fenced code blocks (triple backquotes, i.e., ```) to avoid referencing in Github. Otherwise, # in the output is interpreted as a Github reference. Do not directly paste the ctest output in commit messages either for the same reason.
  2. Receive at least one approval from reviewers.
  3. PR title should be concise since it is going to be the commit message in the main branch after merging and squashing.

Dependencies