mirror of
https://github.com/torvalds/linux.git
synced 2026-06-03 20:14:06 +02:00
perf test workload: Add thread count argument to thloop
Allow the number of threads for the thloop workload to be increased beyond the normal 2. Add error checking to the parsed time and thread count values. Signed-off-by: Ian Rogers <irogers@google.com> Acked-by: Arnaldo Carvalho de Melo <acme@redhat.com> Signed-off-by: Namhyung Kim <namhyung@kernel.org>
This commit is contained in:
parent
2fee899c06
commit
3528647874
|
|
@ -31,21 +31,52 @@ static void *thfunc(void *arg)
|
|||
|
||||
static int thloop(int argc, const char **argv)
|
||||
{
|
||||
int sec = 1;
|
||||
pthread_t th;
|
||||
int nt = 2, sec = 1, err = 1;
|
||||
pthread_t *thread_list = NULL;
|
||||
|
||||
if (argc > 0)
|
||||
sec = atoi(argv[0]);
|
||||
|
||||
if (sec <= 0) {
|
||||
fprintf(stderr, "Error: seconds (%d) must be >= 1\n", sec);
|
||||
return 1;
|
||||
}
|
||||
|
||||
if (argc > 1)
|
||||
nt = atoi(argv[1]);
|
||||
|
||||
if (nt <= 0) {
|
||||
fprintf(stderr, "Error: thread count (%d) must be >= 1\n", nt);
|
||||
return 1;
|
||||
}
|
||||
|
||||
signal(SIGINT, sighandler);
|
||||
signal(SIGALRM, sighandler);
|
||||
|
||||
thread_list = calloc(nt, sizeof(pthread_t));
|
||||
if (thread_list == NULL) {
|
||||
fprintf(stderr, "Error: malloc failed for %d threads\n", nt);
|
||||
goto out;
|
||||
}
|
||||
for (int i = 1; i < nt; i++) {
|
||||
int ret = pthread_create(&thread_list[i], NULL, thfunc, test_loop);
|
||||
|
||||
if (ret) {
|
||||
fprintf(stderr, "Error: failed to create thread %d\n", i);
|
||||
done = 1; // Ensure started threads terminate.
|
||||
goto out;
|
||||
}
|
||||
}
|
||||
alarm(sec);
|
||||
|
||||
pthread_create(&th, NULL, thfunc, test_loop);
|
||||
test_loop();
|
||||
pthread_join(th, NULL);
|
||||
|
||||
return 0;
|
||||
err = 0;
|
||||
out:
|
||||
for (int i = 1; i < nt; i++) {
|
||||
if (thread_list && thread_list[i])
|
||||
pthread_join(thread_list[i], /*retval=*/NULL);
|
||||
}
|
||||
free(thread_list);
|
||||
return err;
|
||||
}
|
||||
|
||||
DEFINE_WORKLOAD(thloop);
|
||||
|
|
|
|||
Loading…
Reference in New Issue
Block a user