diff --git a/drivers/amlogic/media/osd/osd.h b/drivers/amlogic/media/osd/osd.h
index cc2e8fbe7cfc..2884b1fd8be2 100644
--- a/drivers/amlogic/media/osd/osd.h
+++ b/drivers/amlogic/media/osd/osd.h
@@ -278,6 +278,7 @@ enum cpuid_type_e {
 	__MESON_CPU_MAJOR_ID_G12B,
 	__MESON_CPU_MAJOR_ID_TL1,
 	__MESON_CPU_MAJOR_ID_SM1,
+	__MESON_CPU_MAJOR_ID_TM2,
 	__MESON_CPU_MAJOR_ID_UNKNOWN,
 };
 
diff --git a/drivers/amlogic/media/osd/osd_fb.c b/drivers/amlogic/media/osd/osd_fb.c
index 35b78be758d1..5209af990f56 100644
--- a/drivers/amlogic/media/osd/osd_fb.c
+++ b/drivers/amlogic/media/osd/osd_fb.c
@@ -4055,6 +4055,10 @@ static const struct of_device_id meson_fb_dt_match[] = {
 		.compatible = "amlogic, meson-sm1",
 		.data = &osd_sm1,
 	},
+	{
+		.compatible = "amlogic, meson-tm2",
+		.data = &osd_tm2,
+	},
 	{},
 };
 
diff --git a/drivers/amlogic/media/osd/osd_hw.c b/drivers/amlogic/media/osd/osd_hw.c
index 72c5d1d649f1..64d2869c56b6 100644
--- a/drivers/amlogic/media/osd/osd_hw.c
+++ b/drivers/amlogic/media/osd/osd_hw.c
@@ -1680,6 +1680,7 @@ static u32 osd_get_hw_reset_flag(void)
 	case __MESON_CPU_MAJOR_ID_G12B:
 	case __MESON_CPU_MAJOR_ID_TL1:
 	case __MESON_CPU_MAJOR_ID_SM1:
+	case __MESON_CPU_MAJOR_ID_TM2:
 		{
 		int i, afbc_enable = 0;
 
@@ -5052,8 +5053,7 @@ static void osd_update_disp_osd_rotate(u32 index)
 	const struct vinfo_s *vinfo;
 	int out_y_crop_start, out_y_crop_end;
 
-	if (osd_hw.osd_meson_dev.cpu_id < __MESON_CPU_MAJOR_ID_G12B ||
-		osd_hw.osd_meson_dev.cpu_id == __MESON_CPU_MAJOR_ID_SM1)
+	if (osd_hw.osd_meson_dev.cpu_id != __MESON_CPU_MAJOR_ID_G12B)
 		return;
 	src_fmt = get_viu2_src_format();
 	src_data.x = 0;