mirror of
https://github.com/bytecodealliance/wasm-micro-runtime.git
synced 2024-10-16 13:02:32 +00:00
216dc43ab4
Multiple threads generated from the same module should use the same lock to protect the atomic operations. Before this PR, each thread used a different lock to protect atomic operations (e.g. atomic add), making the lock ineffective. Fix #1958.
48 lines
1.0 KiB
C
48 lines
1.0 KiB
C
/*
|
|
* Copyright (C) 2023 Amazon.com Inc. or its affiliates. All rights reserved.
|
|
* SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
|
|
*/
|
|
|
|
#include <stdio.h>
|
|
#include <pthread.h>
|
|
|
|
#define MAX_NUM_THREADS 4
|
|
#define NUM_ITER 1000
|
|
|
|
int g_count = 0;
|
|
|
|
static void *
|
|
thread(void *arg)
|
|
{
|
|
for (int i = 0; i < NUM_ITER; i++) {
|
|
__atomic_fetch_add(&g_count, 1, __ATOMIC_SEQ_CST);
|
|
}
|
|
|
|
return NULL;
|
|
}
|
|
|
|
int
|
|
main(int argc, char **argv)
|
|
{
|
|
pthread_t tids[MAX_NUM_THREADS];
|
|
|
|
for (int i = 0; i < MAX_NUM_THREADS; i++) {
|
|
if (pthread_create(&tids[i], NULL, thread, NULL) != 0) {
|
|
printf("Thread creation failed\n");
|
|
}
|
|
}
|
|
|
|
for (int i = 0; i < MAX_NUM_THREADS; i++) {
|
|
if (pthread_join(tids[i], NULL) != 0) {
|
|
printf("Thread join failed\n");
|
|
}
|
|
}
|
|
|
|
printf("Value of counter after update: %d (expected=%d)\n", g_count,
|
|
MAX_NUM_THREADS * NUM_ITER);
|
|
if (g_count != MAX_NUM_THREADS * NUM_ITER) {
|
|
__builtin_trap();
|
|
}
|
|
|
|
return -1;
|
|
} |