Add sm 90 for hugectr (#907)

Author: EmmaQiaoCh

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 && \

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 && \