#!/bin/bash

# Copyright 2020 The IREE Authors
#
# Licensed under the Apache License v2.0 with LLVM Exceptions.
# See https://llvm.org/LICENSE.txt for license information.
# SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception

# Checks that CUDA is working correctly and logs some useful information

set -xeuo pipefail

if ! dpkg -l | grep nvidia; then
  echo "Cannot find installed nvidia package"
  exit 1
fi
if ! nvidia-smi; then
  ret=$?
  echo "nvidia-smi failed"
  exit "${ret}"
fi