Derive coordinate types

tenstorrent · Oct 23, 2024 · 7f5e04a · 7f5e04a
1 parent 506b270
commit 7f5e04a
Showing 1 changed file with 43 additions and 11 deletions.
diff --git a/device/tt_xy_pair.h b/device/tt_xy_pair.h
@@ -13,14 +13,46 @@
 using tt_xy_pair = tt::umd::xy_pair;
 using tt_cxy_pair = tt::umd::cxy_pair;
 
-using tt_physical_coords = tt::umd::xy_pair;
-using tt_chip_physical_coords = tt::umd::cxy_pair;
-
-using tt_logical_coords = tt::umd::xy_pair;
-using tt_chip_logical_coords = tt::umd::cxy_pair;
-
-using tt_virtual_coords = tt::umd::xy_pair;
-using tt_chip_virtual_coords = tt::umd::cxy_pair;
-
-using tt_translated_coords = tt::umd::xy_pair;
-using tt_chip_translated_coords = tt::umd::cxy_pair;
+struct tt_physical_coords : public tt_xy_pair {
+    tt_physical_coords() : tt_xy_pair() {}
+    tt_physical_coords(std::size_t x, std::size_t y) : tt_xy_pair(x, y) {}
+};
+
+struct tt_chip_physical_coords : public tt_cxy_pair {
+    tt_chip_physical_coords() : tt_cxy_pair() {}
+    tt_chip_physical_coords(std::size_t ichip, xy_pair pair) : tt_cxy_pair(ichip, pair) {}
+    tt_chip_physical_coords(std::size_t ichip, std::size_t x, std::size_t y) : tt_cxy_pair(ichip, x, y) {}
+};
+
+struct tt_logical_coords : public tt_xy_pair {
+    tt_logical_coords() : tt_xy_pair() {}
+    tt_logical_coords(std::size_t x, std::size_t y) : tt_xy_pair(x, y) {}
+};
+
+struct tt_chip_logical_coords : public tt_cxy_pair {
+    tt_chip_logical_coords() : tt_cxy_pair() {}
+    tt_chip_logical_coords(std::size_t ichip, xy_pair pair) : tt_cxy_pair(ichip, pair) {}
+    tt_chip_logical_coords(std::size_t ichip, std::size_t x, std::size_t y) : tt_cxy_pair(ichip, x, y) {}
+};
+
+struct tt_virtual_coords : public tt_xy_pair {
+    tt_virtual_coords() : tt_xy_pair() {}
+    tt_virtual_coords(std::size_t x, std::size_t y) : tt_xy_pair(x, y) {}
+};
+
+struct tt_chip_virtual_coords : public tt_cxy_pair {
+    tt_chip_virtual_coords() : tt_cxy_pair() {}
+    tt_chip_virtual_coords(std::size_t ichip, xy_pair pair) : tt_cxy_pair(ichip, pair) {}
+    tt_chip_virtual_coords(std::size_t ichip, std::size_t x, std::size_t y) : tt_cxy_pair(ichip, x, y) {}
+};
+
+struct tt_translated_coords : public tt_xy_pair {
+    tt_translated_coords() : tt_xy_pair() {}
+    tt_translated_coords(std::size_t x, std::size_t y) : tt_xy_pair(x, y) {}
+};
+
+struct tt_chip_translated_coords : public tt_cxy_pair {
+    tt_chip_translated_coords() : tt_cxy_pair() {}
+    tt_chip_translated_coords(std::size_t ichip, xy_pair pair) : tt_cxy_pair(ichip, pair) {}
+    tt_chip_translated_coords(std::size_t ichip, std::size_t x, std::size_t y) : tt_cxy_pair(ichip, x, y) {}
+};