Skip to content

Latest commit

 

History

History
55 lines (40 loc) · 1.76 KB

README.md

File metadata and controls

55 lines (40 loc) · 1.76 KB
Raw

PaddlePaddle Custom Device Implementation for Enflame GCU

English | 简体中文

Please refer to the following steps to compile, install and verify the hardware backend (GCU).

Prepare environment and source code

# 1) Pull PaddlePaddle development docker image，and install Enflame GCU development kit.

# 2) Clone the source code.
git clone https://github.com/PaddlePaddle/PaddleCustomDevice
cd PaddleCustomDevice

PaddleCustomDevice Installation and Verification

Source code compilation and installation

# 1) Enter the hardware backend (Enflame GCU) directory.
cd backends/gcu

# 2) Before compiling, you need to ensure that the PaddlePaddle installation package is installed in the environment.
#    Just install the PaddlePaddle CPU version directly.
python -m pip install paddlepaddle==0.0.0 -f https://www.paddlepaddle.org.cn/whl/linux/cpu-mkl/develop.html

# 3) Start compiling, and submodules will be downloaded on demand during compilation.
mkdir -p build && cd build
cmake .. -DWITH_TESTING=ON -DCMAKE_EXPORT_COMPILE_COMMANDS=ON
make -j $(nproc)

# 4) The compiled product is in the build/dist path and installed using pip.
python -m pip install --force-reinstall -U build/dist/paddle_custom_gcu*.whl

Functional Verification

# 1) List available hardware backends.
python -c "import paddle; print(paddle.device.get_all_custom_device_type())"
# Expect the following output.
['gcu']

# 2) Check currently installed version.
python -c "import paddle_custom_device; paddle_custom_device.gcu.version()"
# Expect to get output like this.
version: 0.0.0.9e03b0a
commit: 9e03b0a42a530d07fb60e141ee618fc02595bd96
tops-sdk: 2.5.20231128

# 3) Unit test, compiled with -DWITH_TESTING=ON and executed in the build directory.
ctest