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
0ba41814
Commit
0ba41814
authored
Sep 07, 2022
by
Po-Yen, Chen
Browse files
Rename 'BlockToTileMap' as 'Block2TileMap'
parent
c2e5822c
Changes
1
Show whitespace changes
Inline
Side-by-side
Showing
1 changed file
with
11 additions
and
4 deletions
+11
-4
include/ck/tensor_operation/gpu/grid/gridwise_copy.hpp
include/ck/tensor_operation/gpu/grid/gridwise_copy.hpp
+11
-4
No files found.
include/ck/tensor_operation/gpu/grid/gridwise_copy.hpp
View file @
0ba41814
...
@@ -16,16 +16,23 @@
...
@@ -16,16 +16,23 @@
namespace
ck
{
namespace
ck
{
namespace
detail
{
namespace
detail
{
template
<
typename
TileDims
,
typename
GridDescriptor
>
template
<
typename
TileDims
,
typename
GridDescriptor
>
struct
Block
To
TileMap
struct
Block
2
TileMap
{
{
static
constexpr
auto
I0
=
Number
<
0
>
{};
static
constexpr
auto
I0
=
Number
<
0
>
{};
static
constexpr
index_t
NumDim
=
TileDims
::
Size
();
static
constexpr
index_t
NumDim
=
TileDims
::
Size
();
static_assert
(
NumDim
==
GridDescriptor
::
GetNumOfDimension
());
static_assert
(
NumDim
==
GridDescriptor
::
GetNumOfDimension
());
BlockToTileMap
()
=
default
;
Block2TileMap
()
=
delete
;
Block2TileMap
(
const
Block2TileMap
&
)
=
default
;
Block2TileMap
(
Block2TileMap
&&
)
=
delete
;
BlockToTileMap
(
const
GridDescriptor
&
desc
)
:
desc_
(
desc
)
{}
~
Block2TileMap
()
=
default
;
Block2TileMap
&
operator
=
(
const
Block2TileMap
&
)
=
delete
;
Block2TileMap
&
operator
=
(
Block2TileMap
&&
)
=
delete
;
explicit
Block2TileMap
(
const
GridDescriptor
&
desc
)
:
desc_
(
desc
)
{}
__host__
constexpr
index_t
CalculateGridSize
(
const
GridDescriptor
&
desc
)
const
__host__
constexpr
index_t
CalculateGridSize
(
const
GridDescriptor
&
desc
)
const
{
{
...
@@ -126,7 +133,7 @@ struct GridwiseCopy
...
@@ -126,7 +133,7 @@ struct GridwiseCopy
using
ThisThreadBlock
=
ThisThreadBlock
<
BlockSize
>
;
using
ThisThreadBlock
=
ThisThreadBlock
<
BlockSize
>
;
using
DefaultBlock2TileMap
=
using
DefaultBlock2TileMap
=
detail
::
Block
To
TileMap
<
Sequence
<
NPerBlock
,
HPerBlock
,
WPerBlock
>
,
InGrid1dDesc
>
;
detail
::
Block
2
TileMap
<
Sequence
<
NPerBlock
,
HPerBlock
,
WPerBlock
>
,
InGrid1dDesc
>
;
__host__
__device__
static
constexpr
auto
GetInBlockDescriptor
()
__host__
__device__
static
constexpr
auto
GetInBlockDescriptor
()
{
{
...
...
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