void SetUp()
{
// create PMEM partition
int err = memkind_create_pmem(PMEM_DIR, PMEM_PART_SIZE, &pmem_kind);
ASSERT_EQ(0, err);
ASSERT_TRUE(NULL != pmem_kind);
}
void SetUp()
{
allocator_factory.initialize_allocator(AllocatorTypes::STANDARD_ALLOCATOR);
int err = memkind_create_pmem(PMEM_DIR, PMEM_PART_SIZE, &MEMKIND_PMEM_MOCKUP);
ASSERT_EQ(0, err);
ASSERT_NE(nullptr, MEMKIND_PMEM_MOCKUP);
}
int main(int argc, char *argv[])
{
struct memkind *pmem_kind_unlimited = NULL;
int err = 0;
if (argc > 2) {
fprintf(stderr, "Usage: %s [pmem_kind_dir_path]\n", argv[0]);
return 1;
} else if (argc == 2 && (realpath(argv[1], path) == NULL)) {
fprintf(stderr, "Incorrect pmem_kind_dir_path %s\n", argv[1]);
return 1;
}
fprintf(stdout,
"This example shows how to use multithreading with one main pmem kind."
"\nPMEM kind directory: %s\n", path);
// Create PMEM partition with unlimited size
err = memkind_create_pmem(path, 0, &pmem_kind_unlimited);
if (err) {
print_err_message(err);
return 1;
}
// Create a few threads which will access to our main pmem_kind
pthread_t pmem_threads[NUM_THREADS];
int *pmem_tint[NUM_THREADS][NUM_ALLOCS];
int t = 0, i = 0;
struct arg_struct *args[NUM_THREADS];
for (t = 0; t<NUM_THREADS; t++) {
args[t] = malloc(sizeof(struct arg_struct));
args[t]->id = t;
args[t]->ptr = &pmem_tint[t][0];
args[t]->kind = pmem_kind_unlimited;
if (pthread_create(&pmem_threads[t], NULL, thread_onekind,
(void *)args[t])!= 0) {
fprintf(stderr, "Unable to create a thread.\n");
return 1;
}
}
sleep(1);
if (pthread_cond_broadcast(&cond) != 0) {
fprintf(stderr, "Unable to broadcast a condition.\n");
return 1;
}
for (t = 0; t < NUM_THREADS; t++) {
if (pthread_join(pmem_threads[t], NULL) != 0) {
fprintf(stderr, "Thread join failed.\n");
return 1;
}
}
// Check if we can read the values outside of threads and free resources
for (t = 0; t < NUM_THREADS; t++) {
for (i = 0; i < NUM_ALLOCS; i++) {
if(*pmem_tint[t][i] != t) {
fprintf(stderr,
"pmem_tint value has not been saved correctly in the thread.\n");
return 1;
}
memkind_free(args[t]->kind, *(args[t]->ptr+i));
}
free(args[t]);
}
fprintf(stdout, "Threads successfully allocated memory in the PMEM kind.\n");
return 0;
}
