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
9402ee4b
Commit
9402ee4b
authored
May 30, 2022
by
rocking
Browse files
Refine class name
parent
af5b9da7
Changes
2
Show whitespace changes
Inline
Side-by-side
Showing
2 changed files
with
4 additions
and
4 deletions
+4
-4
example/21_gemm_layernorm/gemm_layernorm_xdl_fp16.cpp
example/21_gemm_layernorm/gemm_layernorm_xdl_fp16.cpp
+3
-3
include/ck/tensor_operation/gpu/device/device_5ary_elementwise.hpp
...k/tensor_operation/gpu/device/device_5ary_elementwise.hpp
+1
-1
No files found.
example/21_gemm_layernorm/gemm_layernorm_xdl_fp16.cpp
View file @
9402ee4b
...
@@ -10,7 +10,7 @@
...
@@ -10,7 +10,7 @@
#include "host_tensor.hpp"
#include "host_tensor.hpp"
#include "host_tensor_generator.hpp"
#include "host_tensor_generator.hpp"
#include "device_tensor.hpp"
#include "device_tensor.hpp"
#include "device_5ary_elementwise
_xdl_cshuffle
.hpp"
#include "device_5ary_elementwise.hpp"
#include "device_gemm_reduce_xdl_cshuffle.hpp"
#include "device_gemm_reduce_xdl_cshuffle.hpp"
#include "element_wise_operation.hpp"
#include "element_wise_operation.hpp"
#include "reference_gemm.hpp"
#include "reference_gemm.hpp"
...
@@ -84,7 +84,7 @@ using ReferenceGemmInstance = ck::tensor_operation::host::ReferenceGemm<ADataTyp
...
@@ -84,7 +84,7 @@ using ReferenceGemmInstance = ck::tensor_operation::host::ReferenceGemm<ADataTyp
using
NormalizeFunctor
=
ck
::
tensor_operation
::
element_wise
::
Normalize
;
using
NormalizeFunctor
=
ck
::
tensor_operation
::
element_wise
::
Normalize
;
// A:x, B:E[x], C:E[x^2], D:Gamma, E:Beta , F:y
// A:x, B:E[x], C:E[x^2], D:Gamma, E:Beta , F:y
using
DeviceNormalizeInstance
=
ck
::
tensor_operation
::
device
::
Device5AryElementwise
_Xdl_CShuffle
<
using
DeviceNormalizeInstance
=
ck
::
tensor_operation
::
device
::
Device5AryElementwise
<
CDataType
,
CDataType
,
DDataType
,
DDataType
,
DDataType
,
DDataType
,
...
@@ -295,7 +295,7 @@ int main()
...
@@ -295,7 +295,7 @@ int main()
if
(
!
normalize
.
IsSupportedArgument
(
normalize_argument
))
if
(
!
normalize
.
IsSupportedArgument
(
normalize_argument
))
{
{
throw
std
::
runtime_error
(
"The runtime parameters seems not supported by the "
throw
std
::
runtime_error
(
"The runtime parameters seems not supported by the "
"Device5AryElementwise
_Xdl_CShuffle
instance, exiting!"
);
"Device5AryElementwise instance, exiting!"
);
}
}
// run kernel
// run kernel
...
...
include/ck/tensor_operation/gpu/device/device_5ary_elementwise
_xdl_cshuffle
.hpp
→
include/ck/tensor_operation/gpu/device/device_5ary_elementwise.hpp
View file @
9402ee4b
...
@@ -29,7 +29,7 @@ template <typename ADataType,
...
@@ -29,7 +29,7 @@ template <typename ADataType,
index_t
DScalarPerVector
,
index_t
DScalarPerVector
,
index_t
EScalarPerVector
,
index_t
EScalarPerVector
,
index_t
FScalarPerVector
>
index_t
FScalarPerVector
>
struct
Device5AryElementwise
_Xdl_CShuffle
:
public
BaseOperator
struct
Device5AryElementwise
:
public
BaseOperator
{
{
static
constexpr
auto
I0
=
Number
<
0
>
{};
static
constexpr
auto
I0
=
Number
<
0
>
{};
...
...
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