|
|
|
@ -36,8 +36,6 @@ DEFINE_string(cuda_dir, "",
|
|
|
|
|
|
|
|
|
|
DEFINE_string(warpctc_dir, "", "Specify path for loading libwarpctc.so.");
|
|
|
|
|
|
|
|
|
|
DEFINE_string(lapack_dir, "", "Specify path for loading liblapack.so.");
|
|
|
|
|
|
|
|
|
|
DEFINE_string(nccl_dir, "",
|
|
|
|
|
"Specify path for loading nccl library, such as libcublas, "
|
|
|
|
|
"libcurand. For instance, /usr/local/cuda/lib64. If default, "
|
|
|
|
@ -189,14 +187,6 @@ void* GetWarpCTCDsoHandle() {
|
|
|
|
|
#endif
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
void* GetLapackDsoHandle() {
|
|
|
|
|
#if defined(__APPLE__) || defined(__OSX__)
|
|
|
|
|
return GetDsoHandleFromSearchPath(FLAGS_lapack_dir, "liblapacke.dylib");
|
|
|
|
|
#else
|
|
|
|
|
return GetDsoHandleFromSearchPath(FLAGS_lapack_dir, "liblapacke.so");
|
|
|
|
|
#endif
|
|
|
|
|
}
|
|
|
|
|
|
|
|
|
|
void* GetNCCLDsoHandle() {
|
|
|
|
|
#if defined(__APPLE__) || defined(__OSX__)
|
|
|
|
|
return GetDsoHandleFromSearchPath(FLAGS_nccl_dir, "libnccl.dylib");
|
|
|
|
|