Huawei_Technology
/
MindSpore-AKG
Not watched
1
0
Fork 0
Code
Releases 1 Wiki evaluate Activity
Issues 0 Pull Requests 0 Datasets Model Cloudbrain HPC
You can not select more than 25 topics Topics must start with a chinese character,a letter or number, can include dashes ('-') and can be up to 35 characters long.
80 Commits
2 Branches
7.6 MB
526 Download
 
 
 
 
 
 
Tree: 395aafbbac
Branches Tags
${ item.name }
Create branch ${ searchTerm }
from '395aafbbac'
${ noResults }
MindSpore-AKG/third_party/incubator-tvm
mindspore-ci-bot 58d0f85c2f !49 update the profiling module on Ascend 5 years ago
..
.gitee init 5 years ago
3rdparty init 5 years ago
apps init 5 years ago
cmake init 5 years ago
conda init 5 years ago
docker init 5 years ago
docs init 5 years ago
golang init 5 years ago
include/tvm update binary lib mark 5 years ago
jvm init 5 years ago
licenses init 5 years ago
nnvm init 5 years ago
python support composite topi by irbuilder 5 years ago
rust init 5 years ago
src !49 update the profiling module on Ascend 5 years ago
tests init 5 years ago
topi init 5 years ago
tutorials init 5 years ago
vta init 5 years ago
web init 5 years ago
.clang-format init 5 years ago
CMakeLists.txt init 5 years ago
CONTRIBUTORS.md init 5 years ago
DISCLAIMER init 5 years ago
Jenkinsfile init 5 years ago
KEYS init 5 years ago
LICENSE init 5 years ago
NEWS.md init 5 years ago
NOTICE init 5 years ago
README.md init 5 years ago
version.py init 5 years ago

README.md

This is a revised apache/incubator-tvm 0.6 for Auto Kernel Generator.
What we are using:

  • TVM IR
  • Arith Simplify
  • Several passes
  • DSL api
  • topi

Compared with original apache/incubator-tvm 0.6, we have modified some codes:

  1. Changed namespace from tvm to air.
  2. For these modules(listed above) that we used, we have added or deleted some codes targeted to Auto Kernel Generator.
    For detailed changes, please refer to the source codes, where we have listed the modification at the head of file.
Delete task

Are you sure you want to delete this task? Once this task is deleted, it cannot be recovered.

cancel
confirm

AKG(Auto Kernel Generator)对深度神经网络中的算子进行优化，并提供特定模式下的算子自动融合功能。AKG与MindSpore的图算融合功能协同工作，可提升在不同硬件后端上运行网络的性能。

Python C++ Text Cuda CMake other