From fce1d920293cadad1a69dffd1548bcf485c9d7d9 Mon Sep 17 00:00:00 2001
From: "xiyu.zk" <xiyu.zk@alibaba-inc.com>
Date: Wed, 13 Mar 2024 09:05:31 +0800
Subject: [PATCH] [GLUTEN-4903][CELEBORN] Support multiple versions of Celeborn

---
 .../spark/shuffle/gluten/celeborn/CelebornShuffleManager.java    | 1 +
 1 file changed, 1 insertion(+)

diff --git a/gluten-celeborn/common/src/main/java/org/apache/spark/shuffle/gluten/celeborn/CelebornShuffleManager.java b/gluten-celeborn/common/src/main/java/org/apache/spark/shuffle/gluten/celeborn/CelebornShuffleManager.java
index a762fd5e945b7..c2420583785bb 100644
--- a/gluten-celeborn/common/src/main/java/org/apache/spark/shuffle/gluten/celeborn/CelebornShuffleManager.java
+++ b/gluten-celeborn/common/src/main/java/org/apache/spark/shuffle/gluten/celeborn/CelebornShuffleManager.java
@@ -299,6 +299,7 @@ public <K, V> ShuffleWriter<K, V> getWriter(
         byte[] extension;
         try {
           Field field = CelebornShuffleHandle.class.getDeclaredField("extension");
+          field.setAccessible(true);
           extension = (byte[]) field.get(handle);
 
         } catch (NoSuchFieldException e) {