Remove early dso check

Author: trevor-m

xla/backends/gpu/collectives/BUILD

@@ -292,7 +292,6 @@ cc_library(
         "@com_google_absl//absl/types:span",
         "@nvshmem//:nvshmem_lib",
         "@tsl//tsl/platform:casts",
-        "@tsl//tsl/platform:dso_loader",
         "@tsl//tsl/platform:numbers",
     ],
     alwayslink = True,  # registers collectives implementation
@@ -358,7 +357,7 @@ xla_cc_test(
         "//xla/pjrt/distributed:client",
         "//xla/pjrt/distributed:service",
         "//xla/pjrt/gpu:se_gpu_pjrt_client",
-	    "//xla/pjrt/gpu:se_gpu_pjrt_compiler",
+        "//xla/pjrt/gpu:se_gpu_pjrt_compiler",
         "//xla/tests:literal_test_util",
         "//xla/tsl/platform:errors",
         "//xla/tsl/platform:status",

xla/backends/gpu/collectives/nvshmem_collectives.cc

@@ -36,7 +36,6 @@ limitations under the License.
 #include "xla/tsl/platform/errors.h"
 #include "xla/tsl/platform/statusor.h"
 #include "tsl/platform/casts.h"
-#include "tsl/platform/dso_loader.h"
 #include "tsl/platform/numbers.h"
 
 namespace xla::gpu {
@@ -78,11 +77,6 @@ absl::Status NvshmemCollectives::InitializeOnce() {
       LOG(FATAL) << "NVSHMEM API is only supported with one device per process";
     }
 
-    // First, check that DSO can be found. Otherwise if it cannot be found,
-    // non-leader ranks will be stuck at kv_store->Get().
-    TF_RETURN_IF_ERROR(
-        tsl::internal::DsoLoader::GetNvshmemDsoHandle().status());
-
     nvshmemx_init_attr_t nvshmem_init_attr = NVSHMEMX_INIT_ATTR_INITIALIZER;
     nvshmemx_uniqueid_t nvshmem_id = NVSHMEMX_UNIQUEID_INITIALIZER;