Skip to content
GitLab
Menu
Projects
Groups
Snippets
Loading...
Help
Help
Support
Community forum
Keyboard shortcuts
?
Submit feedback
Contribute to GitLab
Sign in / Register
Toggle navigation
Menu
Open sidebar
jerrrrry
infinicore
Commits
911115fb
Commit
911115fb
authored
Mar 07, 2025
by
YdrMaster
Browse files
issue/87/style: 文件改名
Signed-off-by:
YdrMaster
<
ydrml@hotmail.com
>
parent
39bf75c5
Changes
4
Hide whitespace changes
Inline
Side-by-side
Showing
4 changed files
with
7 additions
and
7 deletions
+7
-7
src/infiniop/devices/cuda/cuda_common.cu
src/infiniop/devices/cuda/cuda_common.cu
+0
-0
src/infiniop/devices/cuda/cuda_common.cuh
src/infiniop/devices/cuda/cuda_common.cuh
+4
-4
src/infiniop/devices/cuda/cuda_handle.cuh
src/infiniop/devices/cuda/cuda_handle.cuh
+1
-1
src/infiniop/ops/matmul/cuda/matmul_cuda.cu
src/infiniop/ops/matmul/cuda/matmul_cuda.cu
+2
-2
No files found.
src/infiniop/devices/cuda/cuda_
handle
.cu
→
src/infiniop/devices/cuda/cuda_
common
.cu
View file @
911115fb
File moved
src/infiniop/devices/cuda/cuda_
functi
on
s
.cuh
→
src/infiniop/devices/cuda/cuda_
comm
on.cuh
View file @
911115fb
#ifndef __INFINIOP_CUDA_
FUNCTI
ON
S
_CUH__
#define __INFINIOP_CUDA_
FUNCTI
ON
S
_CUH__
#ifndef __INFINIOP_CUDA_
COMM
ON_CUH__
#define __INFINIOP_CUDA_
COMM
ON_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_
FUNCTI
ON
S
_CUH__
#endif // __INFINIOP_CUDA_
COMM
ON_CUH__
src/infiniop/devices/cuda/cuda_handle.cuh
View file @
911115fb
...
...
@@ -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)
...
...
src/infiniop/ops/matmul/cuda/matmul_cuda.cu
View file @
911115fb
...
...
@@ -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
;
}
...
...
Write
Preview
Markdown
is supported
0%
Try again
or
attach a new file
.
Attach a file
Cancel
You are about to add
0
people
to the discussion. Proceed with caution.
Finish editing this message first!
Cancel
Please
register
or
sign in
to comment