diff --git a/third_party/tsl/third_party/gpus/cuda/hermetic/cuda_redist_versions.bzl b/third_party/tsl/third_party/gpus/cuda/hermetic/cuda_redist_versions.bzl
index 88b2c2afd5de7c..6934b75b47852d 100644
--- a/third_party/tsl/third_party/gpus/cuda/hermetic/cuda_redist_versions.bzl
+++ b/third_party/tsl/third_party/gpus/cuda/hermetic/cuda_redist_versions.bzl
@@ -93,6 +93,10 @@ CUDNN_REDIST_JSON_DICT = {
         "https://developer.download.nvidia.com/compute/cudnn/redist/redistrib_9.3.0.json",
         "d17d9a7878365736758550294f03e633a0b023bec879bf173349bfb34781972e",
     ],
+    "9.4.0": [
+        "https://developer.download.nvidia.com/compute/cudnn/redist/redistrib_9.4.0.json",
+        "6eeaafc5cc3d4bb2f283e6298e4c55d4c59d7c83c5d9fd8721a2c0e55aee4e54",
+    ],
 }
 
 # The versions are different for x86 and aarch64 architectures because only
diff --git a/xla/python/ifrt/mock.h b/xla/python/ifrt/mock.h
index 77209214543d14..d929ae00eb0d37 100644
--- a/xla/python/ifrt/mock.h
+++ b/xla/python/ifrt/mock.h
@@ -324,6 +324,12 @@ class MockSharding : public llvm::RTTIExtends<MockSharding, Sharding> {
             BasicDeviceList::Create({}), MemoryKind(),
             /*is_fully_replicated=*/false) {}
 
+  explicit MockSharding(tsl::RCReference<DeviceList> devices,
+                        MemoryKind memory_kind = MemoryKind(),
+                        bool is_fully_replicated = false)
+      : llvm::RTTIExtends<MockSharding, Sharding>(devices, memory_kind,
+                                                  is_fully_replicated) {}
+
   MOCK_METHOD(
       (absl::StatusOr<
           std::vector<std::pair<Shape, std::shared_ptr<const Sharding>>>>),