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

Merge branch 'tensorzeros' into 'dev'

Implement Tensor::zeros()

See merge request eclipse/aidge/aidge_backend_cuda!26
parents 60edd799 9729e8e9
No related branches found
No related tags found
2 merge requests!32version 0.2.1,!26Implement Tensor::zeros()
Pipeline #47927 passed
......@@ -89,6 +89,10 @@ public:
std::size_t scalarSize() const noexcept override { return sizeof(T); }
void zeros() override final {
CHECK_CUDA_STATUS(cudaMemset(rawPtr(), T(0), mNbElts * sizeof(T)));
}
void copy(const void *src, NbElts_t length, NbElts_t offset = 0) override {
AIDGE_ASSERT(length <= mData.size() || length <= mNbElts, "TensorImpl_cuda<{}>::copy(): copy length ({}) is above capacity ({})", typeid(T).name(), length, mNbElts);
const T* srcT = static_cast<const T *>(src);
......
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