Go to file

wendy12022 86ec4036ce ADD: add mkl runtime for intel cpu , and add mkl kernel for matmul/conv/convtransposed. (#61 ) * move memory format transformation to TensorObj clang format add MemoryFormat for tensorObj. use post_ops for fused conv/deconv Distinguish mkl op_timer from cuda op timer. add act optype to conv and deconv add operator timer add mkl kernel for convTransposed minor fix for group conv do not use cblas_sgemm_batch CpuRuntimeObj->NativeCpuRuntimeObj add matmul op for mkl * fix: fix bugs when rebasing from master fix: fix bugs when rebasing from master * fix: update api after rebasing * fix: fix format; fix onnx import * fix: fix clang-format * [fix] fix conv_transpose test * [fix] use stronger test case for transposed conv * [fix] remove tensor memory format; fix mkl transpose conv * [fix] add FIXME tag for op_timer python api --------- Co-authored-by: whjthu <haojie0429@gmail.com>		2023-03-27 21:28:49 +08:00
.github/workflows	fix: TensorObj::dataMalloc	2023-03-15 17:23:32 +08:00
3rd-party	test: enhance ci (#62 )	2023-02-12 00:01:36 +08:00
include	ADD: add mkl runtime for intel cpu , and add mkl kernel for matmul/conv/convtransposed. (#61 )	2023-03-27 21:28:49 +08:00
proto	Tensor serialization (#25 )	2022-09-13 11:27:41 +08:00
pyinfinitensor	fix: 删除单独的拷贝函数	2023-03-22 10:13:06 +08:00
python	ADD: add mkl runtime for intel cpu , and add mkl kernel for matmul/conv/convtransposed. (#61 )	2023-03-27 21:28:49 +08:00
src	ADD: add mkl runtime for intel cpu , and add mkl kernel for matmul/conv/convtransposed. (#61 )	2023-03-27 21:28:49 +08:00
test	ADD: add mkl runtime for intel cpu , and add mkl kernel for matmul/conv/convtransposed. (#61 )	2023-03-27 21:28:49 +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	ADD: add mkl runtime for intel cpu , and add mkl kernel for matmul/conv/convtransposed. (#61 )	2023-03-27 21:28:49 +08:00
LICENSE	Initial commit	2022-07-27 22:40:23 +08:00
Makefile	docs: update README.md	2023-03-15 17:23:32 +08:00
README.md	docs: update README.md	2023-03-15 17:23:32 +08:00

README.md

InfiniTensor

Compilation on Lotus

# Enter the root of InfiniTensor
source test/script/env_lotus.sh
make CUDA=ON

Make Commands

make/make build: Builds the project;
make install-python: Builds the project then install the python frontend;
make test-cpp: Builds the project then run cpp unit tests;
make test-onnx: Run python unit tests;

Sets env: CUDA=ON to enable cuda.

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.
If USE_CUDA is ON, cuda have to be installed.

Contributor Guide

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

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.
Receive at least one approval from reviewers.
PR title should be concise since it is going to be the commit message in the main branch after merging and squashing.