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
gaoqiong
composable_kernel
Commits
8e862b7b
"docs/git@developer.sourcefind.cn:OpenDAS/mmdetection3d.git" did not exist on "75c5c6cd590386bd1539a686c5fd2cc45c5480d5"
Commit
8e862b7b
authored
Mar 07, 2023
by
aska-0096
Browse files
Change GridwiseOp nam
parent
c5fd087e
Changes
8
Hide whitespace changes
Inline
Side-by-side
Showing
8 changed files
with
8 additions
and
8 deletions
+8
-8
include/ck/tensor_operation/gpu/device/impl/device_batched_contraction_multiple_d_wmma_cshuffle.hpp
...l/device_batched_contraction_multiple_d_wmma_cshuffle.hpp
+1
-1
include/ck/tensor_operation/gpu/device/impl/device_batched_gemm_softmax_gemm_permute_wmma_cshuffle.hpp
...evice_batched_gemm_softmax_gemm_permute_wmma_cshuffle.hpp
+1
-1
include/ck/tensor_operation/gpu/device/impl/device_gemm_multiple_d_wmma_cshuffle.hpp
.../gpu/device/impl/device_gemm_multiple_d_wmma_cshuffle.hpp
+1
-1
include/ck/tensor_operation/gpu/device/impl/device_gemm_wmma.hpp
.../ck/tensor_operation/gpu/device/impl/device_gemm_wmma.hpp
+1
-1
include/ck/tensor_operation/gpu/device/impl/device_grouped_conv_fwd_multiple_d_wmma_cshuffle.hpp
...impl/device_grouped_conv_fwd_multiple_d_wmma_cshuffle.hpp
+1
-1
include/ck/tensor_operation/gpu/grid/gridwise_batched_gemm_softmax_gemm_wmma_cshuffle.hpp
...grid/gridwise_batched_gemm_softmax_gemm_wmma_cshuffle.hpp
+1
-1
include/ck/tensor_operation/gpu/grid/gridwise_gemm_multiple_d_wmma_cshuffle.hpp
...ation/gpu/grid/gridwise_gemm_multiple_d_wmma_cshuffle.hpp
+1
-1
include/ck/tensor_operation/gpu/grid/gridwise_gemm_wmma.hpp
include/ck/tensor_operation/gpu/grid/gridwise_gemm_wmma.hpp
+1
-1
No files found.
include/ck/tensor_operation/gpu/device/impl/device_batched_contraction_multiple_d_wmma_cshuffle.hpp
View file @
8e862b7b
...
...
@@ -526,7 +526,7 @@ struct DeviceBatchedContractionMultipleD_Wmma_CShuffle
using
BGridDesc
=
decltype
(
DeviceOp
::
MakeBGridDescriptor
({},
{}));
// GridwiseOp
using
GridwiseOp
=
GridwiseGemmMultipleD_
k0mk1_k0nk1_mn_wmma_cshuffle
<
using
GridwiseOp
=
GridwiseGemmMultipleD_
Wmma
<
// DataType Family
ADataType
,
BDataType
,
...
...
include/ck/tensor_operation/gpu/device/impl/device_batched_gemm_softmax_gemm_permute_wmma_cshuffle.hpp
View file @
8e862b7b
...
...
@@ -286,7 +286,7 @@ struct DeviceBatchedGemmSoftmaxGemmPermute_Wmma_CShuffle
};
// GridwiseOp
using
GridwiseOp
=
GridwiseBatchedGemmSoftmaxGemm_Wmma
_CShuffle
<
using
GridwiseOp
=
GridwiseBatchedGemmSoftmaxGemm_Wmma
<
// DataType Family
ADataType
,
B0DataType
,
...
...
include/ck/tensor_operation/gpu/device/impl/device_gemm_multiple_d_wmma_cshuffle.hpp
View file @
8e862b7b
...
...
@@ -263,7 +263,7 @@ struct DeviceGemmMultipleD_Wmma_CShuffle : public DeviceGemmMultipleD<ALayout,
using
EGridDesc_M_N
=
decltype
(
MakeEGridDescriptor_M_N
<
ELayout
>
(
1
,
1
,
1
));
// GridwiseOp
using
GridwiseOp
=
GridwiseGemmMultipleD_
k0mk1_k0nk1_mn_wmma_cshuffle
<
using
GridwiseOp
=
GridwiseGemmMultipleD_
Wmma
<
// DataType Family
ADataType
,
BDataType
,
...
...
include/ck/tensor_operation/gpu/device/impl/device_gemm_wmma.hpp
View file @
8e862b7b
...
...
@@ -222,7 +222,7 @@ struct DeviceGemmWmma_CShuffle : public DeviceGemm<ALayout,
using
CGridDesc_M_N
=
decltype
(
MakeCGridDescriptor_M_N
(
1
,
1
,
1
));
// GridwiseGemm
using
GridwiseGemm
=
GridwiseGemm_
k0mk1_k0nk1_mn_w
mma
<
using
GridwiseGemm
=
GridwiseGemm_
W
mma
<
BlockSize
,
ADataType
,
BDataType
,
...
...
include/ck/tensor_operation/gpu/device/impl/device_grouped_conv_fwd_multiple_d_wmma_cshuffle.hpp
View file @
8e862b7b
...
...
@@ -323,7 +323,7 @@ struct DeviceGroupedConvFwdMultipleD_Wmma_CShuffle
using
BGridDesc
=
decltype
(
DeviceOp
::
MakeBGridDescriptor
<
BLayout
>
({},
{}));
// GridwiseOp
using
GridwiseOp
=
GridwiseGemmMultipleD_
k0mk1_k0nk1_mn_wmma_cshuffle
<
using
GridwiseOp
=
GridwiseGemmMultipleD_
Wmma
<
// DataType Family
ADataType
,
BDataType
,
...
...
include/ck/tensor_operation/gpu/grid/gridwise_batched_gemm_softmax_gemm_wmma_cshuffle.hpp
View file @
8e862b7b
...
...
@@ -182,7 +182,7 @@ template <typename ADataType,
index_t
NumGemmKPrefetchStage
=
1
,
LoopScheduler
LoopSched
=
make_default_loop_scheduler
(),
PipelineVersion
PipelineVer
=
PipelineVersion
::
v1
>
struct
GridwiseBatchedGemmSoftmaxGemm_Wmma
_CShuffle
struct
GridwiseBatchedGemmSoftmaxGemm_Wmma
{
static
constexpr
auto
I0
=
Number
<
0
>
{};
static
constexpr
auto
I1
=
Number
<
1
>
{};
...
...
include/ck/tensor_operation/gpu/grid/gridwise_gemm_multiple_d_wmma_cshuffle.hpp
View file @
8e862b7b
...
...
@@ -321,7 +321,7 @@ template < // DataType Family
index_t
NumGemmKPrefetchStage
=
1
,
LoopScheduler
LoopSched
=
make_default_loop_scheduler
(),
PipelineVersion
PipelineVer
=
PipelineVersion
::
v1
>
struct
GridwiseGemmMultipleD_
k0mk1_k0nk1_mn_wmma_cshuffle
struct
GridwiseGemmMultipleD_
Wmma
{
static
constexpr
index_t
NumDTensor
=
DsDataType
::
Size
();
...
...
include/ck/tensor_operation/gpu/grid/gridwise_gemm_wmma.hpp
View file @
8e862b7b
...
...
@@ -119,7 +119,7 @@ template <index_t BlockSize,
index_t
NumGemmKPrefetchStage
=
1
,
LoopScheduler
LoopSched
=
make_default_loop_scheduler
(),
PipelineVersion
PipelineVer
=
PipelineVersion
::
v1
>
struct
GridwiseGemm_
k0mk1_k0nk1_mn_w
mma
struct
GridwiseGemm_
W
mma
{
static
constexpr
auto
I0
=
Number
<
0
>
{};
static
constexpr
auto
I1
=
Number
<
1
>
{};
...
...
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