From 1882e38181b5c6a46b4a36ec2bf4f01f8f705254 Mon Sep 17 00:00:00 2001 From: Joseph Schuchart Date: Thu, 4 Apr 2024 10:07:58 -0400 Subject: [PATCH] Pass argc/argv to allocator_init They are needed to initialize madness and thus the TA allocators. Signed-off-by: Joseph Schuchart --- examples/matrixtile.h | 2 +- examples/potrf/testing_dlauum.cc | 2 +- examples/potrf/testing_dpoinv.cc | 2 +- examples/potrf/testing_dpotrf.cc | 2 +- examples/potrf/testing_dtrtri.cc | 2 +- 5 files changed, 5 insertions(+), 5 deletions(-) diff --git a/examples/matrixtile.h b/examples/matrixtile.h index a2080f3ed..d703060f8 100644 --- a/examples/matrixtile.h +++ b/examples/matrixtile.h @@ -14,7 +14,7 @@ #if defined(TILEDARRAY_HAS_DEVICE) #define ALLOCATOR TiledArray::device_pinned_allocator -inline void allocator_init() { +inline void allocator_init(int argc, char **argv) { // initialize MADNESS so that TA allocators can be created #if defined(TTG_PARSEC_IMPORTED) madness::ParsecRuntime::initialize_with_existing_context(ttg::default_execution_context().impl().context()); diff --git a/examples/potrf/testing_dlauum.cc b/examples/potrf/testing_dlauum.cc index df87c63a6..bab6d676c 100644 --- a/examples/potrf/testing_dlauum.cc +++ b/examples/potrf/testing_dlauum.cc @@ -59,7 +59,7 @@ int main(int argc, char **argv) ttg::initialize(argc, argv, nthreads); /* set up TA to get the allocator */ - allocator_init(); + allocator_init(argc, argv); auto world = ttg::default_execution_context(); diff --git a/examples/potrf/testing_dpoinv.cc b/examples/potrf/testing_dpoinv.cc index bd10b24b7..646f2477e 100644 --- a/examples/potrf/testing_dpoinv.cc +++ b/examples/potrf/testing_dpoinv.cc @@ -95,7 +95,7 @@ int main(int argc, char **argv) delete[] ttg_argv; /* set up TA to get the allocator */ - allocator_init(); + allocator_init(argc, argv); ttg::trace_on(); diff --git a/examples/potrf/testing_dpotrf.cc b/examples/potrf/testing_dpotrf.cc index 1658dbed3..6b80bb68c 100644 --- a/examples/potrf/testing_dpotrf.cc +++ b/examples/potrf/testing_dpotrf.cc @@ -65,7 +65,7 @@ int main(int argc, char **argv) ttg::initialize(1, argv, nthreads); /* set up TA to get the allocator */ - allocator_init(); + allocator_init(argc, argv); auto world = ttg::default_execution_context(); if(nullptr != prof_filename) { diff --git a/examples/potrf/testing_dtrtri.cc b/examples/potrf/testing_dtrtri.cc index bebe8cccf..3124ab2be 100644 --- a/examples/potrf/testing_dtrtri.cc +++ b/examples/potrf/testing_dtrtri.cc @@ -75,7 +75,7 @@ int main(int argc, char **argv) ttg::initialize(argc, argv, nthreads); /* set up TA to get the allocator */ - allocator_init(); + allocator_init(argc, argv); auto world = ttg::default_execution_context();