fixup! fixup! fixup! fixup! fixup! fixup! fixup! initial commit

Signed-off-by: Sayan Chowdhury <schowdhury@microsoft.com>

kola/tests/misc/nvidia.go

@@ -6,6 +6,7 @@ import (
 	"time"
 
 	"github.com/coreos/pkg/capnslog"
+	"github.com/flatcar/mantle/kola"
 	"github.com/flatcar/mantle/kola/cluster"
 	"github.com/flatcar/mantle/kola/register"
 	"github.com/flatcar/mantle/util"
@@ -31,13 +32,14 @@ func init() {
 }
 
 func verifyNvidiaInstallation(c cluster.TestCluster) {
+	if kola.AzureOptions.Size != "Standard_NC6s_v3" {
+		c.Skip("skipping due to wrong instance size")
+	}
 	m := c.Machines()[0]
-	nvidiaStatusRetry := func() error {
-		out, err := c.SSH(m, "systemctl is-active nvidia")
-		plog.Infof("systemctl is-active nvidia: %v, %v", out, err)
 
+	nvidiaStatusRetry := func() error {
+		out, err := c.SSH(m, "systemctl is-active nvidia.service")
 		if !bytes.Contains(out, []byte("inactive")) {
-			plog.Infof("checks for status contains")
 			return fmt.Errorf("nvidia.service: %q: %v", out, err)
 		}
 		return nil
@@ -46,10 +48,5 @@ func verifyNvidiaInstallation(c cluster.TestCluster) {
 	if err := util.Retry(40, 15*time.Second, nvidiaStatusRetry); err != nil {
 		c.Fatal(err)
 	}
-
-	plog.Infof("Checks for nvidia-smi")
-	out, err := c.SSH(m, "/opt/bin/nvidia-smi")
-	if !bytes.Contains(out, []byte("525.105.17")) {
-		c.Fatal("nvidia-smi: %q, %v", out, err)
-	}
+	c.AssertCmdOutputContains(m, "/opt/bin/nvidia-smi", "Tesla")
 }