p20518467
/
InfiniTensor
forked from jiuyuan/InfiniTensor
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity
29 Commits 70 Branches 0 Tags 141 MiB
C++ 76.3%
Cuda 18%
Python 4.3%
CMake 1.1%
Shell 0.1%
Go to file
wendy12022 13b7a2604b
ADD add/mul/sub/div/pow operators and CPU/CUDA kernels (#26) 
Fix some

remove useless code.

add div/pow kernel

Add add/mul/sub operators.

fix cpu kernel.

add element wise kenerl for cuda.

ADD element wise operator.
 		2022-09-09 13:43:59 +08:00
.github/workflows Add: clang format check github action 2022-08-09 17:58:12 +08:00
3rd-party add code for backtrace (#21) 2022-09-01 20:30:12 +08:00
include ADD add/mul/sub/div/pow operators and CPU/CUDA kernels (#26) 2022-09-09 13:43:59 +08:00
src ADD add/mul/sub/div/pow operators and CPU/CUDA kernels (#26) 2022-09-09 13:43:59 +08:00
test ADD add/mul/sub/div/pow operators and CPU/CUDA kernels (#26) 2022-09-09 13:43:59 +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 Add: graph, tensor, and operator 2022-07-31 21:44:03 +08:00
.gitmodules add code for backtrace (#21) 2022-09-01 20:30:12 +08:00
CMakeLists.txt Operators g2bmm&gbmm transplantation (#24) 2022-09-08 21:31:35 +08:00
LICENSE Initial commit 2022-07-27 22:40:23 +08:00
README.md Add CUDA runtime (#6) 2022-08-22 15:01:03 +08:00

README.md

InfiniTensor

Compilation on Lotus

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