Add allocator api to get alignment based on all buffer types

tt_metal/hw/inc/dataflow_api.h

@@ -1108,7 +1108,8 @@ struct InterleavedPow2AddrGen {
     const uint32_t bank_base_address;
     const uint32_t log_base_2_of_page_size;  // WARNING: This struct is used for optimized get_noc_addr in which case
                                              // you know that bank_unit_size is a power of 2
-    const uint32_t log_base_2_of_allocator_alignment = interleaved_addr_gen::get_log_base2_of_allocator_alignment<DRAM>();
+    static constexpr uint32_t log_base_2_of_allocator_alignment =
+        interleaved_addr_gen::get_log_base2_of_allocator_alignment<DRAM>();
     const uint32_t aligned_log_base_2_of_page_size = this->log_base_2_of_page_size > log_base_2_of_allocator_alignment
                                                          ? this->log_base_2_of_page_size
                                                          : log_base_2_of_allocator_alignment;
@@ -1224,7 +1225,8 @@ template <bool DRAM>
 struct InterleavedPow2AddrGenFast {
     uint32_t bank_base_address;              // Base address for the whole tensor.
     const uint32_t log_base_2_of_page_size;  // Num bytes in bank unit.
-    const uint32_t log_base_2_of_allocator_alignment = interleaved_addr_gen::get_log_base2_of_allocator_alignment<DRAM>();
+    static constexpr uint32_t log_base_2_of_allocator_alignment =
+        interleaved_addr_gen::get_log_base2_of_allocator_alignment<DRAM>();
     const uint32_t aligned_log_base_2_of_page_size = this->log_base_2_of_page_size > log_base_2_of_allocator_alignment
                                                          ? this->log_base_2_of_page_size
                                                          : log_base_2_of_allocator_alignment;

tt_metal/impl/allocator/allocator.cpp

@@ -349,6 +349,18 @@ const std::vector<uint32_t>& bank_ids_from_logical_core(
     return allocator.logical_core_to_bank_ids.at(buffer_type).at(logical_core);
 }
 
+uint32_t get_alignment(const Allocator& alloator, const BufferType& buffer_type) {
+    switch (buffer_type) {
+        case BufferType::DRAM:
+        case BufferType::TRACE: return alloator.config.dram_alignment;
+        case BufferType::L1:
+        case BufferType::L1_SMALL: return alloator.config.l1_alignment;
+        default: {
+            TT_THROW("Allocator does not support buffer ");
+        }
+    }
+}
+
 Statistics get_statistics(const Allocator& allocator, const BufferType& buffer_type) {
     Statistics stats;
     switch (buffer_type) {

tt_metal/impl/allocator/allocator.hpp

@@ -118,6 +118,8 @@ const std::vector<uint32_t>& bank_ids_from_dram_channel(const Allocator& allocat
 const std::vector<uint32_t>& bank_ids_from_logical_core(
     const Allocator& allocator, BufferType buffer_type, const CoreCoord& logical_core);
 
+uint32_t get_alignment(const Allocator& alloator, const BufferType& buffer_type);
+
 Statistics get_statistics(const Allocator& allocator, const BufferType& buffer_type);
 
 void dump_memory_blocks(const Allocator& allocator, const BufferType& buffer_type, std::ofstream& out);

tt_metal/impl/device/device.cpp

@@ -3388,12 +3388,12 @@ allocator::Statistics Device::get_memory_allocation_statistics(const BufferType
 
 uint32_t Device::get_allocator_alignment(const BufferType &buffer_type) const {
     const auto& allocator = this->get_initialized_allocator();
-    return (buffer_type == BufferType::DRAM) ? this->allocator_->config.dram_alignment : this->allocator_->config.l1_alignment;
+    return allocator::get_alignment(*allocator, buffer_type);
 }
 
-uint32_t Device::get_allocator_alignment(SubDeviceId sub_device_id, const BufferType &buffer_type) const {
+uint32_t Device::get_allocator_alignment(const BufferType &buffer_type, SubDeviceId sub_device_id) const {
     const auto& allocator = this->get_initialized_allocator(sub_device_id);
-    return (buffer_type == BufferType::DRAM) ? this->allocator_->config.dram_alignment : this->allocator_->config.l1_alignment;
+    return allocator::get_alignment(*allocator, buffer_type);
 }
 
 size_t Device::get_l1_small_size() const {

tt_metal/impl/device/device.hpp

@@ -197,7 +197,7 @@ class Device {
     allocator::Statistics get_memory_allocation_statistics(const BufferType &buffer_type, SubDeviceId sub_device_id) const;
 
     uint32_t get_allocator_alignment(const BufferType &buffer_type) const;
-    uint32_t get_allocator_alignment(SubDeviceId sub_device_id, const BufferType &buffer_type) const;
+    uint32_t get_allocator_alignment(const BufferType &buffer_type, SubDeviceId sub_device_id) const;
 
     size_t get_l1_small_size() const;
     size_t get_l1_small_size(SubDeviceId sub_device_id) const;