From b2dc97dc84f56174ede9f273999ade2ed059d431 Mon Sep 17 00:00:00 2001 From: Niels Thiele Date: Tue, 15 Jul 2025 11:29:47 +0200 Subject: multi gpu support (#351) --- .../trace/formats/opendc/parquet/ResourceStateReadSupport.kt | 12 ++++++++++++ .../opendc/parquet/ResourceStateRecordMaterializer.kt | 2 +- 2 files changed, 13 insertions(+), 1 deletion(-) (limited to 'opendc-trace') diff --git a/opendc-trace/opendc-trace-api/src/main/kotlin/org/opendc/trace/formats/opendc/parquet/ResourceStateReadSupport.kt b/opendc-trace/opendc-trace-api/src/main/kotlin/org/opendc/trace/formats/opendc/parquet/ResourceStateReadSupport.kt index e7d35630..53e594de 100644 --- a/opendc-trace/opendc-trace-api/src/main/kotlin/org/opendc/trace/formats/opendc/parquet/ResourceStateReadSupport.kt +++ b/opendc-trace/opendc-trace-api/src/main/kotlin/org/opendc/trace/formats/opendc/parquet/ResourceStateReadSupport.kt @@ -110,6 +110,12 @@ internal class ResourceStateReadSupport(private val projection: List?) : Types .required(PrimitiveType.PrimitiveTypeName.DOUBLE) .named("cpuUsage"), + Types + .optional(PrimitiveType.PrimitiveTypeName.INT32) + .named("gpuCount"), + Types + .optional(PrimitiveType.PrimitiveTypeName.DOUBLE) + .named("gpuUsage"), ) .named("resource_state") @@ -137,6 +143,12 @@ internal class ResourceStateReadSupport(private val projection: List?) : Types .required(PrimitiveType.PrimitiveTypeName.DOUBLE) .named("cpu_usage"), + Types + .optional(PrimitiveType.PrimitiveTypeName.INT32) + .named("gpu_count"), + Types + .optional(PrimitiveType.PrimitiveTypeName.DOUBLE) + .named("gpu_usage"), ) .named("resource_state") diff --git a/opendc-trace/opendc-trace-api/src/main/kotlin/org/opendc/trace/formats/opendc/parquet/ResourceStateRecordMaterializer.kt b/opendc-trace/opendc-trace-api/src/main/kotlin/org/opendc/trace/formats/opendc/parquet/ResourceStateRecordMaterializer.kt index 9ad786d5..a53dcdb2 100644 --- a/opendc-trace/opendc-trace-api/src/main/kotlin/org/opendc/trace/formats/opendc/parquet/ResourceStateRecordMaterializer.kt +++ b/opendc-trace/opendc-trace-api/src/main/kotlin/org/opendc/trace/formats/opendc/parquet/ResourceStateRecordMaterializer.kt @@ -87,7 +87,7 @@ internal class ResourceStateRecordMaterializer(schema: MessageType) : RecordMate localCpuUsage = value } } - "gpu_count", "gpu_cores" -> + "gpu_count", "gpuCount", "gpu_cores", "gpuCores" -> object : PrimitiveConverter() { override fun addInt(value: Int) { localGpuCount = value -- cgit v1.2.3