From 5e1759d3028d821e9a36ca0e135cc188abd38cf7 Mon Sep 17 00:00:00 2001
From: Laurent Clouet <laurent@nopnop.net>
Date: Thu, 3 May 2018 18:21:11 +0200
Subject: [PATCH] Bugfix

---
 src/com/sheepit/client/hardware/gpu/GPU.java | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/src/com/sheepit/client/hardware/gpu/GPU.java b/src/com/sheepit/client/hardware/gpu/GPU.java
index e124ba4..d993d8e 100644
--- a/src/com/sheepit/client/hardware/gpu/GPU.java
+++ b/src/com/sheepit/client/hardware/gpu/GPU.java
@@ -22,6 +22,7 @@ package com.sheepit.client.hardware.gpu;
 import java.util.HashMap;
 import java.util.LinkedList;
 import java.util.List;
+import java.util.Map;
 
 import com.sheepit.client.os.OS;
 import com.sun.jna.Native;
@@ -127,7 +128,7 @@ public class GPU {
 		// generate proper cuda id
 		// in theory a set to environment "CUDA_DEVICE_ORDER=PCI_BUS_ID" should be enough but it didn't work
 		int i = 0;
-		for (HashMap.Entry<Integer, GPUDevice> entry : devicesWithPciId.entrySet()){
+		for (Map.Entry<Integer, GPUDevice> entry : devicesWithPciId.entrySet()){
 			GPUDevice aDevice = entry.getValue();
 			aDevice.setCudaName("CUDA_" + Integer.toString(i));
 			devices.add(aDevice);