Test: CI build optimizations (CUDA compute capabilities and non-curses mode from cl/945255455)

.bazelrc

@@ -17,12 +17,25 @@ build:cuda_clang --host_copt=-Wno-error=unused-command-line-argument
 build:cuda_clang --repo_env TF_NEED_TENSORRT=0
 build:cuda_clang --action_env=TF_CUDA_CLANG="1"
 build:cuda_clang --@local_config_cuda//:cuda_compiler=clang
-build:cuda_clang --repo_env=TF_CUDA_COMPUTE_CAPABILITIES="sm_60,sm_70,sm_80,sm_89,compute_90"
+build:cuda_clang --repo_env=TF_CUDA_COMPUTE_CAPABILITIES="sm_60,sm_70,sm_75,sm_80,sm_86,sm_89,compute_90"
+build:cuda_clang --copt=-Dtypeof=__typeof__
 
-common:cuda_clang --repo_env=HERMETIC_CUDA_VERSION="12.5.1"
+common:cuda_clang --repo_env=HERMETIC_CUDA_VERSION="12.2.0"
 common:cuda_clang --repo_env=HERMETIC_CUDNN_VERSION="9.3.0"
 common:cuda_clang --repo_env=HERMETIC_NVSHMEM_VERSION="3.2.5"
-common:cuda_clang --repo_env=HERMETIC_NCCL_VERSION="2.29.7"
+common:cuda_clang --repo_env=HERMETIC_NCCL_VERSION="2.28.9"
+
+common:cuda --repo_env=HERMETIC_CUDA_VERSION="12.2.0"
+common:cuda --repo_env=HERMETIC_CUDNN_VERSION="9.3.0"
+common:cuda --repo_env=HERMETIC_NVSHMEM_VERSION="3.2.5"
+common:cuda --repo_env=HERMETIC_NCCL_VERSION="2.28.9"
+build:cuda --features=-layering_check
+build:cuda --copt=-w
+build:cuda --cxxopt=-w
+build:cuda --copt=-Wno-error
+build:cuda --cxxopt=-Wno-error
+build:cuda --copt=-Wno-header-guard
+build:cuda --cxxopt=-Wno-header-guard
 build:cuda_clang --action_env=GCC_HOST_COMPILER_PATH="/usr/bin/gcc-10"
 build:cuda_clang --action_env=CLANG_CUDA_COMPILER_PATH="/usr/lib/llvm-17/bin/clang"
 
@@ -116,6 +129,9 @@ common --repo_env USE_HERMETIC_CC_TOOLCHAIN=1
 common --repo_env=HERMETIC_PYTHON_VERSION="3.10"
 common --repo_env=USE_PYWRAP_RULES=1
 build --features=-layering_check
+build --host_features=-layering_check
+build --features=-module_maps
+build --host_features=-module_maps
 build --define=framework_shared_object=false
 build --dynamic_mode=off
 

WORKSPACE

@@ -24,8 +24,8 @@ local_repository(
 load("//tensorflow_serving:repo.bzl", "tensorflow_http_archive")
 tensorflow_http_archive(
     name = "org_tensorflow",
-    sha256 = "54bfd56d61dfced2b82bdeee3262a24fb3d59bed4dbb292d19834af002d85b3b",
-    git_commit = "c1be979f3130b85f14e8890c05cbca147784f474",
+    sha256 = "6438396f3b19af5d7ad787cf041f857af7505916dc08092e20b07d1b1f8df492",
+    git_commit = "a481b10260dfdf833a1b16007eead49c1d7febf3",
     patch = "//third_party/tensorflow:tensorflow.patch",
     patch_cmds = [
         "sed -i '/cc_library = _cc_library/d' tensorflow/core/platform/rules_cc.bzl",
@@ -95,17 +95,16 @@ http_archive(
 
 http_archive(
     name = "rules_python",
-    sha256 = "8964aa1e7525fea5244ba737458694a057ada1be96a92998a41caa1983562d00",
-    strip_prefix = "rules_python-1.8.5",
+    sha256 = "fa7dd2c6b7d63b3585028dd8a90a6cf9db83c33b250959c2ee7b583a6c130e12",
+    strip_prefix = "rules_python-1.6.0",
     urls = [
-        "https://storage.googleapis.com/mirror.tensorflow.org/github.com/bazelbuild/rules_python/releases/download/1.8.5/rules_python-1.8.5.tar.gz",
-        "https://github.com/bazelbuild/rules_python/releases/download/1.8.5/rules_python-1.8.5.tar.gz",
+        "https://storage.googleapis.com/mirror.tensorflow.org/github.com/bazelbuild/rules_python/releases/download/1.6.0/rules_python-1.6.0.tar.gz",
+        "https://github.com/bazelbuild/rules_python/releases/download/1.6.0/rules_python-1.6.0.tar.gz",
     ],
     patches = [
-        "@rules_ml_toolchain//third_party/rules_python:rules_python_scope.patch",
-        "@rules_ml_toolchain//third_party/rules_python:rules_python_freethreaded.patch",
-        "@rules_ml_toolchain//third_party/rules_python:rules_python_versions.patch",
-        "@rules_ml_toolchain//third_party/rules_python:rules_python_pip_version.patch",
+        "@org_tensorflow//third_party/py:rules_python_pip_version.patch",
+        "@org_tensorflow//third_party/py:rules_python_freethreaded.patch",
+        "@org_tensorflow//third_party/py:rules_python_versions.patch",
     ],
     patch_args = ["-p1"],
 )
@@ -114,11 +113,15 @@ http_archive(
 # Details: https://github.com/google-ml-infra/rules_ml_toolchain
 http_archive(
     name = "rules_ml_toolchain",
-    sha256 = "0b42f693a60c6050d87db1e0a0eaeb84ab3f54191fce094d86334faedc807da0",
-    strip_prefix = "rules_ml_toolchain-398d613aea7a4c294da49b79a6d6f3f8732bd84c",
+    sha256 = "54c1a357f71f611efdb4891ebd4bcbe4aeb6dfa7e473f14fd7ecad5062096616",
+    strip_prefix = "rules_ml_toolchain-d8cb9c2c168cd64000eaa6eda0781a9615a26ffe",
     urls = [
-        "https://storage.googleapis.com/mirror.tensorflow.org/github.com/google-ml-infra/rules_ml_toolchain/archive/398d613aea7a4c294da49b79a6d6f3f8732bd84c.tar.gz",
-        "https://github.com/google-ml-infra/rules_ml_toolchain/archive/398d613aea7a4c294da49b79a6d6f3f8732bd84c.tar.gz",
+        "https://storage.googleapis.com/mirror.tensorflow.org/github.com/google-ml-infra/rules_ml_toolchain/archive/d8cb9c2c168cd64000eaa6eda0781a9615a26ffe.tar.gz",
+        "https://github.com/google-ml-infra/rules_ml_toolchain/archive/d8cb9c2c168cd64000eaa6eda0781a9615a26ffe.tar.gz",
+    ],
+    patch_cmds = [
+        "mkdir -p third_party/rules_python py && touch third_party/rules_python/BUILD py/BUILD",
+        "cat << 'EOF' >> cc/cuda/clang/clang_cuda_runtime_wrapper.h\n#if defined(__clang__) && defined(__CUDA__)\n#include <vector_types.h>\n#ifndef typeof\n#define typeof __typeof__\n#endif\n#ifndef DISABLE_WRAPPERS_STWT\n__device__ inline void __stwt(uint4* ptr, uint4 value) {\n  asm volatile(\"st.global.wt.v4.u32 [%0], {%1, %2, %3, %4};\" : : \"l\"(ptr), \"r\"(value.x), \"r\"(value.y), \"r\"(value.z), \"r\"(value.w) : \"memory\");\n}\n#endif\n#endif\nEOF",
     ],
 )
 
@@ -146,7 +149,6 @@ python_init_repositories(
         "3.11": "@org_tensorflow//:requirements_lock_3_11.txt",
         "3.12": "@org_tensorflow//:requirements_lock_3_12.txt",
         "3.13": "@org_tensorflow//:requirements_lock_3_13.txt",
-        "3.14": "@org_tensorflow//:requirements_lock_3_14.txt",
     },
 )
 

bazel-tmp_git_fork

@@ -0,0 +1 @@
+/usr/local/google/home/vkarampudi/.cache/bazel/_bazel_vkarampudi/cd19753e2603678f5b4c97f3e42118ca/execroot/tf_serving

kokoro_local_root/src/github/serving

@@ -0,0 +1 @@
+/google/src/cloud/vkarampudi/serving/google3/tmp_git_fork