Skip to content

P
Paddle

六月暴雪飞梨花 / Paddle
与 Fork 源项目一致

Fork自 PaddlePaddle / Paddle

通知 1
Star 0
Fork 0
P
Paddle

体验新版 GitCode，发现更多精彩内容 >>

提交 9ea0661a 编写于 作者: L Luo Tao
浏览文件
操作

clang format off on some cuda .cc file

上级 80c68d38
隐藏空白更改
内联 并排
Showing with 63 addition and 37 deletion
paddle/cuda/src/hl_cuda_cublas.cc
浏览文件 @ 9ea0661a
...@@ -56,9 +56,14 @@ void *cublas_dso_handle = nullptr; ...@@ -56,9 +56,14 @@ void *cublas_dso_handle = nullptr;
#define DYNAMIC_LOAD_CUBLAS_V2_WRAP(__name) DYNAMIC_LOAD_CUBLAS_WRAP(__name) #define DYNAMIC_LOAD_CUBLAS_V2_WRAP(__name) DYNAMIC_LOAD_CUBLAS_WRAP(__name)
// include all needed cublas functions in HPPL // include all needed cublas functions in HPPL
#define CUBLAS_BLAS_ROUTINE_EACH(__macro) \ // clang-format off
__macro(cublasSgemv) __macro(cublasDgemv) __macro(cublasSgemm) \ #define CUBLAS_BLAS_ROUTINE_EACH(__macro) \
__macro(cublasDgemm) __macro(cublasSgeam) __macro(cublasDgeam) __macro(cublasSgemv) \
__macro(cublasDgemv) \
__macro(cublasSgemm) \
__macro(cublasDgemm) \
__macro(cublasSgeam) \
__macro(cublasDgeam) \
DYNAMIC_LOAD_CUBLAS_V2_WRAP(cublasCreate) DYNAMIC_LOAD_CUBLAS_V2_WRAP(cublasCreate)
DYNAMIC_LOAD_CUBLAS_V2_WRAP(cublasDestroy) DYNAMIC_LOAD_CUBLAS_V2_WRAP(cublasDestroy)
...@@ -81,6 +86,7 @@ CUBLAS_BLAS_ROUTINE_EACH(DYNAMIC_LOAD_CUBLAS_V2_WRAP) ...@@ -81,6 +86,7 @@ CUBLAS_BLAS_ROUTINE_EACH(DYNAMIC_LOAD_CUBLAS_V2_WRAP)
} /* namespace dynload */ } /* namespace dynload */
// clang-format on
#ifndef PADDLE_TYPE_DOUBLE #ifndef PADDLE_TYPE_DOUBLE
#define CUBLAS_GEAM dynload::cublasSgeam #define CUBLAS_GEAM dynload::cublasSgeam
#define CUBLAS_GEMV dynload::cublasSgemv #define CUBLAS_GEMV dynload::cublasSgemv
......
paddle/cuda/src/hl_cuda_device.cc
浏览文件 @ 9ea0661a
...@@ -57,10 +57,14 @@ void *curand_dso_handle = nullptr; ...@@ -57,10 +57,14 @@ void *curand_dso_handle = nullptr;
#endif #endif
/* include all needed curand functions in HPPL */ /* include all needed curand functions in HPPL */
#define CURAND_RAND_ROUTINE_EACH(__macro) \ // clang-format off
__macro(curandCreateGenerator) __macro(curandSetStream) \ #define CURAND_RAND_ROUTINE_EACH(__macro) \
__macro(curandSetPseudoRandomGeneratorSeed) \ __macro(curandCreateGenerator) \
__macro(curandGenerateUniform) __macro(curandGenerateUniformDouble) __macro(curandSetStream) \
__macro(curandSetPseudoRandomGeneratorSeed)\
__macro(curandGenerateUniform) \
__macro(curandGenerateUniformDouble)
// clang-format on
CURAND_RAND_ROUTINE_EACH(DYNAMIC_LOAD_CURAND_WRAP) CURAND_RAND_ROUTINE_EACH(DYNAMIC_LOAD_CURAND_WRAP)
...@@ -99,25 +103,38 @@ void *cudart_dso_handle = nullptr; ...@@ -99,25 +103,38 @@ void *cudart_dso_handle = nullptr;
#endif #endif
/* include all needed cuda functions in HPPL */ /* include all needed cuda functions in HPPL */
#define CUDA_ROUTINE_EACH(__macro) \ // clang-format off
__macro(cudaMalloc) __macro(cudaHostAlloc) __macro(cudaFree) \ #define CUDA_ROUTINE_EACH(__macro) \
__macro(cudaFreeHost) __macro(cudaMemcpy) __macro(cudaMemset) __macro( \ __macro(cudaMalloc) \
cudaMemcpyAsync) __macro(cudaSetDevice) __macro(cudaGetDevice) \ __macro(cudaHostAlloc) \
__macro(cudaGetDeviceCount) __macro(cudaGetDeviceProperties) \ __macro(cudaFree) \
__macro(cudaDeviceSynchronize) __macro(cudaDeviceCanAccessPeer) \ __macro(cudaFreeHost) \
__macro(cudaDeviceEnablePeerAccess) \ __macro(cudaMemcpy) \
__macro(cudaStreamCreate) __macro(cudaStreamDestroy) \ __macro(cudaMemset) \
__macro(cudaStreamSynchronize) __macro( \ __macro(cudaMemcpyAsync) \
cudaStreamWaitEvent) __macro(cudaEventCreate) \ __macro(cudaSetDevice) \
__macro(cudaEventRecord) __macro(cudaEventQuery) \ __macro(cudaGetDevice) \
__macro(cudaEventDestroy) __macro( \ __macro(cudaGetDeviceCount) \
cudaEventSynchronize) \ __macro(cudaGetDeviceProperties) \
__macro(cudaEventElapsedTime) __macro( \ __macro(cudaDeviceSynchronize) \
cudaSetDeviceFlags) \ __macro(cudaDeviceCanAccessPeer) \
__macro(cudaGetLastError) __macro( \ __macro(cudaDeviceEnablePeerAccess) \
cudaFuncSetCacheConfig) \ __macro(cudaStreamCreate) \
__macro(cudaRuntimeGetVersion) \ __macro(cudaStreamDestroy) \
__macro(cudaGetErrorString) __macro(cudaStreamSynchronize) \
__macro(cudaStreamWaitEvent) \
__macro(cudaEventCreate) \
__macro(cudaEventRecord) \
__macro(cudaEventQuery) \
__macro(cudaEventDestroy) \
__macro(cudaEventSynchronize) \
__macro(cudaEventElapsedTime) \
__macro(cudaSetDeviceFlags) \
__macro(cudaGetLastError) \
__macro(cudaFuncSetCacheConfig) \
__macro(cudaRuntimeGetVersion) \
__macro(cudaGetErrorString)
// clang-format on
CUDA_ROUTINE_EACH(DYNAMIC_LOAD_CUDART_WRAP) CUDA_ROUTINE_EACH(DYNAMIC_LOAD_CUDART_WRAP)
......
paddle/cuda/src/hl_cudart_wrap.cc
浏览文件 @ 9ea0661a
...@@ -47,17 +47,20 @@ extern void *cudart_dso_handle; ...@@ -47,17 +47,20 @@ extern void *cudart_dso_handle;
} __name; /* struct DynLoad__##__name */ } __name; /* struct DynLoad__##__name */
/* include all needed cuda functions in HPPL */ /* include all needed cuda functions in HPPL */
#define CUDA_ROUTINE_EACH(__macro) \ // clang-format off
__macro(cudaLaunch, cudaError_t) __macro(cudaSetupArgument, cudaError_t) \ #define CUDA_ROUTINE_EACH(__macro) \
__macro(cudaConfigureCall, cudaError_t) \ __macro(cudaLaunch, cudaError_t) \
__macro(__cudaRegisterFatBinary, void **) \ __macro(cudaSetupArgument, cudaError_t) \
__macro(__cudaUnregisterFatBinary, void) \ __macro(cudaConfigureCall, cudaError_t) \
__macro(__cudaRegisterFunction, void) \ __macro(__cudaRegisterFatBinary, void**) \
__macro(__cudaRegisterVar, void) \ __macro(__cudaUnregisterFatBinary, void) \
__macro(__cudaRegisterManagedVar, void) \ __macro(__cudaRegisterFunction, void) \
__macro(__cudaInitModule, char) \ __macro(__cudaRegisterVar, void) \
__macro(__cudaRegisterTexture, void) \ __macro(__cudaRegisterManagedVar, void) \
__macro(__cudaRegisterSurface, void) __macro(__cudaInitModule, char) \
__macro(__cudaRegisterTexture, void) \
__macro(__cudaRegisterSurface, void)
// clang-format on
CUDA_ROUTINE_EACH(DYNAMIC_LOAD_CUDART_WRAP) CUDA_ROUTINE_EACH(DYNAMIC_LOAD_CUDART_WRAP)
......
Markdown is supported
0% 或 .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑！
想要评论请 注册 或