1 files changed, 7 insertions, 1 deletions

diff --git a/drivers/gpu/nvgpu/os/linux/sysfs.c b/drivers/gpu/nvgpu/os/linux/sysfs.c
index 2ab29649..84a311de 100644
--- a/drivers/gpu/nvgpu/os/linux/sysfs.c
+++ b/drivers/gpu/nvgpu/os/linux/sysfs.c
@@ -259,13 +259,19 @@ static ssize_t railgate_enable_store(struct device *dev,
         unsigned long railgate_enable = 0;
         /* dev is guaranteed to be valid here. Ok to de-reference */
         struct gk20a *g = get_gk20a(dev);
+        struct gk20a_platform *platform = dev_get_drvdata(dev);
         bool enabled = nvgpu_is_enabled(g, NVGPU_CAN_RAILGATE);
         int err;
 
         if (kstrtoul(buf, 10, &railgate_enable) < 0)
                 return -EINVAL;
 
-        if (railgate_enable && !enabled) {
+        if (!platform->can_railgate_init) {
+                nvgpu_err(g, "Railgating is not supported");
+                return -EINVAL;
+        }
+
+        if (railgate_enable) {
                 __nvgpu_set_enabled(g, NVGPU_CAN_RAILGATE, true);
                 pm_runtime_set_autosuspend_delay(dev, g->railgate_delay);
         } else if (railgate_enable == 0 && enabled) {