提交 4e918377 编写于 作者: Q qijun

fix bug in dynload

上级 e6e2bf45
...@@ -11,7 +11,7 @@ proto_library(op_proto SRCS op_proto.proto DEPS attr_type) ...@@ -11,7 +11,7 @@ proto_library(op_proto SRCS op_proto.proto DEPS attr_type)
cc_test(op_proto_test SRCS op_proto_test.cc DEPS op_proto protobuf) cc_test(op_proto_test SRCS op_proto_test.cc DEPS op_proto protobuf)
proto_library(op_desc SRCS op_desc.proto DEPS attr_type) proto_library(op_desc SRCS op_desc.proto DEPS attr_type)
cc_test(op_desc_test SRCS op_desc_test.cc DEPS op_desc protobuf) cc_test(op_desc_test SRCS op_desc_test.cc DEPS op_desc protobuf)
cc_library(operator SRCS operator.cc DEPS op_desc protobuf) cc_library(operator SRCS operator.cc DEPS op_desc protobuf device_context)
cc_test(operator_test SRCS operator_test.cc DEPS operator op_registry place) cc_test(operator_test SRCS operator_test.cc DEPS operator op_registry place)
cc_library(op_registry SRCS op_registry.cc DEPS op_proto op_desc) cc_library(op_registry SRCS op_registry.cc DEPS op_proto op_desc)
cc_test(op_registry_test SRCS op_registry_test.cc DEPS op_registry operator) cc_test(op_registry_test SRCS op_registry_test.cc DEPS op_registry operator)
......
...@@ -67,20 +67,20 @@ extern void *cublas_dso_handle; ...@@ -67,20 +67,20 @@ extern void *cublas_dso_handle;
__macro(cublasSgemm); \ __macro(cublasSgemm); \
__macro(cublasDgemm); \ __macro(cublasDgemm); \
__macro(cublasSgeam); \ __macro(cublasSgeam); \
__macro(cublasDgeam); __macro(cublasDgeam); \
__macro(cublasCreate_v2); \
DECLARE_DYNAMIC_LOAD_CUBLAS_V2_WRAP(cublasCreate); __macro(cublasDestroy_v2); \
DECLARE_DYNAMIC_LOAD_CUBLAS_V2_WRAP(cublasDestroy); __macro(cublasSetStream_v2); \
DECLARE_DYNAMIC_LOAD_CUBLAS_V2_WRAP(cublasSetStream); __macro(cublasSetPointerMode_v2); \
DECLARE_DYNAMIC_LOAD_CUBLAS_V2_WRAP(cublasSetPointerMode); __macro(cublasGetPointerMode_v2); \
DECLARE_DYNAMIC_LOAD_CUBLAS_V2_WRAP(cublasGetPointerMode); __macro(cublasSgemmBatched); \
DECLARE_DYNAMIC_LOAD_CUBLAS_WRAP(cublasSgemmBatched); __macro(cublasDgemmBatched); \
DECLARE_DYNAMIC_LOAD_CUBLAS_WRAP(cublasDgemmBatched); __macro(cublasCgemmBatched); \
DECLARE_DYNAMIC_LOAD_CUBLAS_WRAP(cublasCgemmBatched); __macro(cublasZgemmBatched); \
DECLARE_DYNAMIC_LOAD_CUBLAS_WRAP(cublasZgemmBatched); __macro(cublasSgetrfBatched); \
DECLARE_DYNAMIC_LOAD_CUBLAS_WRAP(cublasSgetrfBatched); __macro(cublasSgetriBatched); \
DECLARE_DYNAMIC_LOAD_CUBLAS_WRAP(cublasSgetriBatched); __macro(cublasDgetrfBatched); \
DECLARE_DYNAMIC_LOAD_CUBLAS_WRAP(cublasDgetrfBatched); __macro(cublasDgetriBatched)
CUBLAS_BLAS_ROUTINE_EACH(DECLARE_DYNAMIC_LOAD_CUBLAS_WRAP); CUBLAS_BLAS_ROUTINE_EACH(DECLARE_DYNAMIC_LOAD_CUBLAS_WRAP);
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册