Update

Update.
microsoft · Jun 13, 2022 · 8481e1c · 8481e1c
1 parent b847903
commit 8481e1c
Show file tree

Hide file tree

Showing 6 changed files with 6 additions and 1 deletion.
diff --git a/dockerfile/rocm4.0-pytorch1.7.0.dockerfile b/dockerfile/rocm4.0-pytorch1.7.0.dockerfile
@@ -28,6 +28,7 @@ RUN wget -qO - http://repo.radeon.com/rocm/rocm.gpg.key | APT_KEY_DONT_WARN_ON_D
     git \
     jq \
     libaio-dev \
+    libboost-program-options-dev \
     libcap2 \
     libnuma-dev \
     libpci-dev \

diff --git a/dockerfile/rocm4.2-pytorch1.7.0.dockerfile b/dockerfile/rocm4.2-pytorch1.7.0.dockerfile
@@ -28,6 +28,7 @@ RUN wget -qO - http://repo.radeon.com/rocm/rocm.gpg.key | APT_KEY_DONT_WARN_ON_D
     git \
     jq \
     libaio-dev \
+    libboost-program-options-dev \
     libcap2 \
     libnuma-dev \
     libpci-dev \

diff --git a/dockerfile/rocm5.0-pytorch1.9.0.dockerfile b/dockerfile/rocm5.0-pytorch1.9.0.dockerfile
@@ -28,6 +28,7 @@ RUN apt-get update && \
     hipify-clang \
     jq \
     libaio-dev \
+    libboost-program-options-dev \
     libcap2 \
     libpci-dev \
     libtinfo5 \

diff --git a/dockerfile/rocm5.0.1-pytorch1.9.0.dockerfile b/dockerfile/rocm5.0.1-pytorch1.9.0.dockerfile
@@ -28,6 +28,7 @@ RUN apt-get update && \
     hipify-clang \
     jq \
     libaio-dev \
+    libboost-program-options-dev \
     libcap2 \
     libpci-dev \
     libtinfo5 \

diff --git a/dockerfile/rocm5.1.1-pytorch1.10.0.dockerfile b/dockerfile/rocm5.1.1-pytorch1.10.0.dockerfile
@@ -28,6 +28,7 @@ RUN apt-get update && \
     hipify-clang \
     jq \
     libaio-dev \
+    libboost-program-options-dev \
     libcap2 \
     libpci-dev \
     libtinfo5 \

diff --git a/third_party/Makefile b/third_party/Makefile
@@ -27,7 +27,7 @@ cuda_cutlass:
 ifneq (,$(wildcard cutlass/CMakeLists.txt))
 	cmake -DCMAKE_INSTALL_BINDIR=$(SB_MICRO_PATH)/bin -DCMAKE_INSTALL_LIBDIR=$(SB_MICRO_PATH)/lib -DCMAKE_BUILD_TYPE=Release \
 		-DCUTLASS_NVCC_ARCHS='70;75;80;86' -DCUTLASS_ENABLE_EXAMPLES=OFF -DCUTLASS_ENABLE_TESTS=OFF -S ./cutlass -B ./cutlass/build
-	cmake --build ./cutlass/build -j --target install
+	cmake --build ./cutlass/build -j $(nproc --ignore=2) --target install
 endif
 
 # Build cuda-samples/Samples/bandwidthTest.