Initial commit

Version 3.3 of the client

src/com/sheepit/client/hardware/gpu/GPU.java

@@ -0,0 +1,135 @@
+/*
+ * Copyright (C) 2013-2014 Laurent CLOUET
+ * Author Laurent CLOUET <laurent.clouet@nopnop.net>
+ *
+ * This program is free software; you can redistribute it and/or 
+ * modify it under the terms of the GNU General Public License
+ * as published by the Free Software Foundation; version 2
+ * of the License.
+ *
+ * This program is distributed in the hope that it will be useful,
+ * but WITHOUT ANY WARRANTY; without even the implied warranty of
+ * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
+ * GNU General Public License for more details.
+ *
+ * You should have received a copy of the GNU General Public License
+ * along with this program; if not, write to the Free Software
+ * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA  02110-1301, USA.
+ */
+
+package com.sheepit.client.hardware.gpu;
+
+import java.util.LinkedList;
+import java.util.List;
+
+import com.sheepit.client.os.OS;
+import com.sun.jna.Native;
+import com.sun.jna.NativeLong;
+
+public class GPU {
+	public static List<GPUDevice> devices = null;
+	
+	public static boolean generate() {
+		OS os = OS.getOS();
+		String path = os.getCUDALib();
+		if (path == null) {
+			System.out.println("GPU.listDevices failed to get CUDA lib");
+			return false;
+		}
+		CUDA cudalib = null;
+		try {
+			cudalib = (CUDA) Native.loadLibrary(path, CUDA.class);
+		}
+		catch (java.lang.UnsatisfiedLinkError e) {
+			System.out.println("GPU.listDevices failed to load CUDA lib");
+			return false;
+		}
+		catch (java.lang.ExceptionInInitializerError e) {
+			System.out.println("GPU.listDevices ExceptionInInitializerError " + e);
+			return false;
+		}
+		catch (Exception e) {
+			System.out.println("GPU.listDevices generic exception " + e);
+			return false;
+		}
+		
+		int result = CUresult.CUDA_ERROR_UNKNOWN;
+		
+		result = cudalib.cuInit(0);
+		if (result != CUresult.CUDA_SUCCESS) {
+			return false;
+		}
+		
+		if (result == CUresult.CUDA_ERROR_NO_DEVICE) {
+			System.out.println("NO DEVICE");
+			return false;
+		}
+		
+		int[] count = new int[1];
+		result = cudalib.cuDeviceGetCount(count);
+		
+		if (result != CUresult.CUDA_SUCCESS) {
+			System.out.println("GPU.listDevices cuDeviceGetCount failed (ret: " + CUresult.stringFor(result) + ")");
+			return false;
+		}
+		
+		devices = new LinkedList<GPUDevice>();
+		
+		for (int num = 0; num < count[0]; num++) {
+			byte name[] = new byte[256];
+			
+			result = cudalib.cuDeviceGetName(name, 256, num);
+			if (result != CUresult.CUDA_SUCCESS) {
+				System.out.println("GPU.listDevices cuDeviceGetName failed (ret: " + CUresult.stringFor(result) + ")");
+				continue;
+			}
+			
+			NativeLong[] ram = new NativeLong[1];
+			result = cudalib.cuDeviceTotalMem(ram, num);
+			
+			if (result != CUresult.CUDA_SUCCESS) {
+				System.out.println("GPU.listDevices cuDeviceTotalMem failed (ret: " + CUresult.stringFor(result) + ")");
+				return false;
+			}
+			
+			devices.add(new GPUDevice(new String(name).trim(), ram[0].longValue(), "CUDA_" + Integer.toString(num)));
+		}
+		return true;
+	}
+	
+	public static List<String> listDevices() {
+		if (devices == null) {
+			generate();
+		}
+		if (devices == null) {
+			return null;
+		}
+		
+		List<String> devs = new LinkedList<String>();
+		for (GPUDevice dev : devices) {
+			devs.add(dev.getModel());
+		}
+		return devs;
+	}
+	
+	public static GPUDevice getGPUDevice(String device_model) {
+		if (device_model == null) {
+			return null;
+		}
+		
+		if (devices == null) {
+			generate();
+		}
+		
+		if (devices == null) {
+			return null;
+		}
+		
+		for (GPUDevice dev : devices) {
+			if (device_model.equals(dev.getCudaName()) || device_model.equals(dev.getModel())) {
+				return dev;
+			}
+		}
+		return null;
+	}
+}