diff --git a/litedram/init.py b/litedram/init.py
index 5d8d53d..4b684c5 100644
--- a/litedram/init.py
+++ b/litedram/init.py
@@ -690,6 +690,8 @@ def get_sdram_phy_c_header(phy_settings, timing_settings):
             'K7LPDDR4PHY', 'V7LPDDR4PHY']:
         r += "#define SDRAM_PHY_WRITE_LATENCY_CALIBRATION_CAPABLE\n"
         r += "#define SDRAM_PHY_READ_LEVELING_CAPABLE\n"
+    if phytype in ['K7LPDDR4PHY', 'V7LPDDR4PHY']:
+        r += "#define SDRAM_PHY_WRITE_DQ_DQS_TRAINING_CAPABLE\n"
     if phytype in ["ECP5DDRPHY"]:
         r += "#define SDRAM_PHY_READ_LEVELING_CAPABLE\n"
     if phytype in ["LPDDR4SIMPHY"]: