Make the code reusable.
Signed-off-by: Tao Zhou <tao.zhou1@amd.com>
Reviewed-by: Hawking Zhang <Hawking.Zhang@amd.com>
Signed-off-by: Alex Deucher <alexander.deucher@amd.com>
 
        return 0;
 }
+
+void jpeg_set_ras_funcs(struct amdgpu_device *adev)
+{
+       if (!adev->jpeg.ras)
+               return;
+
+       amdgpu_ras_register_ras_block(adev, &adev->jpeg.ras->ras_block);
+
+       strcpy(adev->jpeg.ras->ras_block.ras_comm.name, "jpeg");
+       adev->jpeg.ras->ras_block.ras_comm.block = AMDGPU_RAS_BLOCK__JPEG;
+       adev->jpeg.ras->ras_block.ras_comm.type = AMDGPU_RAS_ERROR__POISON;
+       adev->jpeg.ras_if = &adev->jpeg.ras->ras_block.ras_comm;
+
+       /* If don't define special ras_late_init function, use default ras_late_init */
+       if (!adev->jpeg.ras->ras_block.ras_late_init)
+               adev->jpeg.ras->ras_block.ras_late_init = amdgpu_ras_block_late_init;
+}
 
 int amdgpu_jpeg_process_poison_irq(struct amdgpu_device *adev,
                                struct amdgpu_irq_src *source,
                                struct amdgpu_iv_entry *entry);
+void jpeg_set_ras_funcs(struct amdgpu_device *adev);
 
 #endif /*__AMDGPU_JPEG_H__*/
 
                break;
        }
 
-       if (adev->jpeg.ras) {
-               amdgpu_ras_register_ras_block(adev, &adev->jpeg.ras->ras_block);
-
-               strcpy(adev->jpeg.ras->ras_block.ras_comm.name, "jpeg");
-               adev->jpeg.ras->ras_block.ras_comm.block = AMDGPU_RAS_BLOCK__JPEG;
-               adev->jpeg.ras->ras_block.ras_comm.type = AMDGPU_RAS_ERROR__POISON;
-               adev->jpeg.ras_if = &adev->jpeg.ras->ras_block.ras_comm;
-
-               /* If don't define special ras_late_init function, use default ras_late_init */
-               if (!adev->jpeg.ras->ras_block.ras_late_init)
-                       adev->jpeg.ras->ras_block.ras_late_init = amdgpu_ras_block_late_init;
-       }
+       jpeg_set_ras_funcs(adev);
 }