1 files changed, 8 insertions, 0 deletions

diff --git a/drivers/gpu/nvgpu/common/linux/soc.c b/drivers/gpu/nvgpu/common/linux/soc.c
index 305d8bd9..1b27d6f1 100644
--- a/drivers/gpu/nvgpu/common/linux/soc.c
+++ b/drivers/gpu/nvgpu/common/linux/soc.c
@@ -12,6 +12,7 @@
  */
 
 #include <soc/tegra/chip-id.h>
+#include <soc/tegra/fuse.h>
 #include <soc/tegra/tegra_bpmp.h>
 #ifdef CONFIG_TEGRA_HV_MANAGER
 #include <soc/tegra/virt/syscalls.h>
@@ -46,6 +47,13 @@ bool nvgpu_is_bpmp_running(struct gk20a *g)
         return tegra_bpmp_running();
 }
 
+bool nvgpu_is_soc_t194_a01(struct gk20a *g)
+{
+        return ((tegra_get_chip_id() == TEGRA194 &&
+                        tegra_chip_get_revision() == TEGRA194_REVISION_A01) ?
+                true : false);
+}
+
 #ifdef CONFIG_TEGRA_HV_MANAGER
 /* When nvlink is enabled on dGPU, we need to use physical memory addresses.
  * There is no SMMU translation. However, the device initially enumerates as a