From 9478b216e5678b1a2e738abe771a335588f0f072 Mon Sep 17 00:00:00 2001 From: Russell Currey Date: Thu, 11 Aug 2016 16:38:43 +1000 Subject: external: NPU hardware procedure script Performing NPU hardware procedures requires some config space magic. Put all that magic into a script, so you can just specify the target device and the procedure number. Signed-off-by: Russell Currey Signed-off-by: Stewart Smith --- external/npu/run_procedure.sh | 119 ++++++++++++++++++++++++++++++++++++++++++ 1 file changed, 119 insertions(+) create mode 100755 external/npu/run_procedure.sh (limited to 'external/npu') diff --git a/external/npu/run_procedure.sh b/external/npu/run_procedure.sh new file mode 100755 index 0000000..f61ec8d --- /dev/null +++ b/external/npu/run_procedure.sh @@ -0,0 +1,119 @@ +#!/bin/bash + +# Copyright 2016 IBM Corp. +# +# Licensed under the Apache License, Version 2.0 (the "License"); +# you may not use this file except in compliance with the License. +# You may obtain a copy of the License at +# +# http://www.apache.org/licenses/LICENSE-2.0 +# +# Unless required by applicable law or agreed to in writing, software +# distributed under the License is distributed on an "AS IS" BASIS, +# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or +# implied. +# See the License for the specific language governing permissions and +# limitations under the License. +# + +function usage() { + echo -e "$0: run a NPU hardware procedure (requires root)\n" + echo -e "Usage: $0 \n" + echo -e "Example: $0 0008:00:01.1 1" + echo -e "Procedures are documented in skiboot/doc/nvlink.rst" +} + +function check_root() { + if [ "$(id -u)" != "0" ]; then + echo -e "Error: $0 must be run as root\n" 1>&2 + exit 1 + fi +} + +function check_args() { + if [ "$#" -eq 0 ]; then + usage + exit 1 + fi + + if [ "$#" -gt 2 ]; then + echo -e "Error: too many arguments\n" 1>&2 + usage + exit 1 + fi + + if [[ "$1" == "-h" || "$1" == "--help" ]]; then + usage + exit 0 + fi + + if ! [ "$2" -eq "$2" ] 2>/dev/null; then + echo -e "Procedure must be a decimal number\n" 1>&2 + usage + exit 1 + fi + + if [[ "$2" -lt "0" || "$2" -gt "12" ]]; then + echo -e "Invalid procedure number\n" 1>&2 + usage + exit 2 + fi + + pci_check=$(lspci -s $1) + if [[ $? -ne 0 || $pci_check == "" ]]; then + echo -e "Invalid PCI device\n" 1>&2 + usage + exit 2 + fi +} + +function run_procedure() { + # Convert procedure number into hex + procedure=$(echo "obase=16; $2" | bc) + + # Check the status register to make sure we can run a procedure + status=$(setpci -s $1 0x84.L) + if [[ $status == 8000000* ]]; then + echo "Procedure in progress, try again." 1>&2 + echo "If that doesn't work, use procedure 0 to abort." 1>&2 + exit 3 + fi + + # Start the procedure + setpci -s $1 0x88.L=0x0000000$procedure >/dev/null + if [ $? -ne 0 ]; then + echo "Control register write failed!" 1>&2 + exit 3 + fi + + iterations=1 + while [[ $(setpci -s $1 0x84.L) == 8000000* ]]; do + ((iterations++)) + done + + # Check again, procedure should be finished + status=$(setpci -s $1 0x84.L) + + echo "Done in $iterations iteration(s)!" + + if [[ $status == 40000000 ]]; then + echo "Procedure completed successfully." + exit 0 + elif [[ $status == 40000001 ]]; then + echo "Transient failure, try again." 1>&2 + exit 4 + elif [[ $status == 40000002 ]]; then + echo "Permanent failure, reboot required?" 1>&2 + exit 5 + elif [[ $status == 40000003 ]]; then + echo "Procedure aborted." 1>&2 + exit 6 + elif [[ $status == 40000004 ]]; then + echo "Unsupported procedure." 1>&2 + exit 7 + fi +} + +check_args "$@" +check_root +run_procedure "$1" "$2" -- cgit v1.1