[PATCH i-g-t] tests/intel/xe_compute_preempt: Add compute-preempt-many

[Nirmoy Das]

On 2/27/2024 5:39 PM, Zbigniew Kempczyński wrote:
> On Fri, Feb 23, 2024 at 08:39:19PM +0100, Nirmoy Das wrote:
>> Add compute-preempt-many which should exercise multiple
>> mid-thread preemption at a time for multiple contexts.
>>
>> Cc: Janga Rahul Kumar <janga.rahul.kumar at intel.com>
>> Signed-off-by: Nirmoy Das <nirmoy.das at intel.com>
>> ---
>>   lib/intel_compute.c              | 24 +++++++++++++++---------
>>   lib/intel_compute.h              |  4 +++-
>>   tests/intel/xe_compute_preempt.c | 27 ++++++++++++++++++++++++---
>>   3 files changed, 42 insertions(+), 13 deletions(-)
>>
>> diff --git a/lib/intel_compute.c b/lib/intel_compute.c
>> index c5d253ebc..8853b5cd3 100644
>> --- a/lib/intel_compute.c
>> +++ b/lib/intel_compute.c
>> @@ -1686,7 +1686,8 @@ static const struct {
>>   };
>>   
>>   static bool __run_intel_compute_kernel_preempt(int fd,
>> -		struct drm_xe_engine_class_instance *eci)
>> +		struct drm_xe_engine_class_instance *eci,
>> +		unsigned int count)
>>   {
>>   	unsigned int ip_ver = intel_graphics_ver(intel_get_drm_devid(fd));
>>   	unsigned int batch;
>> @@ -1719,12 +1720,16 @@ static bool __run_intel_compute_kernel_preempt(int fd,
>>   	if (!kernels->kernel || !kernels->sip_kernel || !kernels->long_kernel)
>>   		return 0;
>>   
>> -	intel_compute_preempt_batches[batch].compute_exec(fd, kernels->long_kernel,
>> -							  kernels->long_kernel_size,
>> -							  kernels->kernel, kernels->size,
>> -							  kernels->sip_kernel,
>> -							  kernels->sip_kernel_size,
>> -							  eci);
>> +	igt_fork(child, count) {
>> +		intel_compute_preempt_batches[batch].compute_exec(fd, kernels->long_kernel,
>> +								  kernels->long_kernel_size,
>> +								  kernels->kernel, kernels->size,
>> +								  kernels->sip_kernel,
>> +								  kernels->sip_kernel_size,
>> +								  eci);
>> +	}
>> +
>> +	igt_waitchildren();
> Generally it will work, but my personal preferences is to not forking
> inside library part. I mean I would fork in the test itself, keeping
> run_intel_compute_kernel_preempt() intact.

Yes, that makes sense.  I will resend.

Thanks,

Nirmoy

>
> --
> Zbigniew
>
>>   
>>   	return true;
>>   }
>> @@ -1737,7 +1742,8 @@ static bool __run_intel_compute_kernel_preempt(int fd,
>>    * Returns true on success, false otherwise.
>>    */
>>   bool run_intel_compute_kernel_preempt(int fd,
>> -		struct drm_xe_engine_class_instance *eci)
>> +		struct drm_xe_engine_class_instance *eci,
>> +		unsigned int count)
>>   {
>> -	return __run_intel_compute_kernel_preempt(fd, eci);
>> +	return __run_intel_compute_kernel_preempt(fd, eci, count);
>>   }
>> diff --git a/lib/intel_compute.h b/lib/intel_compute.h
>> index fe9637b91..430bae562 100644
>> --- a/lib/intel_compute.h
>> +++ b/lib/intel_compute.h
>> @@ -37,5 +37,7 @@ extern const struct intel_compute_kernels intel_compute_square_kernels[];
>>   
>>   bool run_intel_compute_kernel(int fd);
>>   bool xe_run_intel_compute_kernel_on_engine(int fd, struct drm_xe_engine_class_instance *eci);
>> -bool run_intel_compute_kernel_preempt(int fd, struct drm_xe_engine_class_instance *eci);
>> +bool run_intel_compute_kernel_preempt(int fd,
>> +				      struct drm_xe_engine_class_instance *eci,
>> +				      unsigned int count);
>>   #endif	/* INTEL_COMPUTE_H */
>> diff --git a/tests/intel/xe_compute_preempt.c b/tests/intel/xe_compute_preempt.c
>> index a4e0e1454..fd245b655 100644
>> --- a/tests/intel/xe_compute_preempt.c
>> +++ b/tests/intel/xe_compute_preempt.c
>> @@ -22,11 +22,18 @@
>>    * Description:
>>    *      Exercise compute walker mid thread preemption scenario
>>    * Functionality: compute openCL kernel
>> + *
>> + * SUBTEST: compute-preempt-many
>> + * GPU requirement: LNL
>> + * Description:
>> + *      Exercise parallel compute walker mid thread preemption scenario
>> + * Functionality: compute openCL kernel
>>    */
>>   static void
>> -test_compute_preempt(int fd, struct drm_xe_engine_class_instance *hwe)
>> +test_compute_preempt(int fd, struct drm_xe_engine_class_instance *hwe,
>> +		     unsigned int count)
>>   {
>> -	igt_require_f(run_intel_compute_kernel_preempt(fd, hwe), "GPU not supported\n");
>> +	igt_require_f(run_intel_compute_kernel_preempt(fd, hwe, count), "GPU not supported\n");
>>   }
>>   
>>   igt_main
>> @@ -44,7 +51,21 @@ igt_main
>>   				continue;
>>   
>>   			igt_dynamic_f("engine-%s", xe_engine_class_string(hwe->engine_class))
>> -				test_compute_preempt(xe, hwe);
>> +				test_compute_preempt(xe, hwe, 1);
>> +		}
>> +	}
>> +
>> +	igt_subtest_with_dynamic("compute-preempt-many") {
>> +		xe_for_each_engine(xe, hwe) {
>> +			/*
>> +			 * TODO: Render engine is getting timed out when count is high,
>> +			 * so for now only test with compute engine
>> +			 */
>> +			if (hwe->engine_class != DRM_XE_ENGINE_CLASS_COMPUTE)
>> +				continue;
>> +
>> +			igt_dynamic_f("engine-%s", xe_engine_class_string(hwe->engine_class))
>> +				test_compute_preempt(xe, hwe, 100);
>>   		}
>>   	}
>>   
>> -- 
>> 2.42.0
>>