Merge pull request #280 from devreal/fix_potrf_allocator_init

Pass argc/argv to allocator_init
TESSEorg · Jun 3, 2024 · d083245 · d083245
2 parents 8a00cff + 7a56f2e
commit d083245
Show file tree

Hide file tree

Showing 5 changed files with 6 additions and 6 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());
@@ -28,7 +28,7 @@ inline void allocator_fini() {
 #else  // TILEDARRAY_HAS_DEVICE
 #define ALLOCATOR std::allocator<T>
 
-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
@@ -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();