cublas.cc 339 字节
Newer Older
Y
Yu Yang 已提交
1 2 3 4 5 6 7 8
#include <paddle/platform/dynload/cublas.h>

namespace paddle {
namespace platform {
namespace dynload {
std::once_flag cublas_dso_flag;
void *cublas_dso_handle = nullptr;

Q
qijun 已提交
9
#define DEFINE_WRAP(__name) DynLoad__##__name __name
Y
Yu Yang 已提交
10 11 12 13 14

CUBLAS_BLAS_ROUTINE_EACH(DEFINE_WRAP);

}  // namespace dynload
}  // namespace platform
Q
qijun 已提交
15
}  // namespace paddle