diff --git a/qcom/lahaina-thermal.dtsi b/qcom/lahaina-thermal.dtsi
index ea3307042725..58f5763c1df1 100644
--- a/qcom/lahaina-thermal.dtsi
+++ b/qcom/lahaina-thermal.dtsi
@@ -47,6 +47,13 @@
 		reg = <0x00637000 0x1000>;
 		interrupts = <GIC_SPI 237 IRQ_TYPE_LEVEL_HIGH>;
 	};
+
+	cx_pe: cx_rdpm_pe@0x00635000 {
+		compatible = "qcom,policy-engine";
+		#thermal-sensor-cells = <0>;
+		reg = <0x00635000 0x1000>;
+		interrupts = <GIC_SPI 243 IRQ_TYPE_LEVEL_HIGH>;
+	};
 };
 
 &cpufreq_hw {
@@ -1183,5 +1190,59 @@
 			};
 		};
 	};
+
+	cx-pe-step {
+		polling-delay-passive = <0>;
+		polling-delay = <0>;
+		thermal-sensors = <&cx_pe>;
+		thermal-governor = "step_wise";
+		trips {
+			cx_pe_config1: cx-pe-config1 {
+				temperature = <1>;
+				hysteresis = <1>;
+				type = "passive";
+			};
+
+			cx_pe_config2: cx-pe-config2 {
+				temperature = <2>;
+				hysteresis = <1>;
+				type = "passive";
+			};
+
+			cx_pe_config3: cx-pe-config3 {
+				temperature = <3>;
+				hysteresis = <1>;
+				type = "passive";
+			};
+
+			cx_pe_config4: cx-pe-config4 {
+				temperature = <5>;
+				hysteresis = <2>;
+				type = "passive";
+			};
+		};
+
+		cooling-maps {
+			cx_pe_cdev_1 {
+				trip = <&cx_pe_config1>;
+				cooling-device = <&cdsp_sw 1 1>;
+			};
+
+			cx_pe_cdev_2 {
+				trip = <&cx_pe_config2>;
+				cooling-device = <&cdsp_sw 2 2>;
+			};
+
+			cx_pe_cdev_3 {
+				trip = <&cx_pe_config3>;
+				cooling-device = <&cdsp_sw 3 3>;
+			};
+
+			cx_pe_cdev_4 {
+				trip = <&cx_pe_config4>;
+				cooling-device = <&cdsp_sw 4 4>;
+			};
+		};
+	};
 };