From 833ff5f28d320e56d65544dcc9b9dfc5a4e29120 Mon Sep 17 00:00:00 2001
From: Tarry Singh <tarry.singh@deepkapha.com>
Date: Wed, 12 Apr 2023 15:47:06 +0000
Subject: [PATCH] Upgrade PJRT CUDA Plugin to version 0.2.3, adding NCCL
 support for correct sharding.

---
 runtimes/cuda/cuda.bzl | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/runtimes/cuda/cuda.bzl b/runtimes/cuda/cuda.bzl
index bf7343b..aa85b8e 100644
--- a/runtimes/cuda/cuda.bzl
+++ b/runtimes/cuda/cuda.bzl
@@ -182,8 +182,8 @@ cc_import(
     http_archive(
         name = "libpjrt_cuda",
         build_file = "libpjrt_cuda.BUILD.bazel",
-        url = "https://github.com/zml/pjrt-artifacts/releases/download/v0.2.2/pjrt-cuda_linux-amd64.tar.gz",
-        sha256 = "45e91e8649bcccc43900f90d6dcbf0cfe87d3d2ee76f1763f41263d2ed44d31b",
+        url = "https://github.com/zml/pjrt-artifacts/releases/download/v0.2.3/pjrt-cuda_linux-amd64.tar.gz",
+        sha256 = "14f39ffef0c9ac529b1a8957750b0b5f5d2f6d310c0d997436051c53a9eb1618",
     )
 
     return mctx.extension_metadata(