How to run more PEs on vck5000?

[thhongee]

I'm currently using the VCK5000 to run Vitis AI 3.0, and everything works fine with the model zoo. However, I'm curious about the fact that the VCK5000 has more than 8 PEs. If I want to use more than 8 PEs, where should I start to modify? Right now, I'm looking at the file /opt/xilinx/overlaybins/DPUCVDX8H/8PE/dpu_DPUCVDX8H_8PE_350M_xilinx_vck5000_gen4x8_qdma_base_2.xclbin. Should I generate the xclbin myself?

Here is the Vitis AI tutorial.

Docker Container Environment Variable Setup¶
From inside the docker container, execute one of the following commands to set the required environment variables for the DPU. Note that the chosen xclbin file must be in the /opt/xilinx/overlaybins directory prior to execution. There are four xclbin files to choose from. Select the xclbin that matches your chosen DPU configuration.

For the 4PE 350MHz configuration with pooling, elementwise and depthwise convolution support:

[Docker] $ source /workspace/board_setup/vck5000/setup.sh DPUCVDX8H_4pe_miscdwc
For the 6PE 350MHz configuration with depthwise convolution support:

[Docker] $ source /workspace/board_setup/vck5000/setup.sh DPUCVDX8H_6pe_dwc
For the 6PE 350MHz configuration with pooling and elementwise support:

[Docker] $ source /workspace/board_setup/vck5000/setup.sh DPUCVDX8H_6PE_misc
For the 8PE 350MHz base configuration:

[Docker] $ source /workspace/board_setup/vck5000/setup.sh DPUCVDX8H_8pe_normal