Libwrap checks for LIB.so.1 if LIB.so not found
Change-Id: I6f07f887f828cb2259dcfd496a2ad707db898cf5
Reviewed-on: http://git-master/r/1000162
Reviewed-by: Przemek Tredak <ptredak@nvidia.com>
Tested-by: Przemek Tredak <ptredak@nvidia.com>
[ROCm/rccl commit: caa40b8dd3]
Tá an tiomantas seo le fáil i:
1 tuismitheoir
16379ec27b
tiomantas
60799f9427
1 comhad athraithe
+10
-4
@@ -59,14 +59,20 @@ ncclResult_t wrapSymbols(void) {
|
||||
|
||||
nvmlhandle=dlopen("libnvidia-ml.so", RTLD_NOW);
|
||||
if (!nvmlhandle) {
|
||||
WARN("Failed to open libnvidia-ml.so");
|
||||
goto teardown;
|
||||
nvmlhandle=dlopen("libnvidia-ml.so.1", RTLD_NOW);
|
||||
if (!nvmlhandle) {
|
||||
WARN("Failed to open libnvidia-ml.so[.1]");
|
||||
goto teardown;
|
||||
}
|
||||
}
|
||||
|
||||
cuhandle = dlopen("libcuda.so", RTLD_NOW);
|
||||
if (!cuhandle) {
|
||||
WARN("Failed to open libcuda.so");
|
||||
goto teardown;
|
||||
cuhandle = dlopen("libcuda.so.1", RTLD_NOW);
|
||||
if (!cuhandle) {
|
||||
WARN("Failed to open libcuda.so[.1]");
|
||||
goto teardown;
|
||||
}
|
||||
}
|
||||
|
||||
#define LOAD_SYM(handle, symbol, funcptr) do { \
|
||||
|
||||
Tagairt in Eagrán Nua
Cuir bac ar úsáideoir