message: add sm60 on ci
diff: @@ -83,7 +83,7 @@ build_train:
  fi
  - export JOB_DOCKERFILE="train.Dockerfile.${CI_JOB_NAME%%--*}.${CI_PIPELINE_ID}" && echo ${JOB_DOCKERFILE}
  - echo "FROM ${IMAGE_TRAIN}" > ${JOB_DOCKERFILE}
  - - echo "ARG SM=\"70;75;80\"" >> ${JOB_DOCKERFILE}
  + - echo "ARG SM=\"60;61;70;75;80\"" >> ${JOB_DOCKERFILE}
  - echo "
  RUN git clone https://gitlab-ci-token:${CI_JOB_TOKEN}@gitlab-master.nvidia.com/dl/hugectr/hugectr.git ${WORK_DIR} && cd ${WORK_DIR} && git checkout ${CI_COMMIT_SHA} && git submodule update --init --recursive && mkdir build && cd build && mkdir build_single && cd build_single && cmake -DCMAKE_BUILD_TYPE=Release -DKEY_HIT_RATIO=ON -DSM=\$SM ../.. && make -j\$(nproc) && cd .. && mkdir build_multi_nccl && cd build_multi_nccl && cmake -DCMAKE_BUILD_TYPE=Release -DENABLE_MULTINODES=ON -DKEY_HIT_RATIO=ON -DSM=\$SM ../.. && make -j\$(nproc) && make -j\$(nproc) && cd .. && mkdir build_inference && cd build_inference && cmake -DENABLE_INFERENCE=ON -DCMAKE_BUILD_TYPE=Release -DSM=\$SM ../.. && make -j\$(nproc)
  " >> ${JOB_DOCKERFILE}
  @@ -109,7 +109,7 @@ build_embedding:
  fi
  - export JOB_DOCKERFILE="embedding.Dockerfile.${CI_JOB_NAME%%--*}.${CI_PIPELINE_ID}" && echo ${JOB_DOCKERFILE}
  - echo "FROM ${IMAGE_EMBEDDING}" > ${JOB_DOCKERFILE}
  - - echo "ARG SM=\"70;75;80\"" >> ${JOB_DOCKERFILE}
  + - echo "ARG SM=\"60;61;70;75;80\"" >> ${JOB_DOCKERFILE}
  - echo "
  RUN git clone https://gitlab-ci-token:${CI_JOB_TOKEN}@gitlab-master.nvidia.com/dl/hugectr/hugectr.git ${WORK_DIR} && cd ${WORK_DIR} && git checkout ${CI_COMMIT_SHA} && git submodule update --init --recursive && cd sparse_operation_kit/Deprecated && mkdir build && cd build && cmake -DCMAKE_BUILD_TYPE=Release -DSM=\$SM .. && make -j\$(nproc) && make install
  " >> ${JOB_DOCKERFILE}
  @@ -833,7 +833,7 @@ sample_dlrm_selene:
  --container-mounts ${CI_PROJECT_DIR}:${CI_PROJECT_DIR}
  bash -cx "
  cd ${CI_PROJECT_DIR};
  - mkdir build && cd build && cmake -DCMAKE_BUILD_TYPE=Release -DSM=70\;75\;80 .. && make -j;
  + mkdir build && cd build && cmake -DCMAKE_BUILD_TYPE=Release -DSM=60\;61\;70\;75\;80 .. && make -j;
  ./bin/embedding_test
  "
  timeout: 2 hours
  