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
d6d30a86
Commit
d6d30a86
authored
Sep 25, 2023
by
Qianfeng Zhang
Browse files
Two tiny fixes
parent
c0c52268
Changes
1
Show whitespace changes
Inline
Side-by-side
Showing
1 changed file
with
2 additions
and
1 deletion
+2
-1
include/ck/tensor_operation/gpu/device/impl/device_batched_dropout.hpp
...nsor_operation/gpu/device/impl/device_batched_dropout.hpp
+2
-1
No files found.
include/ck/tensor_operation/gpu/device/impl/device_batched_dropout.hpp
View file @
d6d30a86
...
@@ -71,6 +71,7 @@ __global__ void
...
@@ -71,6 +71,7 @@ __global__ void
ignore
=
p_z_grid
;
ignore
=
p_z_grid
;
ignore
=
c_grid_desc_m0_n0_m1_n1_m2_n2_m3_m4_m5_n3
;
ignore
=
c_grid_desc_m0_n0_m1_n1_m2_n2_m3_m4_m5_n3
;
ignore
=
block_2_ctile_map
;
ignore
=
block_2_ctile_map
;
ignore
=
num_gemm0_m_block_outer_loop
;
ignore
=
batch_count
;
ignore
=
batch_count
;
ignore
=
compute_base_ptr_of_batch
;
ignore
=
compute_base_ptr_of_batch
;
ignore
=
seed
;
ignore
=
seed
;
...
@@ -135,7 +136,7 @@ struct DeviceBatchedDropout : public ck::tensor_operation::device::BaseOperator
...
@@ -135,7 +136,7 @@ struct DeviceBatchedDropout : public ck::tensor_operation::device::BaseOperator
static
auto
MakeZGridDescriptor_M_N
(
const
std
::
vector
<
index_t
>&
z_gs_m_n_lengths
,
static
auto
MakeZGridDescriptor_M_N
(
const
std
::
vector
<
index_t
>&
z_gs_m_n_lengths
,
const
std
::
vector
<
index_t
>&
z_gs_m_n_strides
)
const
std
::
vector
<
index_t
>&
z_gs_m_n_strides
)
{
{
return
Transform
::
MakeCGridDescriptor_M_N
(
z_gs_m_n_lengths
,
z_gs_m_n_strides
);
return
Transform
::
MakeC
0
GridDescriptor_M_N
(
z_gs_m_n_lengths
,
z_gs_m_n_strides
);
}
}
using
ZGridDesc_G_M_N
=
decltype
(
Transform
::
MakeCGridDescriptor_G_M_N
({},
{}));
using
ZGridDesc_G_M_N
=
decltype
(
Transform
::
MakeCGridDescriptor_G_M_N
({},
{}));
...
...
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