#!/bin/sh

exec_if_exists() {
    if [ -x "$1" ]; then
        exec "$@"
    fi
}

die() {
    echo "$@" >&2
    exit 1
}

case "$(uname -m)" in
    x86_64) arch=x64 ;;
    aarch64) arch=sbsa-armv8 ;;
    ppc64le) arch=ppc64le ;;
    *)
        die "Error: Nsight Systems is not supported on $(uname -m)"
        ;;
esac

DIR=$(dirname "$(readlink -f -- "$0")")
CUDA_INSTALL_DIR=$(cd "$DIR/.." && pwd)
exec_if_exists /usr/lib64/nsight-systems-2025.1.3/target-linux-"$arch"/nsys "$@"
exec_if_exists /opt/nvidia/nsight-systems/2025.1.3/target-linux-"$arch"/nsys "$@"
exec_if_exists /opt/nvidia/nsight-systems-cli/2025.1.3/target-linux-"$arch"/nsys "$@"

die "Error: Nsight Systems 2025.1.3 hasn't been installed with CUDA Toolkit 12.9"
