Add sm 90 for hugectr (#907)

NVIDIA-Merlin · Apr 26, 2023 · a51efde · a51efde
1 parent 94c038b
commit a51efde
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 5 deletions.
diff --git a/docker/dockerfile.ctr b/docker/dockerfile.ctr
@@ -76,9 +76,9 @@ RUN if [[ "${HUGECTR_DEV_MODE}" == "false" ]]; then \
         LD_LIBRARY_PATH=/usr/local/cuda/lib64/stubs/:$LD_LIBRARY_PATH && \
         export PATH=$PATH:/usr/local/cuda-${CUDA_SHORT_VERSION}/compat && \
         if [[ "${INSTALL_HDFS}" == "false" ]]; then \
-            cmake -DCMAKE_BUILD_TYPE=Release -DSM="60;61;70;75;80" -DENABLE_MULTINODES=ON .. \
+            cmake -DCMAKE_BUILD_TYPE=Release -DSM="60;61;70;75;80;90" -DENABLE_MULTINODES=ON .. \
         ; else \
-            cmake -DCMAKE_BUILD_TYPE=Release -DSM="60;61;70;75;80" -DENABLE_MULTINODES=ON -DENABLE_HDFS=ON .. \
+            cmake -DCMAKE_BUILD_TYPE=Release -DSM="60;61;70;75;80;90" -DENABLE_MULTINODES=ON -DENABLE_HDFS=ON .. \
         ; fi && \
         make -j$(nproc) && \
         make install && \

diff --git a/docker/dockerfile.merlin b/docker/dockerfile.merlin
@@ -379,17 +379,17 @@ RUN if [ "${HUGECTR_DEV_MODE}" == "false" ]; then \
         mkdir build && \
         cd build && \
         if [[ "${INSTALL_HDFS}" == "false" ]]; then \
-            cmake -DCMAKE_BUILD_TYPE=Release -DSM="70;75;80" -DENABLE_INFERENCE=ON .. \
+            cmake -DCMAKE_BUILD_TYPE=Release -DSM="70;75;80;90" -DENABLE_INFERENCE=ON .. \
         ; else \
-            cmake -DCMAKE_BUILD_TYPE=Release -DSM="70;75;80" -DENABLE_INFERENCE=ON -DENABLE_HDFS=ON .. \
+            cmake -DCMAKE_BUILD_TYPE=Release -DSM="70;75;80;90" -DENABLE_INFERENCE=ON -DENABLE_HDFS=ON .. \
         ; fi && \
         make -j$(nproc) && \
         make install && \
         # Install HPS trt pugin
         cd ../hps_trt && \
         mkdir build && \
         cd build && \
-        cmake -DSM="70;75;80" .. && \
+        cmake -DSM="70;75;80;90" .. && \
         make -j$(nproc) && \
         make install && \
         cd / && rm -rf /hugectr && \