2 files changed, 2 insertions, 6 deletions

diff --git a/drivers/gpu/nvgpu/common/bus.c b/drivers/gpu/nvgpu/common/bus.c
index c88262a0..a84348cd 100644
--- a/drivers/gpu/nvgpu/common/bus.c
+++ b/drivers/gpu/nvgpu/common/bus.c
@@ -31,11 +31,6 @@ int nvgpu_get_timestamps_zipper(struct gk20a *g,
         int err = 0;
         unsigned int i = 0;
 
-        if (source_id != NVGPU_GPU_GET_CPU_TIME_CORRELATION_INFO_SRC_ID_TSC) {
-                nvgpu_err(g, "source_id %u not supported", source_id);
-                return -EINVAL;
-        }
-
         if (gk20a_busy(g)) {
                 nvgpu_err(g, "GPU not powered on\n");
                 err = -EINVAL;
diff --git a/drivers/gpu/nvgpu/common/linux/ioctl_ctrl.c b/drivers/gpu/nvgpu/common/linux/ioctl_ctrl.c
index 30988462..88c8debf 100644
--- a/drivers/gpu/nvgpu/common/linux/ioctl_ctrl.c
+++ b/drivers/gpu/nvgpu/common/linux/ioctl_ctrl.c
@@ -558,7 +558,8 @@ static int nvgpu_gpu_get_cpu_time_correlation_info(
         int err;
         u32 i;
 
-        if (args->count > NVGPU_GPU_GET_CPU_TIME_CORRELATION_INFO_MAX_COUNT)
+        if (args->count > NVGPU_GPU_GET_CPU_TIME_CORRELATION_INFO_MAX_COUNT ||
+            args->source_id != NVGPU_GPU_GET_CPU_TIME_CORRELATION_INFO_SRC_ID_TSC)
                 return -EINVAL;
 
         samples = nvgpu_kzalloc(g, args->count *