Commit 911115fb authored by YdrMaster's avatar YdrMaster
Browse files

issue/87/style: 文件改名


Signed-off-by: default avatarYdrMaster <ydrml@hotmail.com>
parent 39bf75c5
#ifndef __INFINIOP_CUDA_FUNCTIONS_CUH__ #ifndef __INFINIOP_CUDA_COMMON_CUH__
#define __INFINIOP_CUDA_FUNCTIONS_CUH__ #define __INFINIOP_CUDA_COMMON_CUH__
#include "cuda_handle.cuh"
#include "infinicore.h" #include "infinicore.h"
#include <cudnn.h>
namespace device::cuda { namespace device::cuda {
...@@ -40,4 +40,4 @@ indexToOffset( ...@@ -40,4 +40,4 @@ indexToOffset(
} // namespace device::cuda } // namespace device::cuda
#endif // __INFINIOP_CUDA_FUNCTIONS_CUH__ #endif // __INFINIOP_CUDA_COMMON_CUH__
...@@ -3,9 +3,9 @@ ...@@ -3,9 +3,9 @@
#include "../../../utils.h" #include "../../../utils.h"
#include "../pool.h" #include "../pool.h"
#include "cuda_functions.cuh"
#include "cuda_handle.h" #include "cuda_handle.h"
#include <cublas_v2.h> #include <cublas_v2.h>
#include <cudnn.h>
#include <functional> #include <functional>
#define CHECK_CUBLAS(API) CHECK_INTERNAL(API, CUBLAS_STATUS_SUCCESS) #define CHECK_CUBLAS(API) CHECK_INTERNAL(API, CUBLAS_STATUS_SUCCESS)
......
...@@ -76,7 +76,7 @@ infiniStatus_t Descriptor::calculate( ...@@ -76,7 +76,7 @@ infiniStatus_t Descriptor::calculate(
auto op_a = _info.a_matrix.row_stride == 1 ? CUBLAS_OP_N : CUBLAS_OP_T; auto op_a = _info.a_matrix.row_stride == 1 ? CUBLAS_OP_N : CUBLAS_OP_T;
auto op_b = _info.b_matrix.row_stride == 1 ? CUBLAS_OP_N : CUBLAS_OP_T; auto op_b = _info.b_matrix.row_stride == 1 ? CUBLAS_OP_N : CUBLAS_OP_T;
_opaque->internal->useCublas( CHECK_STATUS(_opaque->internal->useCublas(
(cudaStream_t)stream, (cudaStream_t)stream,
[&](cublasHandle_t handle) { [&](cublasHandle_t handle) {
CHECK_CUBLAS( CHECK_CUBLAS(
...@@ -105,7 +105,7 @@ infiniStatus_t Descriptor::calculate( ...@@ -105,7 +105,7 @@ infiniStatus_t Descriptor::calculate(
compute_type, compute_type,
CUBLAS_GEMM_DEFAULT_TENSOR_OP)); CUBLAS_GEMM_DEFAULT_TENSOR_OP));
return INFINI_STATUS_SUCCESS; return INFINI_STATUS_SUCCESS;
}); }));
return INFINI_STATUS_SUCCESS; return INFINI_STATUS_SUCCESS;
} }
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment