Raincloud/sheepit-shadow-nabber
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Feat: send cuda compute capability to the server

Browse Source
This commit is contained in:
Laurent Clouet
2025-06-26 14:42:16 +02:00
parent faab3392bd
commit 19788260cb
4 changed files with 18 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
src/main/java/com/sheepit/client/hardware/gpu/GPUDevice.java
View File

@@ -33,6 +33,8 @@ public class GPUDevice {
private String driverVersion; private String driverVersion;
private float computeCapability;
public GPUDevice(String type, String model, long ram, String id) { public GPUDevice(String type, String model, long ram, String id) {
this.type = type; this.type = type;
this.model = model; this.model = model;
@@ -46,7 +48,7 @@ public class GPUDevice {
} }
@Override public String toString() { @Override public String toString() {
return "GPUDevice [type=" + type + ", model='" + model + "', memory=" + memory + ", id=" + id + ", driverVersion=" + driverVersion + "]"; return "GPUDevice [type=" + type + ", model='" + model + "', memory=" + memory + ", id=" + id + ", driverVersion=" + driverVersion + ", computeCapability=" + computeCapability + "]";
} }
public static int compareVersions(String version1, String version2) { public static int compareVersions(String version1, String version2) {

5
src/main/java/com/sheepit/client/hardware/gpu/nvidia/CUDeviceAttribute.java
View File

@@ -43,4 +43,9 @@ public class CUDeviceAttribute {
* Compute capability major version. Return value needs to be > 5 for OptiX support * Compute capability major version. Return value needs to be > 5 for OptiX support
*/ */
public static final int CU_DEVICE_ATTRIBUTE_COMPUTE_CAPABILITY_MAJOR = 75; public static final int CU_DEVICE_ATTRIBUTE_COMPUTE_CAPABILITY_MAJOR = 75;
/**
* Compute capability minor version.
*/
public static final int CU_DEVICE_ATTRIBUTE_COMPUTE_CAPABILITY_MINOR = 76;
} }

8
src/main/java/com/sheepit/client/hardware/gpu/nvidia/Nvidia.java
View File

@@ -226,6 +226,13 @@ public class Nvidia implements GPULister {
continue; continue;
} }
IntByReference computeCapabilityMinor = new IntByReference();
result = cudalib.cuDeviceGetAttribute(computeCapabilityMinor, CUDeviceAttribute.CU_DEVICE_ATTRIBUTE_COMPUTE_CAPABILITY_MINOR, aDevice.getValue());
if (result != CUresult.CUDA_SUCCESS) {
Log.getInstance().error("Nvidia::getGpus cuDeviceGetAttribute for CU_DEVICE_ATTRIBUTE_COMPUTE_CAPABILITY_MINOR failed (ret: " + CUresult.stringFor(result) + ")");
continue;
}
IntByReference pciDomainId = new IntByReference(); IntByReference pciDomainId = new IntByReference();
IntByReference pciBusId = new IntByReference(); IntByReference pciBusId = new IntByReference();
@@ -307,6 +314,7 @@ public class Nvidia implements GPULister {
gpu.setDriverVersion(driverVersion); gpu.setDriverVersion(driverVersion);
// for backward compatibility generate a CUDA_N id // for backward compatibility generate a CUDA_N id
gpu.setOldId(TYPE + "_" + num); gpu.setOldId(TYPE + "_" + num);
gpu.setComputeCapability(computeCapabilityMajor.getValue() + (float)computeCapabilityMinor.getValue() / 10);
devices.add(gpu); devices.add(gpu);
} }

2
src/main/java/com/sheepit/client/network/ServerRequest.java
View File

@@ -352,7 +352,9 @@ public class ServerRequest extends Thread {
urlBuilder.addQueryParameter("gpu_model", user_config.getGPUDevice().getModel()) urlBuilder.addQueryParameter("gpu_model", user_config.getGPUDevice().getModel())
.addQueryParameter("gpu_ram", String.valueOf(user_config.getGPUDevice().getMemory())) .addQueryParameter("gpu_ram", String.valueOf(user_config.getGPUDevice().getMemory()))
.addQueryParameter("gpu_type", user_config.getGPUDevice().getType()) .addQueryParameter("gpu_type", user_config.getGPUDevice().getType())
.addQueryParameter("gpu_cuda_compute_capability", String.valueOf(user_config.getGPUDevice().getComputeCapability()))
.addQueryParameter("gpu_driver_version", user_config.getGPUDevice().getDriverVersion()); .addQueryParameter("gpu_driver_version", user_config.getGPUDevice().getDriverVersion());
} }
Response response = this.HTTPRequest(urlBuilder, RequestBody.create(this.generateXMLForMD5cache(), MediaType.parse("application/xml"))); Response response = this.HTTPRequest(urlBuilder, RequestBody.create(this.generateXMLForMD5cache(), MediaType.parse("application/xml")));