Skip to content
Snippets Groups Projects

Changes following MR

Merged Olivier BICHLER requested to merge unified_params into main
All threads resolved!
2 files
+ 7
3
Compare changes
  • Side-by-side
  • Inline
Files
2
@@ -58,7 +58,11 @@ class TensorImpl_cuda : public TensorImpl, public TensorImpl_cuda_ {
@@ -58,7 +58,11 @@ class TensorImpl_cuda : public TensorImpl, public TensorImpl_cuda_ {
void *rawPtr() override {
void *rawPtr() override {
lazyInit(reinterpret_cast<void**>(&mData));
lazyInit(reinterpret_cast<void**>(&mData));
return mData;
return mData;
};
}
 
 
void* getRaw(std::size_t idx) {
 
return static_cast<void*>(static_cast<T*>(rawPtr()) + idx);
 
}
const cudnnTensorDescriptor_t& getCudnnTensorDesc() const override {
const cudnnTensorDescriptor_t& getCudnnTensorDesc() const override {
if (mCudnnTensor == nullptr) {
if (mCudnnTensor == nullptr) {
Loading