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
9e03c383
"examples/vscode:/vscode.git/clone" did not exist on "5f150c4cef2140342c66ef44ba8f36b5844a51b1"
Commit
9e03c383
authored
Sep 14, 2022
by
Po-Yen, Chen
Browse files
Use type alias to replace duplicated code
parent
95f21ea5
Changes
1
Hide whitespace changes
Inline
Side-by-side
Showing
1 changed file
with
6 additions
and
5 deletions
+6
-5
library/include/ck/library/utility/host_tensor.hpp
library/include/ck/library/utility/host_tensor.hpp
+6
-5
No files found.
library/include/ck/library/utility/host_tensor.hpp
View file @
9e03c383
...
@@ -235,7 +235,8 @@ auto make_ParallelTensorFunctor(F f, Xs... xs)
...
@@ -235,7 +235,8 @@ auto make_ParallelTensorFunctor(F f, Xs... xs)
template
<
typename
T
>
template
<
typename
T
>
struct
Tensor
struct
Tensor
{
{
using
Data
=
std
::
vector
<
T
>
;
using
Descriptor
=
HostTensorDescriptor
;
using
Data
=
std
::
vector
<
T
>
;
template
<
typename
X
>
template
<
typename
X
>
Tensor
(
std
::
initializer_list
<
X
>
lens
)
:
mDesc
(
lens
),
mData
(
mDesc
.
GetElementSpaceSize
())
Tensor
(
std
::
initializer_list
<
X
>
lens
)
:
mDesc
(
lens
),
mData
(
mDesc
.
GetElementSpaceSize
())
...
@@ -253,7 +254,7 @@ struct Tensor
...
@@ -253,7 +254,7 @@ struct Tensor
{
{
}
}
Tensor
(
const
HostTensor
Descriptor
&
desc
)
:
mDesc
(
desc
),
mData
(
mDesc
.
GetElementSpaceSize
())
{}
Tensor
(
const
Descriptor
&
desc
)
:
mDesc
(
desc
),
mData
(
mDesc
.
GetElementSpaceSize
())
{}
template
<
typename
OutT
>
template
<
typename
OutT
>
Tensor
<
OutT
>
CopyAsType
()
const
Tensor
<
OutT
>
CopyAsType
()
const
...
@@ -280,9 +281,9 @@ struct Tensor
...
@@ -280,9 +281,9 @@ struct Tensor
{
{
}
}
const
std
::
vector
<
std
::
size_t
>&
GetLengths
()
const
{
return
mDesc
.
GetLengths
();
}
decltype
(
auto
)
GetLengths
()
const
{
return
mDesc
.
GetLengths
();
}
const
std
::
vector
<
std
::
size_t
>&
GetStrides
()
const
{
return
mDesc
.
GetStrides
();
}
decltype
(
auto
)
GetStrides
()
const
{
return
mDesc
.
GetStrides
();
}
std
::
size_t
GetNumOfDimension
()
const
{
return
mDesc
.
GetNumOfDimension
();
}
std
::
size_t
GetNumOfDimension
()
const
{
return
mDesc
.
GetNumOfDimension
();
}
...
@@ -441,6 +442,6 @@ struct Tensor
...
@@ -441,6 +442,6 @@ struct Tensor
typename
Data
::
const_pointer
data
()
const
{
return
mData
.
data
();
}
typename
Data
::
const_pointer
data
()
const
{
return
mData
.
data
();
}
HostTensor
Descriptor
mDesc
;
Descriptor
mDesc
;
Data
mData
;
Data
mData
;
};
};
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