update sok to 1.20 (DeepRec-AI#898)

Signed-off-by: aleliu <aleliu@nvidia.com>
BrunoScaglione · Jul 4, 2023 · e810009 · e810009
1 parent 2c90ada
commit e810009
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/tensorflow/tools/pip_package/build_sok.sh b/tensorflow/tools/pip_package/build_sok.sh
@@ -17,4 +17,4 @@ export SOK_COMPILE_GPU_SM="70;75;80"
 cd ./bazel-DeepRec/external/hugectr/sparse_operation_kit
 
 "${PYTHON_BIN_PATH:-python}" setup.py bdist_wheel
-pip install ./dist/merlin_sok-1.1.4-cp38-cp38-linux_x86_64.whl
+pip install ./dist/merlin_sok-1.2.0-cp38-cp38-linux_x86_64.whl
diff --git a/tensorflow/workspace.bzl b/tensorflow/workspace.bzl
@@ -1372,9 +1372,9 @@ def tf_repositories(path_prefix = "", tf_repo_name = ""):
     http_archive(
         name = "hugectr",                                     # Apache License 2.0
         build_file = "//third_party:hugectr.BUILD",
-        strip_prefix = "HugeCTR-23.05.01",
+        strip_prefix = "HugeCTR-23.06.00",
         urls = [
-            "https://github.com/NVIDIA-Merlin/HugeCTR/archive/refs/tags/v23.05.01.tar.gz",
+            "https://github.com/NVIDIA-Merlin/HugeCTR/archive/refs/tags/v23.06.00.tar.gz",
         ],
     )