[PATCH v4 20/24] drm/amdkfd: enable pc sampling start

[James Zhu]

Enable pc sampling start.

Signed-off-by: James Zhu <james....@amd.com>
---
 drivers/gpu/drm/amd/amdkfd/kfd_pc_sampling.c | 27 +++++++++++++++++---
 1 file changed, 24 insertions(+), 3 deletions(-)

diff --git a/drivers/gpu/drm/amd/amdkfd/kfd_pc_sampling.c 
b/drivers/gpu/drm/amd/amdkfd/kfd_pc_sampling.c
index df2f4bfd0cda..6f50ba1f8989 100644
--- a/drivers/gpu/drm/amd/amdkfd/kfd_pc_sampling.c
+++ b/drivers/gpu/drm/amd/amdkfd/kfd_pc_sampling.c
@@ -95,9 +95,30 @@ static int kfd_pc_sample_query_cap(struct kfd_process_device 
*pdd,
        return 0;
 }
 
-static int kfd_pc_sample_start(struct kfd_process_device *pdd)
+static int kfd_pc_sample_start(struct kfd_process_device *pdd,
+                                       struct pc_sampling_entry *pcs_entry)
 {
-       return -EINVAL;
+       bool pc_sampling_start = false;
+
+       pcs_entry->enabled = true;
+       mutex_lock(&pdd->dev->pcs_data.mutex);
+
+       kfd_process_set_trap_pc_sampling_flag(&pdd->qpd,
+               pdd->dev->pcs_data.hosttrap_entry.base.pc_sample_info.method, 
true);
+
+       if (!pdd->dev->pcs_data.hosttrap_entry.base.active_count)
+               pc_sampling_start = true;
+       pdd->dev->pcs_data.hosttrap_entry.base.active_count++;
+       mutex_unlock(&pdd->dev->pcs_data.mutex);
+
+       while (pc_sampling_start) {
+               if 
(READ_ONCE(pdd->dev->pcs_data.hosttrap_entry.base.stop_enable))
+                       usleep_range(1000, 2000);
+               else
+                       break;
+       }
+
+       return 0;
 }
 
 static int kfd_pc_sample_stop(struct kfd_process_device *pdd,
@@ -269,7 +290,7 @@ int kfd_pc_sample(struct kfd_process_device *pdd,
                if (pcs_entry->enabled)
                        return -EALREADY;
                else
-                       return kfd_pc_sample_start(pdd);
+                       return kfd_pc_sample_start(pdd, pcs_entry);
 
        case KFD_IOCTL_PCS_OP_STOP:
                if (!pcs_entry->enabled)
-- 
2.25.1