Pass argc/argv to allocator_init

They are needed to initialize madness and thus the TA allocators. Signed-off-by: Joseph Schuchart <[email protected]>
TESSEorg · Apr 4, 2024 · 1882e38 · 1882e38
1 parent ae62f5b
commit 1882e38
Show file tree

Hide file tree

Showing 5 changed files with 5 additions and 5 deletions.
diff --git a/examples/matrixtile.h b/examples/matrixtile.h
@@ -14,7 +14,7 @@
 #if defined(TILEDARRAY_HAS_DEVICE)
 #define ALLOCATOR TiledArray::device_pinned_allocator<T>
 
-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
@@ -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
@@ -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
@@ -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
@@ -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();