1 files changed, 7 insertions, 0 deletions

diff --git a/drivers/gpu/drm/tegra/gr2d.c b/drivers/gpu/drm/tegra/gr2d.c
index 02cd3e37a6ec..fbe0b8b25b42 100644
--- a/drivers/gpu/drm/tegra/gr2d.c
+++ b/drivers/gpu/drm/tegra/gr2d.c
@@ -109,10 +109,17 @@ static int gr2d_is_addr_reg(struct device *dev, u32 class, u32 offset)
         return 0;
 }
 
+static int gr2d_is_valid_class(u32 class)
+{
+        return (class == HOST1X_CLASS_GR2D ||
+                class == HOST1X_CLASS_GR2D_SB);
+}
+
 static const struct tegra_drm_client_ops gr2d_ops = {
         .open_channel = gr2d_open_channel,
         .close_channel = gr2d_close_channel,
         .is_addr_reg = gr2d_is_addr_reg,
+        .is_valid_class = gr2d_is_valid_class,
         .submit = tegra_drm_submit,
 };