36 lines
1.2 KiB
Bash
Executable File
36 lines
1.2 KiB
Bash
Executable File
#!/bin/bash
|
|
|
|
# Check if cuBB_SDK is defined, if not, use default path
|
|
cuBB_Path="${cuBB_SDK:-/opt/nvidia/cuBB}"
|
|
|
|
cd "$cuBB_Path" || exit 1
|
|
# Add gdrcopy to LD_LIBRARY_PATH
|
|
export LD_LIBRARY_PATH=/usr/local/nvidia/lib:/usr/local/nvidia/lib64:/opt/mellanox/dpdk/lib/x86_64-linux-gnu:/opt/mellanox/doca/lib/x86_64-linux-gnu:/opt/nvidia/cuBB/cuPHY-CP/external/gdrcopy/build/x86_64/
|
|
export LD_LIBRARY_PATH=$LD_LIBRART_PATH:"$cuBB_Path/gpu-dpdk/build/install/lib/x86_64-linux-gnu:$cuBB_Path/build/cuPHY-CP/cuphydriver/src"
|
|
|
|
# Restart MPS
|
|
# Export variables
|
|
export CUDA_DEVICE_MAX_CONNECTIONS=8
|
|
export CUDA_MPS_PIPE_DIRECTORY=/var
|
|
export CUDA_MPS_LOG_DIRECTORY=/var
|
|
|
|
# Stop existing MPS
|
|
sudo -E echo quit | sudo -E nvidia-cuda-mps-control
|
|
|
|
# Start MPS
|
|
sudo -E nvidia-cuda-mps-control -d
|
|
sudo -E echo start_server -uid 0 | sudo -E nvidia-cuda-mps-control
|
|
|
|
# Start cuphycontroller_scf
|
|
# Check if an argument is provided
|
|
if [ $# -eq 0 ]; then
|
|
# No argument provided, use default value
|
|
argument="P5G_FXN"
|
|
else
|
|
# Argument provided, use it
|
|
argument="$1"
|
|
fi
|
|
|
|
sudo -E "$cuBB_Path"/build/cuPHY-CP/cuphycontroller/examples/cuphycontroller_scf "$argument"
|
|
sudo -E ./build/cuPHY-CP/gt_common_libs/nvIPC/tests/pcap/pcap_collect
|