Pytorch デバッグモードでコンパイルする

デバッグモードでコンパイルする

pytorch/tools/setup_helpers/cmake.py に下記を追加。（ついでにCMAKE_VERBOSE_MAKEFILEも追加しています）

--- a/pytorch/tools/setup_helpers/cmake.py
+++ b/pytorch/tools/setup_helpers/cmake.py
@@ -305,6 +305,9 @@ class CMake:
             sys.exit(1)
         build_options.update(cmake__options)
 
+        build_options["CMAKE_BUILD_TYPE"] = os.getenv("CMAKE_BUILD_TYPE", "Release")
+        build_options["CMAKE_VERBOSE_MAKEFILE"] = os.getenv("CMAKE_VERBOSE_MAKEFILE", "0")
+
         CMake.defines(
             args,
             PYTHON_EXECUTABLE=sys.executable,

コンパイル時にCMAKE_BUILD_TYPE=Debugをつければok。例えば次のようにする。

CMAKE_BUILD_TYPE=Debug BUILD_SPLIT_CUDA=ON USE_GLOG=ON USE_NUMPY=ON USE_CUDA=ON USE_MPI=ON CMAKE_C_COMPILER=$(which mpicc) CMAKE_CXX_COMPILER=$(which mpicxx) python setup.py build develop install

関連