Skip to content

Latest commit

 

History

History
48 lines (34 loc) · 1.6 KB

README.md

File metadata and controls

48 lines (34 loc) · 1.6 KB

Ascend910 OP Test Samples

How to Run

Pre-requisites: a server with Ascend910, refer to https://www.hiascend.com/software/cann/community

  1. Get runtime docker image

Note: the installed CANN version in this image is '5.0.2.alpha005'.

# x86_64 host
docker pull registry.baidubce.com/device/paddle-npu:cann512-x86_64-gcc75

# aarch64 host
docker pull registry.baidubce.com/device/paddle-npu:cann512-aarch64-gcc75
  1. Start docker container
docker run -it --name cann512 -v `pwd`:/workspace --workdir=/workspace \
      --pids-limit 409600 -v /home/datasets:/datasets \
      --privileged --network=host --shm-size=128G \
      -v /usr/local/Ascend/driver:/usr/local/Ascend/driver \
      -v /usr/local/bin/npu-smi:/usr/local/bin/npu-smi \
      -v /usr/local/dcmi:/usr/local/dcmi \
      registry.baidubce.com/device/paddle-npu:cann512-x86_64-gcc82 /bin/bash
  1. Compile and run by run_demo.sh, for example:
# Run FillV2 OP
sh run_demo.sh FillV2

# Run ResizeNearestNeighborV2 OP
sh run_demo.sh ResizeNearestNeighborV2
  1. Tracking issues here (i.e. issue links to Ascend community)