#0: Add grid_size() to CoreRange

tenstorrent · Jan 25, 2024 · cff69f7 · cff69f7
1 parent 66d2a9b
commit cff69f7
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/tt_metal/common/core_coord.h b/tt_metal/common/core_coord.h
@@ -161,6 +161,8 @@ struct CoreRange {
     std::string str() const { return "[" + start.str() + " - " + end.str() + "]"; }
 
     size_t size() const { return (this->end.x - this->start.x + 1) * (this->end.y - this->start.y + 1); }
+
+    CoreCoord grid_size() const { return {this->end.x - this->start.x + 1, this->end.y - this->start.y + 1}; }
 };
 
 constexpr inline bool operator==(const CoreRange &a, const CoreRange &b) { return a.start == b.start && a.end == b.end; }