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
0cfbefce
Commit
0cfbefce
authored
Sep 14, 2022
by
Po-Yen, Chen
Browse files
Add Tensor<>::data()
parent
31875367
Changes
1
Hide whitespace changes
Inline
Side-by-side
Showing
1 changed file
with
15 additions
and
9 deletions
+15
-9
library/include/ck/library/utility/host_tensor.hpp
library/include/ck/library/utility/host_tensor.hpp
+15
-9
No files found.
library/include/ck/library/utility/host_tensor.hpp
View file @
0cfbefce
...
@@ -3,13 +3,13 @@
...
@@ -3,13 +3,13 @@
#pragma once
#pragma once
#include <thread>
#include <vector>
#include <numeric>
#include <algorithm>
#include <algorithm>
#include <utility>
#include <cassert>
#include <cassert>
#include <iostream>
#include <iostream>
#include <numeric>
#include <thread>
#include <utility>
#include <vector>
#include "ck/utility/data_type.hpp"
#include "ck/utility/data_type.hpp"
...
@@ -235,6 +235,8 @@ auto make_ParallelTensorFunctor(F f, Xs... xs)
...
@@ -235,6 +235,8 @@ auto make_ParallelTensorFunctor(F f, Xs... xs)
template
<
typename
T
>
template
<
typename
T
>
struct
Tensor
struct
Tensor
{
{
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
())
{
{
...
@@ -427,14 +429,18 @@ struct Tensor
...
@@ -427,14 +429,18 @@ struct Tensor
return
mData
[
mDesc
.
GetOffsetFromMultiIndex
(
idx
)];
return
mData
[
mDesc
.
GetOffsetFromMultiIndex
(
idx
)];
}
}
typename
std
::
vector
<
T
>::
iterator
begin
()
{
return
mData
.
begin
();
}
typename
Data
::
iterator
begin
()
{
return
mData
.
begin
();
}
typename
Data
::
iterator
end
()
{
return
mData
.
end
();
}
typename
Data
::
pointer
data
()
{
return
mData
.
data
();
}
typename
std
::
vector
<
T
>::
iterator
end
()
{
return
mData
.
end
();
}
typename
Data
::
const_iterator
begin
()
const
{
return
mData
.
begin
();
}
typename
std
::
vector
<
T
>
::
const_iterator
begin
()
const
{
return
mData
.
begin
();
}
typename
Data
::
const_iterator
end
()
const
{
return
mData
.
end
();
}
typename
std
::
vector
<
T
>::
const_iterator
end
()
const
{
return
mData
.
end
();
}
typename
Data
::
const_pointer
data
()
const
{
return
mData
.
data
();
}
HostTensorDescriptor
mDesc
;
HostTensorDescriptor
mDesc
;
std
::
vector
<
T
>
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