Skip to content
GitLab
Explore
Sign in
Primary navigation
Search or go to…
Project
aidge_backend_cpu
Manage
Activity
Members
Labels
Plan
Issues
Issue boards
Milestones
Iterations
Wiki
Requirements
Code
Merge requests
Repository
Branches
Commits
Tags
Repository graph
Compare revisions
Snippets
Locked files
Build
Pipelines
Jobs
Pipeline schedules
Test cases
Artifacts
Deploy
Releases
Model registry
Operate
Environments
Monitor
Incidents
Analyze
Value stream analytics
Contributor analytics
CI/CD analytics
Repository analytics
Code review analytics
Issue analytics
Insights
Model experiments
Help
Help
Support
GitLab documentation
Compare GitLab plans
Community forum
Contribute to GitLab
Provide feedback
Terms and privacy
Keyboard shortcuts
?
Snippets
Groups
Projects
Show more breadcrumbs
Eclipse Projects
aidge
aidge_backend_cpu
Commits
ab92e03e
Commit
ab92e03e
authored
1 year ago
by
Cyril Moineau
Browse files
Options
Downloads
Plain Diff
Merge branch 'dev' of gitlab.eclipse.org:eclipse/aidge/aidge_backend_cpu into dev
parents
b94de272
bec69644
No related branches found
No related tags found
1 merge request
!31
Adding INT64 Tensor support
Pipeline
#36991
failed
1 year ago
Stage: build
Stage: test
Changes
1
Pipelines
1
Hide whitespace changes
Inline
Side-by-side
Showing
1 changed file
include/aidge/backend/cpu/data/TensorImpl.hpp
+28
-19
28 additions, 19 deletions
include/aidge/backend/cpu/data/TensorImpl.hpp
with
28 additions
and
19 deletions
include/aidge/backend/cpu/data/TensorImpl.hpp
+
28
−
19
View file @
ab92e03e
/********************************************************************************
* Copyright (c) 2023 CEA-List
*
* This program and the accompanying materials are made available under the
* terms of the Eclipse Public License 2.0 which is available at
* http://www.eclipse.org/legal/epl-2.0.
*
* SPDX-License-Identifier: EPL-2.0
*
********************************************************************************/
#ifndef AIDGE_CPU_DATA_TENSORIMPL_H_
#define AIDGE_CPU_DATA_TENSORIMPL_H_
...
...
@@ -10,9 +21,10 @@
#include
"aidge/utils/future_std/span.hpp"
namespace
Aidge
{
template
<
class
T
>
class
TensorImpl_cpu
:
public
TensorImpl
{
private:
private:
const
Tensor
&
mTensor
;
// Impl needs to access Tensor information, but is not
// supposed to change it!
/// Pointer to the data and its capacity
...
...
@@ -20,7 +32,7 @@ class TensorImpl_cpu : public TensorImpl {
/// If this instance own the data, std::unique_ptr manages it
std
::
unique_ptr
<
T
[]
>
mDataOwner
;
public:
public:
static
constexpr
const
char
*
Backend
=
"cpu"
;
TensorImpl_cpu
(
const
Tensor
&
tensor
)
:
TensorImpl
(
Backend
),
mTensor
(
tensor
)
{}
...
...
@@ -31,8 +43,8 @@ class TensorImpl_cpu : public TensorImpl {
std
::
size_t
i
=
0
;
for
(;
i
<
mTensor
.
size
()
&&
mData
[
i
]
==
typedOtherImpl
.
data
()[
i
]
;
++
i
)
{
*
(
mData
.
data
()
+
i
)
==
*
static_cast
<
const
T
*>
(
typedOtherImpl
.
rawPtr
(
i
))
;
++
i
)
{
}
return
i
==
mTensor
.
size
();
}
...
...
@@ -41,23 +53,20 @@ class TensorImpl_cpu : public TensorImpl {
return
std
::
make_unique
<
TensorImpl_cpu
<
T
>>
(
tensor
);
}
// native interface
auto
data
()
const
->
decltype
(
mData
.
data
())
{
return
mData
.
data
();
}
std
::
size_t
size
()
const
override
{
return
mData
.
size
();
}
std
::
size_t
scalarSize
()
const
override
{
return
sizeof
(
T
);
}
inline
std
::
size_t
size
()
const
noexcept
override
final
{
return
mData
.
size
();
}
inline
std
::
size_t
scalarSize
()
const
noexcept
override
final
{
return
sizeof
(
T
);
}
void
setDevice
(
DeviceIdx_t
device
)
override
{
void
setDevice
(
DeviceIdx_t
device
)
override
final
{
AIDGE_ASSERT
(
device
==
0
,
"device cannot be != 0 for CPU backend"
);
}
void
copy
(
const
void
*
src
,
NbElts_t
length
,
NbElts_t
offset
=
0
)
override
{
void
copy
(
const
void
*
src
,
NbElts_t
length
,
NbElts_t
offset
=
0
)
override
final
{
AIDGE_ASSERT
(
length
<=
mData
.
size
()
||
length
<=
mTensor
.
size
(),
"copy length is above capacity"
);
std
::
copy
(
static_cast
<
const
T
*>
(
src
),
static_cast
<
const
T
*>
(
src
)
+
length
,
static_cast
<
T
*>
(
rawPtr
())
+
offset
);
}
void
copyCast
(
const
void
*
src
,
NbElts_t
length
,
const
DataType
srcDt
)
override
{
void
copyCast
(
const
void
*
src
,
NbElts_t
length
,
const
DataType
srcDt
)
override
final
{
if
(
length
==
0
)
{
return
;
}
...
...
@@ -115,39 +124,39 @@ class TensorImpl_cpu : public TensorImpl {
}
}
void
copyFromDevice
(
const
void
*
src
,
NbElts_t
length
,
const
std
::
pair
<
std
::
string
,
DeviceIdx_t
>&
device
)
override
{
void
copyFromDevice
(
const
void
*
src
,
NbElts_t
length
,
const
std
::
pair
<
std
::
string
,
DeviceIdx_t
>&
device
)
override
final
{
AIDGE_ASSERT
(
device
.
first
==
Backend
,
"backend must match"
);
AIDGE_ASSERT
(
device
.
second
==
0
,
"device cannot be != 0 for CPU backend"
);
copy
(
src
,
length
);
}
void
copyFromHost
(
const
void
*
src
,
NbElts_t
length
)
override
{
inline
void
copyFromHost
(
const
void
*
src
,
NbElts_t
length
)
override
final
{
copy
(
src
,
length
);
}
void
copyToHost
(
void
*
dst
,
NbElts_t
length
)
const
override
{
void
copyToHost
(
void
*
dst
,
NbElts_t
length
)
const
override
final
{
AIDGE_ASSERT
(
length
<=
mData
.
size
()
||
length
<=
mTensor
.
size
(),
"copy length is above capacity"
);
const
T
*
src
=
static_cast
<
const
T
*>
(
rawPtr
());
std
::
copy
(
static_cast
<
const
T
*>
(
src
),
static_cast
<
const
T
*>
(
src
)
+
length
,
static_cast
<
T
*>
(
dst
));
}
void
*
rawPtr
(
NbElts_t
offset
=
0
)
override
{
void
*
rawPtr
(
NbElts_t
offset
=
0
)
override
final
{
lazyInit
();
return
(
mData
.
data
()
+
offset
);
};
const
void
*
rawPtr
(
NbElts_t
offset
=
0
)
const
override
{
const
void
*
rawPtr
(
NbElts_t
offset
=
0
)
const
override
final
{
AIDGE_ASSERT
(
mData
.
size
()
>=
mTensor
.
size
(),
"accessing uninitialized const rawPtr"
);
return
(
mData
.
data
()
+
offset
);
};
void
*
hostPtr
(
NbElts_t
offset
=
0
)
override
{
void
*
hostPtr
(
NbElts_t
offset
=
0
)
override
final
{
lazyInit
();
return
(
mData
.
data
()
+
offset
);
};
const
void
*
hostPtr
(
NbElts_t
offset
=
0
)
const
override
{
const
void
*
hostPtr
(
NbElts_t
offset
=
0
)
const
override
final
{
AIDGE_ASSERT
(
mData
.
size
()
>=
mTensor
.
size
(),
"accessing uninitialized const hostPtr"
);
return
(
mData
.
data
()
+
offset
);
};
...
...
This diff is collapsed.
Click to expand it.
Preview
0%
Loading
Try again
or
attach a new file
.
Cancel
You are about to add
0
people
to the discussion. Proceed with caution.
Finish editing this message first!
Save comment
Cancel
Please
register
or
sign in
to comment