From 7a56f2e5e11aa3941d365988355fcb3366626b05 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 | 4 ++-- 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, 6 insertions(+), 6 deletions(-) diff --git a/examples/matrixtile.h b/examples/matrixtile.h index a2080f3ed..a759b1c3c 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()); @@ -28,7 +28,7 @@ inline void allocator_fini() { #else // TILEDARRAY_HAS_DEVICE #define ALLOCATOR std::allocator -inline void allocator_init() { } +inline void allocator_init(int argc, char **argv) { } inline void allocator_fini() { } 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();