add TT_FATAL check

tenstorrent · Mar 6, 2025 · bcf6d29 · bcf6d29
1 parent 6f1e5de
commit bcf6d29
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/ttnn/cpp/ttnn/distributed/api.cpp b/ttnn/cpp/ttnn/distributed/api.cpp
@@ -117,6 +117,7 @@ Tensor aggregate_as_tensor(
         auto storage = MultiDeviceHostStorage{config, std::move(host_owned_buffers), specs};
         return Tensor(std::move(storage), reference_shard.get_tensor_spec());
     } else {
+        TT_FATAL(storage_type == StorageType::DEVICE, "Unexpected storage type {}", storage_type);
         std::vector<int> ordered_device_ids;
         std::unordered_map<int, ttnn::TensorSpec> specs;
         std::unordered_map<int, std::shared_ptr<Buffer>> device_buffers;