diff options
Diffstat (limited to 'opendc-trace/opendc-trace-api/src/main/kotlin')
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 |