From 797cf2119347e3fb4c668b30f7cdad9f39d98101 Mon Sep 17 00:00:00 2001
From: Bryan Wilder Field Lozano <blozano@tenstorrent.com>
Date: Tue, 5 Nov 2024 09:53:42 -0800
Subject: [PATCH] #14636: Update device_command.hpp (#14709)

---
 tt_metal/impl/dispatch/device_command.hpp | 6 +++---
 1 file changed, 3 insertions(+), 3 deletions(-)

diff --git a/tt_metal/impl/dispatch/device_command.hpp b/tt_metal/impl/dispatch/device_command.hpp
index a9ecc2ef4d0..54ef442bec2 100644
--- a/tt_metal/impl/dispatch/device_command.hpp
+++ b/tt_metal/impl/dispatch/device_command.hpp
@@ -270,7 +270,7 @@ class DeviceCommand {
         } else {
             initialize_mcast_cmd(mcast_cmd_dst);
         }
-        this->cmd_write_offsetB = align(this->cmd_write_offsetB, PCIE_ALIGNMENT);
+        this->cmd_write_offsetB = align(this->cmd_write_offsetB, this->pcie_alignment);
     }
 
     void add_notify_dispatch_s_go_signal_cmd(uint8_t wait) {
@@ -289,7 +289,7 @@ class DeviceCommand {
         } else {
             initialize_sem_update_cmd(dispatch_s_sem_update_dst);
         }
-        this->cmd_write_offsetB = align(this->cmd_write_offsetB, PCIE_ALIGNMENT);
+        this->cmd_write_offsetB = align(this->cmd_write_offsetB, this->pcie_alignment);
     }
 
     template <bool inline_data = false>
@@ -394,7 +394,7 @@ class DeviceCommand {
             initialize_set_unicast_only_cores_cmd(set_unicast_only_cores_cmd_dst);
         }
         uint32_t data_sizeB = noc_encodings.size() * sizeof(uint32_t);
-        uint32_t increment_sizeB = align(data_sizeB, PCIE_ALIGNMENT);
+        uint32_t increment_sizeB = align(data_sizeB, this->pcie_alignment);
         this->add_data(noc_encodings.data(), data_sizeB, increment_sizeB);
     }