Skip to content
Snippets Groups Projects
Commit 78b99eab authored by Olivier BICHLER's avatar Olivier BICHLER
Browse files

Fixed pybind

parent 13fdaf81
No related branches found
No related tags found
2 merge requests!50version 0.2.0,!32New proposal for handling tensor views
Pipeline #36965 failed
...@@ -21,7 +21,7 @@ class TensorImpl_cpu : public TensorImpl { ...@@ -21,7 +21,7 @@ class TensorImpl_cpu : public TensorImpl {
public: public:
static constexpr const char *Backend = "cpu"; 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 { bool operator==(const TensorImpl &otherImpl) const override final {
const auto& typedOtherImpl = reinterpret_cast<const TensorImpl_cpu<T> &>(otherImpl); const auto& typedOtherImpl = reinterpret_cast<const TensorImpl_cpu<T> &>(otherImpl);
...@@ -35,7 +35,7 @@ class TensorImpl_cpu : public TensorImpl { ...@@ -35,7 +35,7 @@ class TensorImpl_cpu : public TensorImpl {
return i == mNbElts; 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); return std::make_shared<TensorImpl_cpu<T>>(device, length);
} }
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment