Files
OBLITERATUS/tests/fixtures/distributed_checkpoints/v1/cases/world2-uneven-1d/case.json
T

151 lines
3.1 KiB
JSON

{
"case_id": "world2-uneven-1d",
"components": [
"full_model"
],
"expected_manifest_digest": "sha256:d0e2e96c6bdfb6ebd29470ef1f789b1a6a8f1ebc89624538cf56600ad3f4cb88",
"features": [
"padding",
"uneven_1d"
],
"fragments": [
{
"component_id": "model",
"dtype": "int64",
"element_extent": [
3
],
"element_offset": [
0
],
"evidence_refs": [
"synthetic:w2-r0"
],
"fqn": "model.weight",
"fragment_digest": "sha256:ab25350e3e65efebe24584461683ecda68725576e825e550038b90e7b1479946",
"fragment_id": "w2-r0",
"global_shape": [
7
],
"local_shape": [
4
],
"logical_tensor_id": "model.weight",
"padding": {
"after": [
1
],
"before": [
0
],
"semantic": "producer_declared"
},
"partition_axes": [
0
],
"payload_file": "rank-00000.safetensors",
"payload_key": "w2-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",
0
]
]
},
{
"component_id": "model",
"dtype": "int64",
"element_extent": [
4
],
"element_offset": [
3
],
"evidence_refs": [
"synthetic:w2-r1"
],
"fqn": "model.weight",
"fragment_digest": "sha256:6df0e128a82e6d8c332ed546fca7d48406970ea60531b4d64ee41bdffc3d5da4",
"fragment_id": "w2-r1",
"global_shape": [
7
],
"local_shape": [
4
],
"logical_tensor_id": "model.weight",
"padding": {
"after": [
0
],
"before": [
0
],
"semantic": "none"
},
"partition_axes": [
0
],
"payload_file": "rank-00001.safetensors",
"payload_key": "w2-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",
1
]
]
}
],
"oracle_file": "oracles.safetensors",
"oracles": [
{
"dtype": "int64",
"logical_tensor_id": "model.weight",
"payload_key": "tensor_000",
"sha256": "sha256:81845a01dafa45c9b26e10a7af52a92e8604d5d8ef690f1e3ccdcfe3b5c6ae98",
"shape": [
7
]
}
],
"schema_id": "obliteratus.checkpoint-fixture-case",
"schema_version": "1.0.0",
"topology": {
"source": {
"tp": 2
},
"target": {
"world_size": 1
}
},
"world_size": 2
}