41 lines
1016 B
C
41 lines
1016 B
C
|
#include <dlfcn.h>
|
||
|
|
#include <string.h>
|
||
|
|
|
||
|
|
void *zmlxcuda_dlopen(const char *filename, int flags)
|
||
|
|
{
|
||
|
|
if (filename != NULL)
|
||
|
|
{
|
||
|
|
char *replacements[] = {
|
||
|
|
"libcublas.so",
|
||
|
|
"libcublas.so.12",
|
||
|
|
"libcublasLt.so",
|
||
|
|
"libcublasLt.so.12",
|
||
|
|
"libcudart.so",
|
||
|
|
"libcudart.so.12",
|
||
|
|
"libcudnn.so",
|
||
|
|
"libcudnn.so.9",
|
||
|
|
"libcufft.so",
|
||
|
|
"libcufft.so.11",
|
||
|
|
"libcupti.so",
|
||
|
|
"libcupti.so.12",
|
||
|
|
"libcusolver.so",
|
||
|
|
"libcusolver.so.11",
|
||
|
|
"libcusparse.so",
|
||
|
|
"libcusparse.so.12",
|
||
|
|
"libnccl.so",
|
||
|
|
"libnccl.so.2",
|
||
|
|
NULL,
|
||
|
|
NULL,
|
||
|
|
};
|
||
|
|
for (int i = 0; replacements[i] != NULL; i += 2)
|
||
|
|
{
|
||
|
|
if (strcmp(filename, replacements[i]) == 0)
|
||
|
|
{
|
||
|
|
filename = replacements[i + 1];
|
||
|
|
break;
|
||
|
|
}
|
||
|
|
}
|
||
|
|
}
|
||
|
|
return dlopen(filename, flags);
|
||
|
|
}
|