sglang.0.4.8.post1/nvshmem_src/scripts/create_imex.sh

27 lines
910 B
Bash
Executable File

#!/bin/bash
# Following the guidelines from B200/GB200 bringup to use mknod(2) to create IMEX channels
# This is admin level operation
find_major_number=$(cat /proc/devices | grep nvidia-caps-imex-channels | wc -l)
if [[ $find_major_number -eq 0 ]];
then
echo "No nvidia-caps-imex-channels device found. Exiting..."
exit 1
fi
# /dev/nvidia-caps-imex-channels dir should be present before create a char dev under it.
IMEX_DIRECTORY="/dev/nvidia-caps-imex-channels"
if [ ! -d "$IMEX_DIRECTORY" ]; then
mkdir -p "$IMEX_DIRECTORY" || { echo "Failed to create directory $IMEX_DIRECTORY"; exit 1; }
fi
major_number=$(cat /proc/devices | grep nvidia-caps-imex-channels | cut -d' ' -f1)
create_cmd="sudo mknod /dev/nvidia-caps-imex-channels/channel0 c $major_number 0"
eval $create_cmd
exit_status=$?
if [[ $exit_status -ne 0 ]];
then
echo "Unable to create IMEX channel0. StatusCode = $exit_status"
exit 1
fi