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
bb314592
"...composable_kernel.git" did not exist on "6dfb92bbef33b4caea55f6b4ed7c449927ae771c"
Commit
bb314592
authored
May 25, 2022
by
rocking
Browse files
Remove epislon
parent
459f63a8
Changes
1
Hide whitespace changes
Inline
Side-by-side
Showing
1 changed file
with
3 additions
and
3 deletions
+3
-3
include/ck/tensor_operation/gpu/device/device_normalize_xdl_cshuffle.hpp
...or_operation/gpu/device/device_normalize_xdl_cshuffle.hpp
+3
-3
No files found.
include/ck/tensor_operation/gpu/device/device_normalize_xdl_cshuffle.hpp
View file @
bb314592
...
@@ -83,7 +83,6 @@ struct DeviceNormalize_Xdl_CShuffle : public BaseOperator
...
@@ -83,7 +83,6 @@ struct DeviceNormalize_Xdl_CShuffle : public BaseOperator
const
GammaDataType
*
p_gamma
,
const
GammaDataType
*
p_gamma
,
const
BetaDataType
*
p_beta
,
const
BetaDataType
*
p_beta
,
OutDataType
*
p_output
,
OutDataType
*
p_output
,
MeanSquareDataType
epsilon
,
const
std
::
vector
<
index_t
>&
shape
,
const
std
::
vector
<
index_t
>&
shape
,
const
std
::
vector
<
index_t
>&
stride_x
,
const
std
::
vector
<
index_t
>&
stride_x
,
const
std
::
vector
<
index_t
>&
stride_mean
,
const
std
::
vector
<
index_t
>&
stride_mean
,
...
@@ -98,7 +97,6 @@ struct DeviceNormalize_Xdl_CShuffle : public BaseOperator
...
@@ -98,7 +97,6 @@ struct DeviceNormalize_Xdl_CShuffle : public BaseOperator
p_gamma_
(
p_gamma
),
p_gamma_
(
p_gamma
),
p_beta_
(
p_beta
),
p_beta_
(
p_beta
),
p_output_
(
p_output
),
p_output_
(
p_output
),
epsilon_
(
epsilon
),
shape_
(
shape
),
shape_
(
shape
),
stride_x_
(
stride_x
),
stride_x_
(
stride_x
),
stride_mean_
(
stride_mean
),
stride_mean_
(
stride_mean
),
...
@@ -124,7 +122,6 @@ struct DeviceNormalize_Xdl_CShuffle : public BaseOperator
...
@@ -124,7 +122,6 @@ struct DeviceNormalize_Xdl_CShuffle : public BaseOperator
const
GammaDataType
*
p_gamma_
;
const
GammaDataType
*
p_gamma_
;
const
BetaDataType
*
p_beta_
;
const
BetaDataType
*
p_beta_
;
OutDataType
*
p_output_
;
OutDataType
*
p_output_
;
MeanSquareDataType
epsilon_
;
std
::
vector
<
index_t
>
shape_
;
std
::
vector
<
index_t
>
shape_
;
GridDesc_M0
x_grid_desc_m0_
;
GridDesc_M0
x_grid_desc_m0_
;
GridDesc_M0
mean_grid_desc_m0_
;
GridDesc_M0
mean_grid_desc_m0_
;
...
@@ -179,6 +176,9 @@ struct DeviceNormalize_Xdl_CShuffle : public BaseOperator
...
@@ -179,6 +176,9 @@ struct DeviceNormalize_Xdl_CShuffle : public BaseOperator
if
(
pArg
==
nullptr
)
if
(
pArg
==
nullptr
)
return
false
;
return
false
;
if
(
pArg
->
shape_
.
size
()
!=
Dim
)
return
false
;
if
(
pArg
->
shape_
.
back
()
%
M0PerThread
!=
0
)
if
(
pArg
->
shape_
.
back
()
%
M0PerThread
!=
0
)
return
false
;
return
false
;
...
...
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