diff --git a/tt_metal/distributed/coordinate_translation.cpp b/tt_metal/distributed/coordinate_translation.cpp index 4870acf0e31..5e4be86b0b8 100644 --- a/tt_metal/distributed/coordinate_translation.cpp +++ b/tt_metal/distributed/coordinate_translation.cpp @@ -66,7 +66,7 @@ MeshShape get_system_mesh_shape(size_t system_num_devices) { std::pair get_system_mesh_coordinate_translation_map() { static const auto* cached_translation_map = new std::pair([] { - auto system_num_devices = tt::Cluster::instance().number_of_devices(); + auto system_num_devices = tt::Cluster::instance().number_of_user_devices(); std::string galaxy_mesh_descriptor = "TG.json"; if (tt::Cluster::instance().number_of_pci_devices() == system_num_devices) { diff --git a/tt_metal/distributed/system_mesh.cpp b/tt_metal/distributed/system_mesh.cpp index f1eaf92adb4..e75ff5f41ca 100644 --- a/tt_metal/distributed/system_mesh.cpp +++ b/tt_metal/distributed/system_mesh.cpp @@ -95,9 +95,11 @@ std::vector SystemMesh::Impl::get_mapped_physical_device_ids(const Me // First check if total size fits TT_FATAL( requested_num_rows * requested_num_cols <= system_mesh_rows * system_mesh_cols, - "Requested submesh is too big: {}x{}", + "Requested submesh is too big: {}x{}, SystemMesh shape: {}x{}", requested_num_rows, - requested_num_cols); + requested_num_cols, + system_mesh_rows, + system_mesh_cols); bool is_single_row_or_column = requested_num_rows == 1 or requested_num_cols == 1; if (is_single_row_or_column) {