OBLITERATUS/tests/fixtures/distributed_checkpoints/v1/cases/world4-dp-replicas/case.json

{
  "case_id": "world4-dp-replicas",
  "components": [
    "full_model"
  ],
  "expected_manifest_digest": "sha256:d45ab4d7db81d4f7bd55e7ccb04bfb7c936b0bd320f1b57e77d7d02d3d488ff3",
  "features": [
    "dp_replica"
  ],
  "fragments": [
    {
      "component_id": "model",
      "dtype": "float32",
      "element_extent": [
        3
      ],
      "element_offset": [
        0
      ],
      "evidence_refs": [
        "synthetic:dp-r0"
      ],
      "fqn": "model.weight",
      "fragment_digest": "sha256:7f8a2918261b8d80cae770a9aae209b29d8b39654c62aee0013a423796ddb655",
      "fragment_id": "dp-r0",
      "global_shape": [
        3
      ],
      "local_shape": [
        3
      ],
      "logical_tensor_id": "model.weight",
      "padding": {
        "after": [
          0
        ],
        "before": [
          0
        ],
        "semantic": "none"
      },
      "partition_axes": [],
      "payload_file": "rank-00000.safetensors",
      "payload_key": "dp-r0",
      "replica": {
        "group_id": "dp-full",
        "member_count": 4,
        "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": [
        [
          "dp",
          0
        ],
        [
          "rank",
          0
        ]
      ]
    },
    {
      "component_id": "model",
      "dtype": "float32",
      "element_extent": [
        3
      ],
      "element_offset": [
        0
      ],
      "evidence_refs": [
        "synthetic:dp-r1"
      ],
      "fqn": "model.weight",
      "fragment_digest": "sha256:7f8a2918261b8d80cae770a9aae209b29d8b39654c62aee0013a423796ddb655",
      "fragment_id": "dp-r1",
      "global_shape": [
        3
      ],
      "local_shape": [
        3
      ],
      "logical_tensor_id": "model.weight",
      "padding": {
        "after": [
          0
        ],
        "before": [
          0
        ],
        "semantic": "none"
      },
      "partition_axes": [],
      "payload_file": "rank-00001.safetensors",
      "payload_key": "dp-r1",
      "replica": {
        "group_id": "dp-full",
        "member_count": 4,
        "member_index": 1
      },
      "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": [
        [
          "dp",
          1
        ],
        [
          "rank",
          1
        ]
      ]
    },
    {
      "component_id": "model",
      "dtype": "float32",
      "element_extent": [
        3
      ],
      "element_offset": [
        0
      ],
      "evidence_refs": [
        "synthetic:dp-r2"
      ],
      "fqn": "model.weight",
      "fragment_digest": "sha256:7f8a2918261b8d80cae770a9aae209b29d8b39654c62aee0013a423796ddb655",
      "fragment_id": "dp-r2",
      "global_shape": [
        3
      ],
      "local_shape": [
        3
      ],
      "logical_tensor_id": "model.weight",
      "padding": {
        "after": [
          0
        ],
        "before": [
          0
        ],
        "semantic": "none"
      },
      "partition_axes": [],
      "payload_file": "rank-00002.safetensors",
      "payload_key": "dp-r2",
      "replica": {
        "group_id": "dp-full",
        "member_count": 4,
        "member_index": 2
      },
      "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": [
        [
          "dp",
          2
        ],
        [
          "rank",
          2
        ]
      ]
    },
    {
      "component_id": "model",
      "dtype": "float32",
      "element_extent": [
        3
      ],
      "element_offset": [
        0
      ],
      "evidence_refs": [
        "synthetic:dp-r3"
      ],
      "fqn": "model.weight",
      "fragment_digest": "sha256:7f8a2918261b8d80cae770a9aae209b29d8b39654c62aee0013a423796ddb655",
      "fragment_id": "dp-r3",
      "global_shape": [
        3
      ],
      "local_shape": [
        3
      ],
      "logical_tensor_id": "model.weight",
      "padding": {
        "after": [
          0
        ],
        "before": [
          0
        ],
        "semantic": "none"
      },
      "partition_axes": [],
      "payload_file": "rank-00003.safetensors",
      "payload_key": "dp-r3",
      "replica": {
        "group_id": "dp-full",
        "member_count": 4,
        "member_index": 3
      },
      "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": [
        [
          "dp",
          3
        ],
        [
          "rank",
          3
        ]
      ]
    }
  ],
  "oracle_file": "oracles.safetensors",
  "oracles": [
    {
      "dtype": "float32",
      "logical_tensor_id": "model.weight",
      "payload_key": "tensor_000",
      "sha256": "sha256:7f8a2918261b8d80cae770a9aae209b29d8b39654c62aee0013a423796ddb655",
      "shape": [
        3
      ]
    }
  ],
  "schema_id": "obliteratus.checkpoint-fixture-case",
  "schema_version": "1.0.0",
  "topology": {
    "source": {
      "dp": 4
    },
    "target": {
      "world_size": 1
    }
  },
  "world_size": 4
}