This is a converter to transfer Caffe-CPU prototxt to HUAWEI Atlas prototxt. It generates the Atlas AICORE kernel codes, compiler the code, and append the kernel info to prototxt.
It relies on Caffe-CPU to figure out what AICORE kernel codes are needed. The "name" of these AICORE kernel codes is automatically collected into a python file.
The python file also contains the necessary code to compile the code. It compiles the code first and then collects the kernels' information, such as their names and num_cores. It then appends the information to the original caffe prototxt, which is used by caffe-atlas.
HUAWEI only provides binary files for its compiler, which is for Ubuntu 18.04. This converter also runs on Ubuntu 18.04 only.
Caffe can be compiled follow link, but you must compiler the one in this repo!
cd REPO_HOME
cd caffe
./data/mnist/get_mnist.sh
./examples/mnist/create_mnist.sh
./examples/mnist/train_lenet.sh
cd ..
It generates a python file gen_compile_net.py. Some path needs to be specified, but you can just run
sh run.sh
A directory named 'kernel_meta' is generated. It contains the *.o kernel codes and the aicore_YOU_PROTOTXT_NAME.prototxt file.
You can then use caffe-atlas and these files for training.