Skip to content

qili93/CANN-ACL-OpTest

Repository files navigation

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)

About

sample and log files for ascend 910 ops

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages