From 78b99eab839e1fbba8a43afcc1e380e9983e584c Mon Sep 17 00:00:00 2001
From: Olivier BICHLER <olivier.bichler@cea.fr>
Date: Mon, 15 Jan 2024 16:08:07 +0100
Subject: [PATCH] Fixed pybind
---
include/aidge/backend/cpu/data/TensorImpl.hpp | 4 ++--
1 file changed, 2 insertions(+), 2 deletions(-)
diff --git a/include/aidge/backend/cpu/data/TensorImpl.hpp b/include/aidge/backend/cpu/data/TensorImpl.hpp
index 984e5c94..02f1ee5f 100644
--- a/include/aidge/backend/cpu/data/TensorImpl.hpp
+++ b/include/aidge/backend/cpu/data/TensorImpl.hpp
@@ -21,7 +21,7 @@ class TensorImpl_cpu : public TensorImpl {
public:
static constexpr const char *Backend = "cpu";
- TensorImpl_cpu(int device, NbElts_t length) : TensorImpl(Backend, device, length) {}
+ TensorImpl_cpu(DeviceIdx_t device, NbElts_t length) : TensorImpl(Backend, device, length) {}
bool operator==(const TensorImpl &otherImpl) const override final {
const auto& typedOtherImpl = reinterpret_cast<const TensorImpl_cpu<T> &>(otherImpl);
@@ -35,7 +35,7 @@ class TensorImpl_cpu : public TensorImpl {
return i == mNbElts;
}
- static std::shared_ptr<TensorImpl_cpu> create(int device, NbElts_t length) {
+ static std::shared_ptr<TensorImpl_cpu> create(DeviceIdx_t device, NbElts_t length) {
return std::make_shared<TensorImpl_cpu<T>>(device, length);
}
--
GitLab