Log parallelization information.

gputnam · gputnam · commit 8402f9231a9f · 2025-04-16T17:27:30.000-05:00
diff --git a/pytorch/src/TorchService.cxx b/pytorch/src/TorchService.cxx
@@ -5,6 +5,7 @@
 #include "WireCellUtil/Persist.h"
 
 #include <omp.h>
+#include <ATen/Parallel.h>
 
 WIRECELL_FACTORY(TorchService, 
                  WireCell::Pytorch::TorchService,
@@ -30,6 +31,7 @@ Pytorch::TorchService::TorchService()
       log->critical("error interpreting OMP_NUM_THREADS as integer ({})", env_str);
     } // env var not set
   }
+  log->info("TorchService parallel info:\n{}",  at::get_parallel_info());
 }
 
 Configuration Pytorch::TorchService::default_configuration() const

Original file line number	Diff line number	Diff line change
`@@ -5,6 +5,7 @@`
`5`	`5`	`#include "WireCellUtil/Persist.h"`
`6`	`6`
`7`	`7`	`#include <omp.h>`
	`8`	`+#include <ATen/Parallel.h>`
`8`	`9`
`9`	`10`	`WIRECELL_FACTORY(TorchService,`
`10`	`11`	`WireCell::Pytorch::TorchService,`
`@@ -30,6 +31,7 @@ Pytorch::TorchService::TorchService()`
`30`	`31`	`log->critical("error interpreting OMP_NUM_THREADS as integer ({})", env_str);`
`31`	`32`	`} // env var not set`
`32`	`33`	`}`
	`34`	`+ log->info("TorchService parallel info:\n{}", at::get_parallel_info());`
`33`	`35`	`}`
`34`	`36`
`35`	`37`	`Configuration Pytorch::TorchService::default_configuration() const`