OBLITERATUS/tests/fixtures/distributed_checkpoints/v1/cases/tp2-pp2-to-single/case.json

{
  "case_id": "tp2-pp2-to-single",
  "components": [
    "full_model"
  ],
  "expected_manifest_digest": "sha256:03da2d1dafe858e225174f12394d243dcaf70516c2649bca85c57dce91dff323",
  "features": [
    "pipeline_parallel",
    "topology_a_to_b"
  ],
  "fragments": [
    {
      "component_id": "model",
      "dtype": "float32",
      "element_extent": [
        2,
        2
      ],
      "element_offset": [
        0,
        0
      ],
      "evidence_refs": [
        "synthetic:tp-pp-r0"
      ],
      "fqn": "model.weight",
      "fragment_digest": "sha256:36f52612ab7fad5ef2adca0d34f41471c0c38f50cc92d9a47fa662631d3e1268",
      "fragment_id": "tp-pp-r0",
      "global_shape": [
        4,
        4
      ],
      "local_shape": [
        2,
        2
      ],
      "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": "tp-pp-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": [
        [
          "pp",
          0
        ],
        [
          "rank",
          0
        ],
        [
          "tp",
          0
        ]
      ]
    },
    {
      "component_id": "model",
      "dtype": "float32",
      "element_extent": [
        2,
        2
      ],
      "element_offset": [
        0,
        2
      ],
      "evidence_refs": [
        "synthetic:tp-pp-r1"
      ],
      "fqn": "model.weight",
      "fragment_digest": "sha256:762921576e058fd1360006105a48a8225d6baa27085f7d30abed455c288f9308",
      "fragment_id": "tp-pp-r1",
      "global_shape": [
        4,
        4
      ],
      "local_shape": [
        2,
        2
      ],
      "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": "tp-pp-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": [
        [
          "pp",
          0
        ],
        [
          "rank",
          1
        ],
        [
          "tp",
          1
        ]
      ]
    },
    {
      "component_id": "model",
      "dtype": "float32",
      "element_extent": [
        2,
        2
      ],
      "element_offset": [
        2,
        0
      ],
      "evidence_refs": [
        "synthetic:tp-pp-r2"
      ],
      "fqn": "model.weight",
      "fragment_digest": "sha256:5cca39bfe5a82135bad95ae98f2df321fe08ea75b68ea8f939ffa90845138146",
      "fragment_id": "tp-pp-r2",
      "global_shape": [
        4,
        4
      ],
      "local_shape": [
        2,
        2
      ],
      "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": "tp-pp-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": [
        [
          "pp",
          1
        ],
        [
          "rank",
          2
        ],
        [
          "tp",
          0
        ]
      ]
    },
    {
      "component_id": "model",
      "dtype": "float32",
      "element_extent": [
        2,
        2
      ],
      "element_offset": [
        2,
        2
      ],
      "evidence_refs": [
        "synthetic:tp-pp-r3"
      ],
      "fqn": "model.weight",
      "fragment_digest": "sha256:dbb7087f7cd13a205e0dcec5093a66dcfac4f6c237afd8dcafd64d166227083a",
      "fragment_id": "tp-pp-r3",
      "global_shape": [
        4,
        4
      ],
      "local_shape": [
        2,
        2
      ],
      "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": "tp-pp-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": [
        [
          "pp",
          1
        ],
        [
          "rank",
          3
        ],
        [
          "tp",
          1
        ]
      ]
    }
  ],
  "oracle_file": "oracles.safetensors",
  "oracles": [
    {
      "dtype": "float32",
      "logical_tensor_id": "model.weight",
      "payload_key": "tensor_000",
      "sha256": "sha256:58dda328598e2f7fe472621bfc54935aaa354d1a6ebcaf9562cd743fd575eb19",
      "shape": [
        4,
        4
      ]
    }
  ],
  "schema_id": "obliteratus.checkpoint-fixture-case",
  "schema_version": "1.0.0",
  "topology": {
    "source": {
      "pp": 2,
      "tp": 2
    },
    "target": {
      "world_size": 1
    }
  },
  "world_size": 4
}