diff --git a/arch/nvgpu-common.yaml b/arch/nvgpu-common.yaml
index cc7390797..60add3e5a 100644
--- a/arch/nvgpu-common.yaml
+++ b/arch/nvgpu-common.yaml
@@ -139,6 +139,7 @@ nvgpu_err:
 
 pramin:
   safe: yes
+  gpu: dgpu
   owner: Terje B
   sources: [ common/pramin.c,
              include/nvgpu/pramin.h ]
diff --git a/arch/nvgpu-hal-new.yaml b/arch/nvgpu-hal-new.yaml
index 7dd6a0ecc..e6d9b2695 100644
--- a/arch/nvgpu-hal-new.yaml
+++ b/arch/nvgpu-hal-new.yaml
@@ -566,6 +566,7 @@ perf:
 
 pramin:
   safe: yes
+  gpu: dgpu
   sources: [ hal/pramin/pramin_gp10b.c,
              hal/pramin/pramin_gp10b.h,
              hal/pramin/pramin_gv100.c,