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__
#define __INFINIOP_CUDA_FUNCTIONS_CUH__
#ifndef __INFINIOP_CUDA_COMMON_CUH__
#define __INFINIOP_CUDA_COMMON_CUH__
#include "cuda_handle.cuh"
#include "infinicore.h"
#include <cudnn.h>
namespace device::cuda {
......@@ -40,4 +40,4 @@ indexToOffset(
} // namespace device::cuda
#endif // __INFINIOP_CUDA_FUNCTIONS_CUH__
#endif // __INFINIOP_CUDA_COMMON_CUH__
......@@ -3,9 +3,9 @@
#include "../../../utils.h"
#include "../pool.h"
#include "cuda_functions.cuh"
#include "cuda_handle.h"
#include <cublas_v2.h>
#include <cudnn.h>
#include <functional>
#define CHECK_CUBLAS(API) CHECK_INTERNAL(API, CUBLAS_STATUS_SUCCESS)
......
......@@ -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_b = _info.b_matrix.row_stride == 1 ? CUBLAS_OP_N : CUBLAS_OP_T;
_opaque->internal->useCublas(
CHECK_STATUS(_opaque->internal->useCublas(
(cudaStream_t)stream,
[&](cublasHandle_t handle) {
CHECK_CUBLAS(
......@@ -105,7 +105,7 @@ infiniStatus_t Descriptor::calculate(
compute_type,
CUBLAS_GEMM_DEFAULT_TENSOR_OP));
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