Files
OBLITERATUS/tests/fixtures/distributed_checkpoints/v1/cases/world4-uneven-2d/case.json
T

310 lines
6.1 KiB
JSON

{
"case_id": "world4-uneven-2d",
"components": [
"full_model"
],
"expected_manifest_digest": "sha256:356bddfa143562828982c2fd80b83d4a2e0c278814d844568b99aeed1f6cb310",
"features": [
"uneven_2d"
],
"fragments": [
{
"component_id": "model",
"dtype": "float32",
"element_extent": [
2,
3
],
"element_offset": [
0,
0
],
"evidence_refs": [
"synthetic:w4-r0"
],
"fqn": "model.weight",
"fragment_digest": "sha256:e902ad65c5c6702e7ca1da427d13ec926431470bc155ff2983b362cb10d2dff5",
"fragment_id": "w4-r0",
"global_shape": [
5,
7
],
"local_shape": [
2,
3
],
"logical_tensor_id": "model.weight",
"padding": {
"after": [
0,
0
],
"before": [
0,
0
],
"semantic": "none"
},
"partition_axes": [
0,
1
],
"payload_file": "rank-00000.safetensors",
"payload_key": "w4-r0",
"replica": {
"group_id": null,
"member_count": 1,
"member_index": 0
},
"role": "parameter",
"shard_digest_ref": "source-digest:rank-00000.safetensors",
"shard_file_id": "rank-00000.safetensors",
"shared_storage_id": null,
"tie_group_id": null,
"topology_coordinates": [
[
"rank",
0
],
[
"tp_col",
0
],
[
"tp_row",
0
]
]
},
{
"component_id": "model",
"dtype": "float32",
"element_extent": [
2,
4
],
"element_offset": [
0,
3
],
"evidence_refs": [
"synthetic:w4-r1"
],
"fqn": "model.weight",
"fragment_digest": "sha256:8dc8137a498004d656f20c18b4726bf12aab21d7010b9980480317d6df38e8c8",
"fragment_id": "w4-r1",
"global_shape": [
5,
7
],
"local_shape": [
2,
4
],
"logical_tensor_id": "model.weight",
"padding": {
"after": [
0,
0
],
"before": [
0,
0
],
"semantic": "none"
},
"partition_axes": [
0,
1
],
"payload_file": "rank-00001.safetensors",
"payload_key": "w4-r1",
"replica": {
"group_id": null,
"member_count": 1,
"member_index": 0
},
"role": "parameter",
"shard_digest_ref": "source-digest:rank-00001.safetensors",
"shard_file_id": "rank-00001.safetensors",
"shared_storage_id": null,
"tie_group_id": null,
"topology_coordinates": [
[
"rank",
1
],
[
"tp_col",
1
],
[
"tp_row",
0
]
]
},
{
"component_id": "model",
"dtype": "float32",
"element_extent": [
3,
3
],
"element_offset": [
2,
0
],
"evidence_refs": [
"synthetic:w4-r2"
],
"fqn": "model.weight",
"fragment_digest": "sha256:cd2d91a4484f0cd6236034895436acbee187e02fd96900593a2512d9882f7e0e",
"fragment_id": "w4-r2",
"global_shape": [
5,
7
],
"local_shape": [
3,
3
],
"logical_tensor_id": "model.weight",
"padding": {
"after": [
0,
0
],
"before": [
0,
0
],
"semantic": "none"
},
"partition_axes": [
0,
1
],
"payload_file": "rank-00002.safetensors",
"payload_key": "w4-r2",
"replica": {
"group_id": null,
"member_count": 1,
"member_index": 0
},
"role": "parameter",
"shard_digest_ref": "source-digest:rank-00002.safetensors",
"shard_file_id": "rank-00002.safetensors",
"shared_storage_id": null,
"tie_group_id": null,
"topology_coordinates": [
[
"rank",
2
],
[
"tp_col",
0
],
[
"tp_row",
1
]
]
},
{
"component_id": "model",
"dtype": "float32",
"element_extent": [
3,
4
],
"element_offset": [
2,
3
],
"evidence_refs": [
"synthetic:w4-r3"
],
"fqn": "model.weight",
"fragment_digest": "sha256:3c4e748352f778507efe297633562c96b36d3e68030f1c230266d63fd2994179",
"fragment_id": "w4-r3",
"global_shape": [
5,
7
],
"local_shape": [
3,
4
],
"logical_tensor_id": "model.weight",
"padding": {
"after": [
0,
0
],
"before": [
0,
0
],
"semantic": "none"
},
"partition_axes": [
0,
1
],
"payload_file": "rank-00003.safetensors",
"payload_key": "w4-r3",
"replica": {
"group_id": null,
"member_count": 1,
"member_index": 0
},
"role": "parameter",
"shard_digest_ref": "source-digest:rank-00003.safetensors",
"shard_file_id": "rank-00003.safetensors",
"shared_storage_id": null,
"tie_group_id": null,
"topology_coordinates": [
[
"rank",
3
],
[
"tp_col",
1
],
[
"tp_row",
1
]
]
}
],
"oracle_file": "oracles.safetensors",
"oracles": [
{
"dtype": "float32",
"logical_tensor_id": "model.weight",
"payload_key": "tensor_000",
"sha256": "sha256:471d327907fc83cb6703d3424393e5caeefd627fa86d8b1b2f07d3045b6e1433",
"shape": [
5,
7
]
}
],
"schema_id": "obliteratus.checkpoint-fixture-case",
"schema_version": "1.0.0",
"topology": {
"source": {
"tp_columns": 2,
"tp_rows": 2
},
"target": {
"world_size": 1
}
},
"world_size": 4
}