updated timemory metric tracking around fit #15
Conversation
| #python setup.py install <ARGS> -- <CMAKE_ARGS> | ||
| #python setup.py install --help | ||
| python -m pip install scikit-build | ||
| export CMAKE_CXX_COMPILER=$MPI_CXX |
There was a problem hiding this comment.
Just a side note, this env variable isn't used by CMake. You can use either -DCMAKE_CXX_COMPILER=$MPI_CXX or export CXX=$MPI_CXX. The former will force the compiler to change if there is an existing build and the CXX environment variable is used if there is no existing build and the latter is not specified on the command line.
Also, out of curiosity, is there a particular reason you are using the MPI compiler wrappers here? The find_package for MPI in CMake should take care of that but if you've had build errors, please lmk.
| timemory.settings.flat_profile = False | ||
| timemory.settings.timeline_profile = False | ||
|
|
||
| components = ['wall_clock','peak_rss','read_bytes','written_bytes'] |
There was a problem hiding this comment.
You should add in the 'user_global_bundle' here. It will essentially be zero overhead and it will allow you to add other components on-the-fly without and modifications to the script, e.g. export TIMEMORY_GLOBAL_COMPONENTS="papi_vector, virtual_memory, gperftools_cpu_profiler"
Updated train.py with timemory tracking for communication, I/O, CPU utilization and walltime