multi-gpu test

golemfactory · Aug 21, 2024 · 88d2cb1 · 88d2cb1
1 parent 2c71920
commit 88d2cb1
Show file tree

Hide file tree

Showing 11 changed files with 518 additions and 247 deletions.
diff --git a/docker-stack/golem-reputation-backend/benchmark/types.ts b/docker-stack/golem-reputation-backend/benchmark/types.ts
@@ -22,3 +22,24 @@ export interface Benchmark {
     type: string // 'disk', 'cpu', 'memory', 'network', 'gpu'
     data: any // This should ideally be replaced with a more specific interface based on the structure of your benchmark data
 }
+
+
+export interface GPUInfo {
+    name: string;
+    pcie: number;
+    memory_total: number;
+    memory_free: number;
+    cuda_cap: number;
+    // Remove gpu_burn_gflops from here
+}
+
+export interface GPUBenchmarkData {
+    node_id: string;
+    gpus: (GPUInfo & { quantity: number })[];
+    gpu_burn_gflops: number;
+}
+
+  export interface Benchmark {
+    type: string; // 'disk', 'cpu', 'memory', 'network', 'gpu'
+    data: DiskBenchmarkData | CPUBenchmarkData | MemoryBenchmarkData | NetworkBenchmarkData | GPUBenchmarkData;
+  }