OSSForks/tensorflow
1
0
Fork 0
mirror of https://github.com/zebrajr/tensorflow.git synced 2026-01-15 12:15:41 +00:00
Code Issues Actions 5 Packages Projects Releases Wiki Activity

[TSL] Remove some Windows- and Darwinn-specific GPU code.

Browse Source 
PiperOrigin-RevId: 571535260
This commit is contained in:
Christian Sigg
2023-10-07 01:42:02 -07:00
committed by TensorFlower Gardener
parent 333145db59
commit a47d18e3f3
1 changed files with 0 additions and 23 deletions
Download Patch File Download Diff File Expand all files Collapse all files

23
third_party/xla/third_party/tsl/tsl/platform/default/dso_loader.cc vendored
View File

@@ -60,11 +60,6 @@ StatusOr<void*> GetDsoHandle(const string& name, const string& version) {
auto message = absl::StrCat("Could not load dynamic library '", filename,
"'; dlerror: ", status.message());
#if !defined(PLATFORM_WINDOWS)
if (const char* ld_library_path = getenv("LD_LIBRARY_PATH")) {
message += absl::StrCat("; LD_LIBRARY_PATH: ", ld_library_path);
}
#endif
VLOG(1) << message;
return Status(absl::StatusCode::kFailedPrecondition, message);
}
@@ -72,16 +67,6 @@ StatusOr<void*> GetDsoHandle(const string& name, const string& version) {
namespace DsoLoader {
StatusOr<void*> GetCudaDriverDsoHandle() {
#if defined(PLATFORM_WINDOWS)
return GetDsoHandle("nvcuda", "");
#elif defined(__APPLE__)
// On Mac OS X, CUDA sometimes installs libcuda.dylib instead of
// libcuda.1.dylib.
auto handle_or = GetDsoHandle("cuda", "");
if (handle_or.ok()) {
return handle_or;
}
#endif
return GetDsoHandle("cuda", "1");
}
@@ -126,19 +111,11 @@ StatusOr<void*> GetNcclDsoHandle() {
}
StatusOr<void*> GetNvInferDsoHandle() {
#if defined(PLATFORM_WINDOWS)
return GetDsoHandle("nvinfer", "");
#else
return GetDsoHandle("nvinfer", GetTensorRTVersion());
#endif
}
StatusOr<void*> GetNvInferPluginDsoHandle() {
#if defined(PLATFORM_WINDOWS)
return GetDsoHandle("nvinfer_plugin", "");
#else
return GetDsoHandle("nvinfer_plugin", GetTensorRTVersion());
#endif
}
StatusOr<void*> GetRocblasDsoHandle() { return GetDsoHandle("rocblas", ""); }