diff options
| author | Niels Thiele <noleu66@posteo.net> | 2025-07-15 11:29:47 +0200 |
|---|---|---|
| committer | GitHub <noreply@github.com> | 2025-07-15 11:29:47 +0200 |
| commit | b2dc97dc84f56174ede9f273999ade2ed059d431 (patch) | |
| tree | 1b5d6d775890375f46b533c7aa78e492a88afc3f /opendc-trace/opendc-trace-api | |
| parent | 0203254b709614fa732c114aa25916f61b8b3275 (diff) | |
multi gpu support (#351)
Diffstat (limited to 'opendc-trace/opendc-trace-api')
2 files changed, 13 insertions, 1 deletions
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<String>?) : 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<String>?) : 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 |