1 files changed, 4 insertions, 0 deletions

diff --git a/drivers/gpu/nvgpu/gk20a/platform_gk20a_tegra.c b/drivers/gpu/nvgpu/gk20a/platform_gk20a_tegra.c
index 28f82816..148496dd 100644
--- a/drivers/gpu/nvgpu/gk20a/platform_gk20a_tegra.c
+++ b/drivers/gpu/nvgpu/gk20a/platform_gk20a_tegra.c
@@ -135,6 +135,7 @@ static int gk20a_tegra_secure_alloc(struct platform_device *pdev,
                                     struct gr_ctx_buffer_desc *desc,
                                     size_t size)
 {
+        struct gk20a_platform *platform = platform_get_drvdata(pdev);
         struct device *dev = &pdev->dev;
         DEFINE_DMA_ATTRS(attrs);
         dma_addr_t iova;
@@ -142,6 +143,9 @@ static int gk20a_tegra_secure_alloc(struct platform_device *pdev,
         struct page *page;
         int err = 0;
 
+        if (!platform->secure_alloc_ready)
+                return -EINVAL;
+
         (void)dma_alloc_attrs(&tegra_vpr_dev, size, &iova,
                                       DMA_MEMORY_NOMAP, &attrs);
         if (dma_mapping_error(&tegra_vpr_dev, iova))